test(sponsor): add test cases for sponsor apis

src/sponsors/tests/test_api.py

@@ -0,0 +1,72 @@
+import pytest
+
+from sponsors.models import OpenRole, Sponsor
+
+
+@pytest.fixture(autouse=True, scope='function')
+def test_data():
+    # sponsors
+    # platinum
+    sponsor_1 = Sponsor.objects.create(name='1', level=1)
+    # gold
+    sponsor_2 = Sponsor.objects.create(name='2', level=2, order=2)
+    sponsor_3 = Sponsor.objects.create(name='3', level=2, order=None)  # noqa
+    sponsor_4 = Sponsor.objects.create(name='4', level=2, order=1)  # noqa
+
+    # roles
+    OpenRole.objects.create(sponsor=sponsor_1, name='11', description='...')
+    OpenRole.objects.create(sponsor=sponsor_2, name='21', description='...')
+    OpenRole.objects.create(sponsor=sponsor_2, name='22', description='...')
+
+
+@pytest.mark.django_db
+class TestSponsorAPIView:
+    def test_should_get_sponsor_data(self, api_client):
+        # arrange: test_data fixture
+        # action
+        resp = api_client.get('/api/sponsors/')
+
+        # assert
+        assert resp.status_code == 200
+
+        data = {level_data['level_name']: level_data['sponsors'] for level_data in resp.data['data']}
+        assert list(data.keys()) == ['platinum', 'gold']
+
+        # assert: should have correct sponsor count
+        platinum_sponsors = data['platinum']
+        assert len(platinum_sponsors) == 1
+
+        gold_sponsors = data['gold']
+        assert len(gold_sponsors) == 3
+
+        # assert: gold sponsors should be in correct sequence
+        # small-order > large-order > null-order
+        gold_sponsor_names = [s['name_en_us'] for s in gold_sponsors]
+        assert gold_sponsor_names == ['4', '2', '3']
+
+    def test_should_get_role_data(self, api_client):
+        # arrange: test_data fixture
+        # action
+        resp = api_client.get('/api/sponsors/jobs/')
+
+        # assert
+        assert resp.status_code == 200
+
+        data = resp.json()['data']
+        assert len(data) == 2
+
+        # assert: sponsor 1
+        assert data[0]['sponsor_name'] == '1'
+
+        jobs_1 = data[0]['jobs']
+        assert len(jobs_1) == 1
+        assert jobs_1[0]['job_name_en_us'] == '11'
+
+        # assert: sponsor 2
+        assert data[1]['sponsor_name'] == '2'
+
+        jobs_2 = data[1]['jobs']
+        assert len(jobs_2) == 2
+
+        job_names = [j['job_name_en_us'] for j in jobs_2]
+        assert set(job_names) == {'21', '22'}