add tests

main/tests.py

@@ -1,3 +1,30 @@
-from django.test import TestCase
+from django.test import TestCase, SimpleTestCase, Client
+from .models import Sponsor
 
 # Create your tests here.
+
+
+class TestFrontend(SimpleTestCase):
+
+    def test_main(self):
+        # smoke test front end
+        client = Client()
+        re = client.get("/")
+        self.assertEqual(re.status_code, 200)
+        # giberrish
+        re = client.get("/akldsjfkajsdfk")
+        self.assertEqual(re.status_code, 200)
+
+
+class TestSponsorViewSet(TestCase):
+    fixtures = ["main/fixtures/sponsors.json"]
+
+    def setUp(self):
+        self.c = Client()
+
+    def test_view(self):
+        re = self.c.get("/api/sponsors/")
+        response = re.json()
+        self.assertEqual(re.status_code, 200)
+        self.assertEqual(response["count"],
+                         Sponsor.objects.all().count())