The changes made in this commit include:

- Added `link_preview` import in `models.py`
- Modified `get_preview` method in `Link` model to return `LinkPreview` object
- Added `SignalListViewTest` class in `tests.py`
- Added test methods for `SignalListView` in `tests.py`
- Modified `get_queryset` method in `SignalsListView` to call `get_base_diffs` for each object in the queryset
- Added `get_context_data` method in `SignalsListView`

Author: BrainIsDead

src/base/factories.py

@@ -0,0 +1,18 @@
+from factory import Sequence
+from factory.django import DjangoModelFactory
+from faker import Faker
+
+from base.models import Link, LinkTypeChoices
+
+fake = Faker()
+
+
+class LinkFactory(DjangoModelFactory):
+    """
+    A factory for the Link model.
+    """
+    class Meta:
+        model = Link
+
+    link_type = fake.random_element(LinkTypeChoices.values)
+    url = Sequence(lambda n: f'http://www.{n}.org/')

src/base/models.py

@@ -1,6 +1,6 @@
 from django.db import models
 from django.utils.translation import gettext_lazy as _
-from linkpreview import link_preview
+from linkpreview import LinkPreview, link_preview
 
 
 class LinkTypeChoices(models.TextChoices):
@@ -43,7 +43,7 @@ def __str__(self) -> str:
         """
         return self.url
 
-    def get_preview(self):
+    def get_preview(self) -> LinkPreview:
         """
         Returns a preview of the link using the link_preview library.
 

src/datasources/factories.py

@@ -0,0 +1,32 @@
+from factory import Sequence, SubFactory
+from factory.django import DjangoModelFactory
+from faker import Faker
+
+from datasources.models import DataSource, SourceSubdivision
+
+fake = Faker()
+
+
+class DataSourceFactory(DjangoModelFactory):
+    """
+    A factory for the DataSource model.
+    """
+    class Meta:
+        model = DataSource
+
+    name = Sequence(lambda n: f'{n}')
+    display_name = Sequence(lambda n: f'{n}')
+
+
+class SourceSubdivisionFactory(DjangoModelFactory):
+    """
+    A factory for the SourceSubdivision model.
+    """
+    class Meta:
+        model = SourceSubdivision
+
+    name = Sequence(lambda n: f'{n}')
+    display_name = Sequence(lambda n: f'{n}')
+    description = fake.text()
+    db_source = fake.word()
+    data_source = SubFactory(DataSourceFactory)

src/signals/factories.py

@@ -0,0 +1,114 @@
+from factory import (
+    Sequence,
+    SubFactory,
+    post_generation,
+)
+from factory.django import DjangoModelFactory
+from faker import Faker
+
+from base.factories import LinkFactory
+from datasources.factories import SourceSubdivisionFactory
+from signals.models import (
+    FormatChoices,
+    Geography,
+    HighValuesAreChoices,
+    Pathogen,
+    Signal,
+    SignalCategory,
+    SignalType,
+    TimeLabelChoices,
+    TimeTypeChoices,
+)
+
+fake = Faker()
+
+
+class GeographyFactory(DjangoModelFactory):
+    """
+    A factory for the Geography model.
+    """
+    class Meta:
+        model = Geography
+
+    name = Sequence(lambda n: f'{n}')
+
+
+class PathogenFactory(DjangoModelFactory):
+    """
+    A factory for the Pathogen model.
+    """
+    class Meta:
+        model = Pathogen
+
+    name = Sequence(lambda n: f'{n}')
+
+
+class SignalCategoryFactory(DjangoModelFactory):
+    """
+    A factory for the SignalCategory model.
+    """
+    class Meta:
+        model = SignalCategory
+
+    name = Sequence(lambda n: f'{n}')
+
+
+class SignalTypeFactory(DjangoModelFactory):
+    """
+    A factory for the SignalType model.
+    """
+    class Meta:
+        model = SignalType
+
+    name = Sequence(lambda n: f'{n}')
+
+
+class SignalFactory(DjangoModelFactory):
+    """
+    A factory for the Signal model.
+    """
+    name = Sequence(lambda n: f'{n}')
+    active = fake.boolean()
+    short_description = fake.text(max_nb_chars=500)
+    description = fake.text(max_nb_chars=500)
+    format = fake.random_element(FormatChoices.values)
+    time_type = fake.random_element(TimeTypeChoices.values)
+    time_label = fake.random_element(TimeLabelChoices.values)
+    category = SubFactory(SignalCategoryFactory)
+    is_smoothed = fake.boolean()
+    is_weighted = fake.boolean()
+    is_cumulative = fake.boolean()
+    has_stderr = fake.boolean()
+    high_values_are = fake.random_element(HighValuesAreChoices.values)
+    source = SubFactory(SourceSubdivisionFactory)
+
+    class Meta:
+        model = Signal
+
+    @post_generation
+    def generate_pathogens(self, create, extracted, **kwargs):
+        if not create or not extracted:
+            for _ in range(fake.random_int(min=1, max=10)):
+                self.pathogen.add(PathogenFactory())
+            return
+
+    @post_generation
+    def generate_signal_types(self, create, extracted, **kwargs):
+        if not create or not extracted:
+            for _ in range(fake.random_int(min=1, max=10)):
+                self.signal_type.add(SignalTypeFactory())
+            return
+
+    @post_generation
+    def generate_links(self, create, extracted, **kwargs):
+        if not create or not extracted:
+            for _ in range(fake.random_int(min=1, max=10)):
+                self.links.add(LinkFactory())
+            return
+
+    @post_generation
+    def generate_available_geography(self, create, extracted, **kwargs):
+        if not create or not extracted:
+            for _ in range(fake.random_int(min=1, max=10)):
+                self.available_geography.add(GeographyFactory())
+            return

src/signals/tests.py

@@ -0,0 +1,33 @@
+from django.test import TestCase
+from django.urls import reverse
+from faker import Faker
+
+from signals.factories import SignalFactory
+
+fake = Faker()
+
+
+class SignalListViewTest(TestCase):
+
+    def setUp(self):
+        for i in range(fake.random_int(min=1, max=100)):
+            SignalFactory()
+
+    def test_signal_list_view(self):
+        response = self.client.get(reverse('signals'))
+        self.assertEqual(response.status_code, 200)
+        self.assertTemplateUsed(response, 'signals/signal_list.html')
+
+    def test_signal_list_view_context(self):
+        response = self.client.get(reverse('signals'))
+        self.assertTrue('signals' in response.context)
+        self.assertTrue('is_paginated' in response.context)
+        self.assertTrue('page_obj' in response.context)
+        self.assertTrue('paginator' in response.context)
+        self.assertTrue('filter' in response.context)
+
+    def test_signal_list_view_filter_by_pathogen(self):
+        signal = SignalFactory()
+        response = self.client.get(reverse('signals'), {'pathogen': signal.pathogen.first().id})
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(response.context['signals'].count(), 1)

src/signals/views.py

@@ -16,6 +16,8 @@ class SignalsListView(ListView):
     def get_queryset(self):
         queryset = super().get_queryset()
         f = SignalFilter(self.request.GET, queryset=queryset)
+        for i in queryset:
+            i.get_base_diffs()
         return f.qs
 
     def get_context_data(self, **kwargs):