def test_saving_credentials(self): oauth_credentials = OAuth2Credentials( access_token='aaa', refresh_token='bbb', client_id='12345', client_secret='1111', token_uri=None, token_expiry=datetime.datetime.now(), user_agent=None ) storage = DBCredentialStorage() storage.put(oauth_credentials) credentials = self.session.query(GoogleCredential).first() self.assertEqual(credentials.access_token, 'aaa') self.assertEqual(credentials.client_id, '12345')