@@ -21,8 +21,8 @@ def setUp(self):
21
21
self .user2 = UserFactory (
22
22
username = 'spameggs' ,
23
23
password = 'password' ,
24
- search_visibility = 0 ,
25
- email_privacy = 1 ,
24
+ search_visibility = User . SEARCH_PRIVATE ,
25
+ email_privacy = User . EMAIL_PRIVATE ,
26
26
public_profile = False ,
27
27
)
28
28
@@ -95,6 +95,22 @@ def test_membership_update(self):
95
95
response = self .client .post (url , post_data )
96
96
self .assertEqual (response .status_code , 302 )
97
97
98
+ def test_membership_update_404 (self ):
99
+ url = reverse ('users:user_membership_edit' )
100
+ self .assertFalse (self .user .has_membership )
101
+ self .client .login (username = self .user , password = 'password' )
102
+ response = self .client .get (url )
103
+ self .assertEqual (response .status_code , 404 )
104
+
105
+ def test_user_has_already_have_membership (self ):
106
+ # Should redirect to /membership/edit/ if user already
107
+ # has membership.
108
+ url = reverse ('users:user_membership_create' )
109
+ self .assertTrue (self .user2 .has_membership )
110
+ self .client .login (username = self .user2 , password = 'password' )
111
+ response = self .client .get (url )
112
+ self .assertRedirects (response , reverse ('users:user_membership_edit' ))
113
+
98
114
def test_user_update (self ):
99
115
self .client .login (username = 'username' , password = 'password' )
100
116
url = reverse ('users:user_profile_edit' )
@@ -267,3 +283,9 @@ def test_is_active_login(self):
267
283
response ,
268
284
'{}?next={}' .format (reverse ('account_login' ), url )
269
285
)
286
+
287
+ def test_user_list (self ):
288
+ url = reverse ('users:user_list' )
289
+ response = self .client .get (url )
290
+ self .assertEqual (response .status_code , 200 )
291
+ self .assertEqual (len (response .context ['user_list' ]), 1 )
0 commit comments