@@ -43,11 +43,16 @@ def setUp(self):
43
43
"email": "u@gogs.io",
44
44
"avatar_url": "/avatars/1"
45
45
}"""
46
- self .token = gogs_client .Token ("mytoken" )
46
+ self .token_json_str = """{
47
+ "name": "new token",
48
+ "sha1": "mytoken"
49
+ }"""
47
50
self .username_password = gogs_client .UsernamePassword (
48
51
"auth_username" , "password" )
49
52
self .expected_repo = gogs_client .GogsRepo .from_json (json .loads (self .repo_json_str ))
50
53
self .expected_user = gogs_client .GogsUser .from_json (json .loads (self .user_json_str ))
54
+ self .expected_token = gogs_client .TokenInfo .from_json (json .loads (self .token_json_str ))
55
+ self .token = self .expected_token .as_token ()
51
56
52
57
@responses .activate
53
58
def test_create_repo1 (self ):
@@ -238,6 +243,14 @@ def test_authenticated_user(self):
238
243
user = self .client .authenticated_user (self .token )
239
244
self .assert_users_equals (user , self .expected_user )
240
245
246
+ @responses .activate
247
+ def test_ensure_token (self ):
248
+ uri = self .path ("/users/{}/tokens" .format (self .username_password .username ))
249
+ responses .add (responses .GET , uri , body = "[]" , status = 200 )
250
+ responses .add (responses .POST , uri , body = self .token_json_str , status = 200 )
251
+ token = self .client .ensure_token (self .username_password , self .expected_token .name , self .username_password .username )
252
+ self .assertEqual (token .token , self .expected_token .sha1 )
253
+
241
254
# helper methods
242
255
243
256
@staticmethod
0 commit comments