"""C-06 关注 feed 回归：关注/未关注/互关/取关/匿名/分页对齐。"""
from django.test import TestCase
from rest_framework.test import APIClient

from article.models import Article
from user.models import Follow, FUser


class FollowingFeedTest(TestCase):
    def setUp(self):
        self.client = APIClient()
        self.a = FUser.objects.create_user(username='c06fa', password='x12345678')
        self.b = FUser.objects.create_user(username='c06fb', password='x12345678')
        self.c = FUser.objects.create_user(username='c06fc', password='x12345678')
        self.art_b = Article.objects.create(title='B文', content='x' * 500, author=self.b, status='published')
        self.art_c = Article.objects.create(title='C文', content='y' * 500, author=self.c, status='published')

    def test_following_only_shows_followed(self):
        Follow.objects.create(follower=self.a, following=self.b)
        self.client.force_authenticate(user=self.a)
        r = self.client.get('/article/articles/', {'feed': 'following'})
        self.assertEqual(r.status_code, 200)
        titles = [x['title'] for x in r.json()['data']['results']]
        self.assertEqual(titles, ['B文'])

    def test_unfollowed_empty_state(self):
        self.client.force_authenticate(user=self.a)
        r = self.client.get('/article/articles/', {'feed': 'following'})
        self.assertEqual(r.status_code, 200)
        self.assertEqual(r.json()['data']['count'], 0)
        self.assertEqual(r.json()['data']['results'], [])

    def test_mutual_follow_both_visible(self):
        Follow.objects.create(follower=self.a, following=self.b)
        Follow.objects.create(follower=self.a, following=self.c)
        self.client.force_authenticate(user=self.a)
        r = self.client.get('/article/articles/', {'feed': 'following'})
        self.assertEqual(r.status_code, 200)
        self.assertEqual(r.json()['data']['count'], 2)

    def test_unfollow_removes_from_feed(self):
        f = Follow.objects.create(follower=self.a, following=self.b)
        f.delete()
        self.client.force_authenticate(user=self.a)
        r = self.client.get('/article/articles/', {'feed': 'following'})
        self.assertEqual(r.status_code, 200)
        self.assertEqual(r.json()['data']['count'], 0)

    def test_anonymous_following_empty(self):
        Follow.objects.create(follower=self.a, following=self.b)
        r = self.client.get('/article/articles/', {'feed': 'following'})
        self.assertEqual(r.status_code, 200)
        self.assertEqual(r.json()['data']['count'], 0)

    def test_normal_list_unaffected(self):
        Follow.objects.create(follower=self.a, following=self.b)
        self.client.force_authenticate(user=self.a)
        r = self.client.get('/article/articles/')
        self.assertEqual(r.status_code, 200)
        self.assertEqual(r.json()['data']['count'], 2)

    def test_draft_not_in_feed(self):
        Follow.objects.create(follower=self.a, following=self.b)
        self.art_b.status = 'draft'
        self.art_b.save(update_fields=['status'])
        self.client.force_authenticate(user=self.a)
        r = self.client.get('/article/articles/', {'feed': 'following'})
        self.assertEqual(r.status_code, 200)
        self.assertEqual(r.json()['data']['count'], 0)