← Back to branch summary

~fom-dev/fom/trunk

« back to all changes in this revision

Viewing changes to tests/test_db.py

  • Committer: Nicholas Tollervey
  • Date: 2011-12-03 13:34:09 UTC
  • mfrom: (154.1.1 add-oauth-calls-899576)
  • Revision ID: ntoll@ntoll.org-20111203133409-hm63r8s3zz6zsh8h
Bumped version number

Show diffs side-by-side

added added

removed removed

Lines of Context:
tests/test_db.py
184
184
        db.logout()
185
185
        self.assertFalse('Authorization' in db.headers)
186
186
 
 
187
    def testLoginLogoutOAuth2(self):
 
188
        """
 
189
        Make sure login and logout functions set things up correctly
 
190
        when we use OAuth2.
 
191
        """
 
192
        token = 'kajfjowijmssafuwoisflsjlfsoieuossfh'
 
193
        db = FluidDB(TEST_INSTANCE)
 
194
        # start from a blank slate
 
195
        self.assertFalse('Authorization' in db.headers)
 
196
        self.assertFalse('X-FluidDB-Access-Token' in db.headers)
 
197
        # Login
 
198
        db.login_oauth2(token)
 
199
        self.assertEquals(db.headers['Authorization'], 'oauth2')
 
200
        self.assertEquals(db.headers['X-FluidDB-Access-Token'], token)
 
201
        # Logout
 
202
        db.logout()
 
203
        self.assertFalse('Authorization' in db.headers)
 
204
        self.assertFalse('X-FluidDB-Access-Token' in db.headers)
 
205
 
187
206
 
188
207
if __name__ == '__main__':
189
208
    unittest.main()