test_views.py 文件源码

def test_categories_filter(self):
    """
    Test searching with categories filter returns only results filtered by category
    """
    category_id1 = Category.objects.all().order_by('pk')[0].pk
    category_id2 = Category.objects.all().order_by('pk')[1].pk

    response = self.client.get(reverse("search-projects-list") + "?category=" + str(category_id1), format="json")
    self.assertEqual(len(response.data["results"]), 1)
    self.assertEqual(str(response.data["results"][0]["name"]), "test project")

    response = self.client.get(reverse("search-projects-list") + "?category=" + str(category_id2), format="json")
    self.assertEqual(len(response.data["results"]), 1)
    self.assertEqual(str(response.data["results"][0]["name"]), "test project2")

    response = self.client.get(reverse("search-projects-list") + "?category={},{}".format(category_id1, category_id2), format="json")
    self.assertEqual(len(response.data["results"]), 2)