Add tests

Signed-off-by: Tushar Goel <tushar.goel.dav@gmail.com>

Author: TG1999

vulnerabilities/tests/test_forms.py

@@ -0,0 +1,19 @@
+from django.test import TestCase
+from rest_framework import status
+
+from vulnerabilities.forms import CVEForm
+from vulnerabilities.models import Vulnerability
+
+
+class TestCVEForm(TestCase):
+    def setUp(self) -> None:
+        vuln1 = Vulnerability.objects.create(summary="test-vuln1", vulnerability_id="VCID-1234")
+        self.id = vuln1.id
+
+    def test_cve_form(self):
+        form = CVEForm(data={"vuln_id": "vcid-1234"})
+        assert form.is_valid()
+
+    def test_client(self):
+        response = self.client.get(f"/vulnerabilities/{self.id}?vuln_id=vcid-1234")
+        self.assertContains(response, "test-vuln1", status_code=200)