~rvb/maas/transaction-1.7-bug-1409852

« back to all changes in this revision

Viewing changes to src/maasserver/tests/test_commands.py

merged upstream

Show diffs side-by-side

added added

removed removed

Lines of Context:
194
194
        stderr = BytesIO()
195
195
        out = BytesIO()
196
196
        stdout = getwriter("UTF-8")(out)
197
 
        user = factory.make_user()
 
197
        user = factory.make_User()
198
198
        call_command(
199
199
            'apikey', username=user.username, stderr=stderr, stdout=stdout)
200
200
        self.assertEqual('', stderr.getvalue().strip())
209
209
        stderr = BytesIO()
210
210
        out = BytesIO()
211
211
        stdout = getwriter("UTF-8")(out)
212
 
        user = factory.make_user()
 
212
        user = factory.make_User()
213
213
        num_keys = len(user.get_profile().get_authorisation_tokens())
214
214
        call_command(
215
215
            'apikey', username=user.username, generate=True, stderr=stderr,
226
226
    def test_apikey_deletes_key(self):
227
227
        stderr = BytesIO()
228
228
        stdout = BytesIO()
229
 
        user = factory.make_user()
 
229
        user = factory.make_User()
230
230
        existing_token = get_one(
231
231
            user.get_profile().get_authorisation_tokens())
232
232
        token_string = convert_tuple_to_string(
240
240
        self.assertEqual(0, len(keys_after))
241
241
 
242
242
    def test_apikey_rejects_mutually_exclusive_options(self):
243
 
        user = factory.make_user()
 
243
        user = factory.make_User()
244
244
        error_text = assertCommandErrors(
245
245
            self, 'apikey',
246
246
            username=user.username, generate=True, delete="foo")
249
249
            error_text)
250
250
 
251
251
    def test_apikey_rejects_deletion_of_bad_key(self):
252
 
        user = factory.make_user()
 
252
        user = factory.make_User()
253
253
        error_text = assertCommandErrors(
254
254
            self, 'apikey',
255
255
            username=user.username, delete="foo")
259
259
 
260
260
    def test_api_key_rejects_deletion_of_nonexistent_key(self):
261
261
        stderr = BytesIO()
262
 
        user = factory.make_user()
 
262
        user = factory.make_User()
263
263
        existing_token = get_one(
264
264
            user.get_profile().get_authorisation_tokens())
265
265
        token_string = convert_tuple_to_string(