Commit a50133f2 authored by PidgeyL's avatar PidgeyL
Browse files

more unit test for API

parent 28372eb1
Loading
Loading
Loading
Loading
+15 −8
Original line number Diff line number Diff line
@@ -49,9 +49,16 @@ class TestAPI(unittest.TestCase):
    def response_check(self, url, expected):
        url = 'http://localhost:5000/api/%s'%url
        data = json.loads((urlopen(url).read()).decode('utf8'))
        print(data)
        self.assertEqual(expected, data)

    def test_cpe2_3(self):
        expected = "cpe:2.3:a:test:test"
        self.response_check("cpe2.3/cpe:/a:test:test", expected)

    def test_cpe2_2(self):
        expected = "cpe:/a:test:test"
        self.response_check("cpe2.2/cpe:2.3:a:test:test", expected)

    def test_api_capec(self):
        expected = {"id": "10000",
                    "name": "test_capec",