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

« back to all changes in this revision

Viewing changes to src/maasserver/views/tests/test_prefs.py

merged upstream

Show diffs side-by-side

added added

removed removed

Lines of Context:
170
170
    def test_key_can_be_added_if_same_key_already_setup_for_other_user(self):
171
171
        self.client_log_in()
172
172
        key_string = get_data('data/test_rsa0.pub')
173
 
        key = SSHKey(user=factory.make_user(), key=key_string)
 
173
        key = SSHKey(user=factory.make_User(), key=key_string)
174
174
        key.save()
175
175
        response = self.client.post(
176
176
            reverse('prefs-add-sshkey'), {'key': key_string})
183
183
    def test_delete_key_GET(self):
184
184
        # The 'Delete key' page displays a confirmation page with a form.
185
185
        self.client_log_in()
186
 
        key = factory.make_sshkey(self.logged_in_user)
 
186
        key = factory.make_SSHKey(self.logged_in_user)
187
187
        del_link = reverse('prefs-delete-sshkey', args=[key.id])
188
188
        response = self.client.get(del_link)
189
189
        doc = fromstring(response.content)
199
199
 
200
200
    def test_delete_key_GET_cannot_access_someone_elses_key(self):
201
201
        self.client_log_in()
202
 
        key = factory.make_sshkey(factory.make_user())
 
202
        key = factory.make_SSHKey(factory.make_User())
203
203
        del_link = reverse('prefs-delete-sshkey', args=[key.id])
204
204
        response = self.client.get(del_link)
205
205
 
209
209
        # Deleting a nonexistent key requires no confirmation.  It just
210
210
        # "succeeds" instantaneously.
211
211
        self.client_log_in()
212
 
        key = factory.make_sshkey(self.logged_in_user)
 
212
        key = factory.make_SSHKey(self.logged_in_user)
213
213
        del_link = reverse('prefs-delete-sshkey', args=[key.id])
214
214
        key.delete()
215
215
        response = self.client.get(del_link)
218
218
    def test_delete_key_POST(self):
219
219
        # A POST request deletes the key, and redirects to the prefs.
220
220
        self.client_log_in()
221
 
        key = factory.make_sshkey(self.logged_in_user)
 
221
        key = factory.make_SSHKey(self.logged_in_user)
222
222
        del_link = reverse('prefs-delete-sshkey', args=[key.id])
223
223
        response = self.client.post(del_link, {'post': 'yes'})
224
224
 
229
229
        # Deleting a key that's already been deleted?  Basically that's
230
230
        # success.
231
231
        self.client_log_in()
232
 
        key = factory.make_sshkey(self.logged_in_user)
 
232
        key = factory.make_SSHKey(self.logged_in_user)
233
233
        del_link = reverse('prefs-delete-sshkey', args=[key.id])
234
234
        key.delete()
235
235
        response = self.client.post(del_link, {'post': 'yes'})