我正在针对一个API写一些单元测试,该API要么返回所有书籍,要么只返回查询参数中给定类型的书籍。当我在本地开发服务器中点击它时,这似乎正常。但是,如果在我的单元测试中指定了类型,它甚至不会进入else语句。
我的单元测试如下:
class TitlesAndBlurbsListTestCase(APITestCase):
def setUp(self):
# Creates a lot of books with genre horror
# and books not in the horror genre
def test_horror_genre(self):
# Ensure that screener can see all the available books
self.client.login(username='b', password='b')
response = self.client.get('/api/titles-and-blurbs/?genre=horror')
self.assertEqual(response.status_code, status.HTTP_200_OK)
# Ensure that the screener gets all horror books at first
horror_books = TitlesAndBlurbs.objects.filter(genre='horror')
# I keep getting an assertion error here - it returns all the books
self.assertEqual(len(response.data), horror_books.count())
我的api viewset看起来像这样
class TitlesAndBlurbsListViewSet(viewsets.mixins.ListModelMixin,
viewsets.mixins.RetrieveModelMixin,
viewsets.GenericViewSet):
model = TitlesAndBlurbs
permission_classes = [ScreenerPermissions]
serializer_class = TitlesAndBlurbsSerializer
def get_queryset(self):
if self.action == 'list':
genre = self.request.QUERY_PARAMS.get('genre', None)
if not genre:
print 'i dont have query params of genre'
TitlesAndBlurbs.objects.all()
else:
print genre
TitlesAndBlurbs.objects.filter(genre=genre)
return TitlesAndBlurbs.objects.all()
我的网址/路由器看起来像
router.register(r'api/titles-and-blurbs', TitlesAndBlurbsListViewSet)
当我点击网址' api / titles-and-blurbs /?genre = horror'在我的浏览器中,我得到了打印声明和标题以及具有类型恐怖的模糊。但是,当我打中 测试套件,我没有得到打印语句类型我得到的打印声明“我没有查询参数'”它返回所有书籍。任何帮助都是真的 赞赏。