import pytest

from ii_drf_testtools.utils import ListAPITest
from news.tests.factories import NewsItemFactory


@pytest.mark.django_db(transaction=True)
class TestNewsList(ListAPITest):
    api_url = reverse('rest_api:news:list', kwargs={'version': '1.0'})
    only_custom = False
    status_codes_anonymous = {'ALL': 403}
    status_codes_authenticated = {'ALL': 405, 'GET': 200}

    def setup_list_data(self):
        current_user = self.get_user()

        self.active_newsitem = NewsItemFactory(
            name='Active newsitem', is_active=True, user=current_user)
        self.active_newsitem2 = NewsItemFactory(
            name='Active newsitem2', is_active=True, user=current_user)
        self.inactive_newsitem = NewsItemFactory(
            name='Inactive newsitem', is_active=False, user=current_user)
        self.other_newsteam = NewsItemFactory(
            name='Other newsitem', is_active=True)

    def assert_list_response_authenticated(self, response):
        assert response.json() == [
            {
                'id': self.active_newsitem.id,
                'name': 'Active newsitem',
            },
            {
                'id': self.active_newsitem2.id,
                'name': 'Active newsitem2',
            }]