在測試中使用請求的用戶信息

Question

我使用user.userprofile中的信息來管理權限。

我如何使用它：

# List of status 
STATUS_CHOICES = (
    ('teacher', 'Teacher'), 
    ('student', 'Student'), 
) 

# More field for an user. 
class UserProfile(models.Model): 
    user = models.OneToOneField(User, unique=True) 
    status = models.CharField(max_length=10, choices=STATUS_CHOICES, default='student') 

現在，在視圖中，我有爲例：

# Only the owner view this template 
if request.user.userprofile.status != 'teacher': 
    return redirect('Penelope.views.home') 

如何我特林考：

def login_function(self, username, password, status): 
    # Create a new user 
    user = User.objects.create_user(username=username, password=password) 
    user.userprofile.status = status 
    user.save() 

    # Use test client to perform login 
    user = self.client.login(username=username, password=password) 
    response = self.client.post('/login/') 

class LoggedAsTeacherCase(TestCase): 
    def setUp(self): 
     # I create and log with a teacher user 
     login_function(self, 'teststudent', 'password', 'teacher') 

    def test_login(self): # Verify if login has been validated by server 
     response = self.client.get('/login/') 
     self.assertRedirects(response, '/') # <<< It is ok (My user is detected as connected, he is redirected) 

    def test_dashboard(self): 
     response = self.client.get('/') 
     self.assertEqual(response.status_code, 200) # <<< It is ok (Confirmation, my user is not redirected to login url, perfect) 

    def test_newcourse(self): 
     response = self.client.get('/newcourse/') 
     self.assertEqual(response.status_code, 200) # <<< /!\ It is false, I have a redirection (302), not a 200... 

所以，這意味着當視圖測試用戶是否是老師（cf：我的主題的頂部），它發送了False ...

我如何訪問此信息並執行測試？

Answer 1

我已經找到了問題！

user = User.objects.create_user(username=username, password=password) 
user.userprofile.status = status 
user.save() 

當我這樣做時，我想user = User.objects.create_user(username=username, password=password)返回一個用戶（false）。

所以，我必須做出一個請求：

user = User.objects.create_user(username=username, password=password) 
test = UserProfile.objects.filter(user=username) 
test.status = status 
test.save() 

它是OK！

Answer 2

您的status字段的定義不正確。 CharFields需要一個「max_length」屬性，該屬性是一個正整數。

將其更改爲以下：

models.CharField(max_length=..., choices=STATUS_CHOICES) 

參見here