← Back to branch summary

~nataliabidart/ubuntu-sso-client/find-me-bin-dir

« back to all changes in this revision

Viewing changes to ubuntu_sso/qt/tests/test_proxy_dialog.py

  • Committer: Tarmac
  • Author(s): Manuel de la Pena
  • Date: 2012-02-14 12:52:01 UTC
  • mfrom: (846.7.19 load-creds-retry)
  • Revision ID: tarmac-20120214125201-vj7wzqcemrbq813v
Adds the code that will load the creds from the keyring when we retry.

Show diffs side-by-side

added added

removed removed

Lines of Context:
ubuntu_sso/qt/tests/test_proxy_dialog.py
165
165
        self.assertIn('domain_label', dialog.ui.called)
166
166
        self.assertIn(('setText', domain), dialog.ui.called['domain_label'])
167
167
 
168
 
    def test_retry(self):
 
168
    @defer.inlineCallbacks
 
169
    def test_retry_present_creds(self):
 
170
        """Test that we do set the error label correctly."""
 
171
        creds = dict(username='username', password='password')
 
172
        called = []
 
173
        dialog = proxy_dialog.ProxyCredsDialog(retry=True)
 
174
 
 
175
        def fake_get_credentials(app_name):
 
176
            """Fake the get creds."""
 
177
            called.append(('get_credentials', app_name))
 
178
            return defer.succeed(creds)
 
179
 
 
180
        self.patch(dialog.keyring, 'get_credentials', fake_get_credentials)
 
181
 
 
182
        yield dialog._load_creds()
 
183
        self.assertIn('get_credentials', called[0])
 
184
        self.assertIn('error_label', dialog.ui.called)
 
185
        self.assertIn(('setVisible', True), dialog.ui.called['error_label'])
 
186
        self.assertIn(('setText', creds['username']),
 
187
                      dialog.ui.called['username_entry'])
 
188
        self.assertIn(('setText', creds['password']),
 
189
                      dialog.ui.called['password_entry'])
 
190
 
 
191
    @defer.inlineCallbacks
 
192
    def test_retry_none_creds(self):
 
193
        """Test that we do set the error label correctly."""
 
194
        called = []
 
195
        dialog = proxy_dialog.ProxyCredsDialog(retry=True)
 
196
 
 
197
        def fake_get_credentials(app_name):
 
198
            """Fake the get creds."""
 
199
            called.append(('get_credentials', app_name))
 
200
            return defer.succeed(None)
 
201
 
 
202
        self.patch(dialog.keyring, 'get_credentials', fake_get_credentials)
 
203
 
 
204
        yield dialog._load_creds()
 
205
        self.assertIn('get_credentials', called[0])
 
206
        self.assertIn('error_label', dialog.ui.called)
 
207
        self.assertIn(('setVisible', True), dialog.ui.called['error_label'])
 
208
        self.assertNotIn('username_entry', dialog.ui.called)
 
209
        self.assertNotIn('password_entry', dialog.ui.called)
 
210
 
 
211
    @defer.inlineCallbacks
 
212
    def test_retry_exception_creds(self):
 
213
        """Test that we do set the error label correctly."""
 
214
        called = []
 
215
        dialog = proxy_dialog.ProxyCredsDialog(retry=True)
 
216
 
 
217
        def fake_get_credentials(app_name):
 
218
            """Fake the get creds."""
 
219
            called.append(('get_credentials', app_name))
 
220
            return defer.fail(Exception('Fail test'))
 
221
 
 
222
        self.patch(dialog.keyring, 'get_credentials', fake_get_credentials)
 
223
 
 
224
        yield dialog._load_creds()
 
225
        self.assertIn('get_credentials', called[0])
 
226
        self.assertIn('error_label', dialog.ui.called)
 
227
        self.assertIn(('setVisible', True), dialog.ui.called['error_label'])
 
228
        self.assertNotIn('username_entry', dialog.ui.called)
 
229
        self.assertNotIn('password_entry', dialog.ui.called)
 
230
 
 
231
    def test_is_not_retry(self):
169
232
        """Test that we do set the error label correctly."""
170
233
        dialog = proxy_dialog.ProxyCredsDialog(retry=False)
171
234
        self.assertIn('error_label', dialog.ui.called)
172
235
        self.assertIn(('setVisible', False), dialog.ui.called['error_label'])
173
236
 
174
 
    def test_is_not_retry(self):
175
 
        """Test that we do set the error label correctly."""
176
 
        dialog = proxy_dialog.ProxyCredsDialog(retry=True)
177
 
        self.assertIn('error_label', dialog.ui.called)
178
 
        self.assertIn(('setVisible', True), dialog.ui.called['error_label'])
179
 
 
180
237
    def test_icon(self):
181
238
        """Test that we correctly set the icon."""
182
239
        dialog = proxy_dialog.ProxyCredsDialog()