Unify api serializers (#16)

netbox_slm/tests/test_models.py

@@ -0,0 +1,36 @@
+from django.test import TestCase
+
+from netbox_slm.models import SoftwareProduct, SoftwareProductVersion, SoftwareProductInstallation
+
+
+class ModelTestCase(TestCase):
+    def setUp(self):
+        self.p_name = "test product"
+        self.v_name = "test version"
+
+        self.software_product = SoftwareProduct.objects.create(name=self.p_name)
+        self.software_product_version = SoftwareProductVersion.objects.create(
+            name=self.v_name, software_product=self.software_product
+        )
+        self.software_product_installation = SoftwareProductInstallation.objects.create(
+            software_product=self.software_product, version=self.software_product_version
+        )
+
+    def test_model_name(self):
+        self.assertEqual(self.p_name, str(self.software_product))
+        self.assertEqual(self.v_name, str(self.software_product_version))
+        self.assertTrue(str(self.software_product_installation).isnumeric())  # should be PK
+
+    def test_absolute_url(self):
+        self.assertEqual("/plugins/slm/software-products/1/", self.software_product.get_absolute_url())
+        self.assertEqual("/plugins/slm/versions/1/", self.software_product_version.get_absolute_url())
+        self.assertEqual("/plugins/slm/installations/1/", self.software_product_installation.get_absolute_url())
+
+    def test_get_installation_count(self):
+        installation_ss = '<a href="/plugins/slm/installations/?q={}">1</a>'
+        self.assertEqual(installation_ss.format(self.p_name), self.software_product.get_installation_count())
+        self.assertEqual(installation_ss.format(self.v_name), self.software_product_version.get_installation_count())
+
+    def test_product_installation_methods(self):
+        self.assertEqual("virtualmachine", self.software_product_installation.render_type())
+        self.assertIsNone(self.software_product_installation.get_platform())

netbox_slm/tests/test_smoke.py

@@ -1,16 +0,0 @@
-from django.test import TestCase
-
-from netbox_slm.models import SoftwareProduct, SoftwareProductVersion, SoftwareProductInstallation
-
-
-class ModelTestCase(TestCase):
-    def test_smoke(self):
-        software_product = SoftwareProduct.objects.create(name="test product")
-        software_product_version = SoftwareProductVersion.objects.create(
-            name="test version", software_product=software_product
-        )
-        software_product_installation = SoftwareProductInstallation.objects.create(
-            software_product=software_product, version=software_product_version
-        )
-
-        self.assertEqual('virtualmachine', software_product_installation.render_type())