import pytest
from django.contrib.auth.models import User
from rest_framework.test import APIClient
from books.models import Book, Category
from books.factories import BookFactory, AuthorFactory
 
@pytest.fixture
def api_client():
    return APIClient()
 
@pytest.fixture
def author(db):
    return AuthorFactory()
 
@pytest.fixture
def published_book(db, author):
    return BookFactory(is_published=True, author=author)
 
@pytest.mark.django_db
class TestBookAPI:
    def test_book_list_returns_only_published(self, api_client, published_book):
        response = api_client.get('/api/books/')
        assert response.status_code == 200
        assert len(response.data['results']) == 1
        assert published_book.title in str(response.data)
 
    def test_unpublished_book_not_in_list(self, api_client, db):
        unpublished = BookFactory(is_published=False)
        response = api_client.get('/api/books/')
        assert response.status_code == 200
        assert unpublished.title not in str(response.data)