lib.itmens/users/tests.py

34 lines
1.2 KiB
Python
Raw Normal View History

2020-05-01 22:46:15 +08:00
from django.test import TestCase
2023-07-07 02:02:48 -04:00
from .models import *
2020-05-01 22:46:15 +08:00
2023-07-07 02:02:48 -04:00
class UserTest(TestCase):
def setUp(self):
self.alice = User.objects.create(
mastodon_site="MySpace", mastodon_username="Alice"
)
self.bob = User.objects.create(mastodon_site="KKCity", mastodon_username="Bob")
def test_local_follow(self):
self.alice.follow(self.bob)
self.assertTrue(
Follow.objects.filter(owner=self.alice, target=self.bob).exists()
)
self.assertEqual(self.alice.following, [self.bob.pk])
self.assertTrue(self.alice.is_following(self.bob))
self.assertTrue(self.bob.is_followed_by(self.alice))
self.alice.follow(self.bob)
self.assertEqual(
Follow.objects.filter(owner=self.alice, target=self.bob).count(), 1
)
self.assertEqual(self.alice.following, [self.bob.pk])
self.alice.unfollow(self.bob)
self.assertFalse(
Follow.objects.filter(owner=self.alice, target=self.bob).exists()
)
self.assertFalse(self.alice.is_following(self.bob))
self.assertFalse(self.bob.is_followed_by(self.alice))
self.assertEqual(self.alice.following, [])