Unit test sponsors page

Author: berinhard

tests/common/db/base.py

@@ -46,3 +46,26 @@ def fuzz(self):
         chars = string.ascii_letters + string.digits
         username = "".join(random.choice(chars) for i in range(12))
         return "@".join([username, self.domain])
+
+
+class FuzzyList(fuzzy.BaseFuzzyAttribute):
+    def __init__(self, item_factory, item_kwargs=None, size=1, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.item_factory = item_factory
+        self.item_kwargs = item_kwargs or {}
+        self.size = size
+
+    def fuzz(self):
+        return [self.item_factory(**self.item_kwargs).fuzz() for i in range(self.size)]
+
+
+class FuzzyUrl(fuzzy.BaseFuzzyAttribute):
+    def __init__(self, domain="example.com", is_secure=False, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.domain = domain
+        self.protocol = "https" if is_secure else "http"
+
+    def fuzz(self):
+        chars = string.ascii_letters
+        path = "".join(random.choice(chars) for i in range(12))
+        return f"{self.protocol}://{self.domain}/{path}"

tests/common/db/sponsors.py

@@ -0,0 +1,38 @@
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import datetime
+
+from factory import fuzzy
+
+from warehouse.sponsors.models import Sponsor
+
+from .base import FuzzyList, FuzzyUrl, WarehouseFactory
+
+
+class SponsorFactory(WarehouseFactory):
+    class Meta:
+        model = Sponsor
+
+    name = fuzzy.FuzzyText(length=12)
+    service = fuzzy.FuzzyText(length=12)
+    activity = FuzzyList(fuzzy.FuzzyText, {"length": 30}, size=2)
+
+    url = FuzzyUrl()
+    color_logo_url = FuzzyUrl()
+    white_logo_url = FuzzyUrl()
+
+    footer = True
+    psf_sponsor = True
+    infra_sponsor = False
+    one_time = False
+    sidebar = True

tests/unit/sponsors/__init__.py

@@ -0,0 +1,11 @@
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.

tests/unit/sponsors/test_views.py

@@ -0,0 +1,29 @@
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import pretend
+import pytest
+
+from warehouse.sponsors import views
+from warehouse.sponsors.models import Sponsor
+
+from ...common.db.sponsors import SponsorFactory
+
+
+class TestSponsorsPage:
+    def test_list_sponsors(self, db_request):
+        sponsors = [SponsorFactory.create() for i in range(3)]
+
+        result = views.display_sponsors_page(db_request)
+
+        expected = db_request.db.query(Sponsor).all()
+        assert result == {"sponsors": expected}