moanos/Notfellchen
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: Add test for adoption_notice_fits_search

Browse Source
This commit is contained in:
moanos [he/him] 2025-01-01 17:23:38 +01:00
parent 7035b1642e
commit 164ba7def2
1 changed files with 65 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

82
src/tests/test_search.py
View File

@ -1,43 +1,73 @@
from time import sleep
from django.test import TestCase from django.test import TestCase
from django.urls import reverse
from fellchensammlung.models import SearchSubscription, User, TrustLevel, AdoptionNotice, Location, SexChoicesWithAll, \ from fellchensammlung.models import SearchSubscription, User, TrustLevel, AdoptionNotice, Location, SexChoicesWithAll, \
Animal, Species Animal, Species, AdoptionNoticeNotification, SexChoices
from fellchensammlung.tools import search
from model_bakery import baker from model_bakery import baker
from fellchensammlung.tools.geo import GeoAPI from fellchensammlung.tools.geo import GeoAPI
from fellchensammlung.tools.search import Search from fellchensammlung.tools.search import Search, notify_search_subscribers
class TestSearch(TestCase): class TestSearch(TestCase):
@classmethod @classmethod
def setUpTestData(cls): def setUpTestData(cls):
cls.test_user0 = User.objects.create_user(username='testuser0', cls.test_user0 = User.objects.create_user(username='testuser0',
first_name="Admin", first_name="Admin",
last_name="BOFH", last_name="BOFH",
password='12345') password='12345')
test_user1 = User.objects.create_user(username='testuser1', cls.test_user1 = User.objects.create_user(username='testuser1',
first_name="Max", first_name="Max",
last_name="Müller", last_name="Müller",
password='12345') password='12345')
cls.test_user2 = User.objects.create_user(username='testuser2',
first_name="Miriam",
last_name="Müller",
password='12345')
cls.test_user0.trust_level = TrustLevel.ADMIN cls.test_user0.trust_level = TrustLevel.ADMIN
cls.test_user0.save() cls.test_user0.save()
adoption1 = baker.make(AdoptionNotice, name="TestAdoption1", owner=cls.test_user0)
rat = baker.make(Species, name="Farbratte") rat = baker.make(Species, name="Farbratte")
cls.location_stuttgart = Location.get_location_from_string("Stuttgart")
cls.location_tue = Location.get_location_from_string("Kirchentellinsfurt")
cls.location_berlin = Location.get_location_from_string("Berlin")
cls.adoption1 = baker.make(AdoptionNotice, name="TestAdoption1", owner=cls.test_user0,
location=cls.location_stuttgart)
rat1 = baker.make(Animal, rat1 = baker.make(Animal,
name="Rat1", name="Rat1",
adoption_notice=adoption1, adoption_notice=cls.adoption1,
species=rat, species=rat,
sex=SexChoices.MALE,
description="Eine unglaublich süße Ratte") description="Eine unglaublich süße Ratte")
cls.adoption1.set_active()
adoption2 = baker.make(AdoptionNotice, name="TestAdoption2", owner=cls.test_user0) cls.adoption2 = baker.make(AdoptionNotice, name="TestAdoption2", owner=cls.test_user0, location=cls.location_berlin)
cls.location = Location.get_location_from_string("Stuttgart") rat2 = baker.make(Animal,
name="Rat2",
adoption_notice=cls.adoption2,
species=rat,
sex=SexChoices.FEMALE,
description="Eine unglaublich süße Ratte")
cls.adoption2.set_active()
cls.subscription1 = SearchSubscription.objects.create(owner=cls.test_user1,
location=cls.location_tue,
max_distance=50,
sex=SexChoicesWithAll.MALE)
cls.subscription2 = SearchSubscription.objects.create(owner=cls.test_user2,
location=cls.location_berlin,
max_distance=50,
sex=SexChoicesWithAll.ALL)
def test_equals(self): def test_equals(self):
search_subscription1 = SearchSubscription.objects.create(owner=self.test_user0, search_subscription1 = SearchSubscription.objects.create(owner=self.test_user0,
location=self.location, location=self.location_stuttgart,
sex=SexChoicesWithAll.ALL, sex=SexChoicesWithAll.ALL,
max_distance=100 max_distance=100
) )
@ -52,6 +82,24 @@ class TestSearch(TestCase):
self.assertEqual(search_subscription1, search1) self.assertEqual(search_subscription1, search1)
def test_notification(self): def test_adoption_notice_fits_search(self):
self.client.login(username='testuser0', password='12345') search1 = Search(search_subscription=self.subscription1)
self.assertTrue(search1.adoption_notice_fits_search(self.adoption1))
self.assertFalse(search1.adoption_notice_fits_search(self.adoption2))
search2 = Search(search_subscription=self.subscription2)
self.assertFalse(search2.adoption_notice_fits_search(self.adoption1))
self.assertTrue(search2.adoption_notice_fits_search(self.adoption2))
def test_notification(self):
"""
Simulates as if a new adoption notice is added and checks if Notifications are triggered.
Must simulate adding a new adoption notice, the actual celery task can not be tested as celery can not
be tested as it can't access the in-memory test database.
https://stackoverflow.com/questions/46530784/make-django-test-case-database-visible-to-celery/46564964#46564964
"""
notify_search_subscribers(self.adoption1)
self.assertTrue(AdoptionNoticeNotification.objects.filter(user=self.test_user1).exists())
self.assertFalse(AdoptionNoticeNotification.objects.filter(user=self.test_user1).exists())