def test_settings_storage(self): path = tempfile.mktemp('.json') storage = SettingStorage(path) title = u'Hello solog' storage.set('test:blog:title', title) storage.write() self.assertEqual(storage.get('test:blog:title'), title) fp = open(path, 'r') self.assertTrue(json.load(fp).keys(), ['blog:title']) fp.close() self.assertEqual(storage.get('blog:none', 'default'), 'default') storage.set('test:dropbox:access_token', 'access_token') storage.write() self.assertEqual(json.load(open(path, 'r')).keys(), ['test:blog:title', 'test:dropbox:access_token'])
def test_auth(self, MockSession): with mock.patch('views.SETTINGS_FILE', self.settings_file): AccessToken = namedtuple('AccessToken', "key secret") sess = MockSession() sess.obtain_request_token.return_value = AccessToken(key="key", secret="secret") http_response = views.dropbox_auth() self.assertEqual(http_response.status_code, 302) self.assertTrue(sess.obtain_request_token.called) self.assertEqual(views.dropbox_callback(), 'Oops, something is wrong ...') with mock.patch.object(BaseRequest, 'get_cookie', lambda *args: "key&secret"): sess.obtain_access_token.return_value = AccessToken(key="key", secret="secret") # should redirect to /dropbox/sync self.assertRaises(HTTPResponse, lambda: views.dropbox_callback()) self.assertTrue(sess.obtain_access_token.called) storage = SettingStorage(self.settings_file) self.assertEqual(storage.get('dropbox:access_token_secret'), 'secret') self.assertEqual(storage.get('dropbox:access_token_key'), 'key')