与不同的Django用户调用Django Unittest URL

时间:2016-12-13 11:39:11

标签: python django api

我在Django中有以下unittest类。

class TmpUrlTest(TestCase):
    def setUp(self):
        self.client = Client()
        test_group = Group('test_group')
        test_group.save()
        self.api_url = "/tmp_url/"

        self.test_user = User.objects.create(username='user1')
        self.test_user.set_password('pass')
        self.test_user.save()
        self.test_user.groups.add(test_group)

    def test_url(self):
        response = self.client.get(self.api_url)
        self.assertEqual(response.status_code, 200)

现在,在我的来电中,我希望用户为user1。有什么方法可以在打电话时这样做吗?

1 个答案:

答案 0 :(得分:2)

您需要在发送请求之前以用户身份登录

self.client.login(username='user1', password='pass')