class OAuthTest(TestCase): def setUp(self): self.oauth = OAuth(CONSUMER_KEY, CONSUMER_SECRET) self.oauth.token = oauth2.Token(OAUTH_TOKEN, OAUTH_TOKEN_SECRET) self.tmp = NamedTemporaryFile() def tearDown(self): self.tmp.close() def test_save_token(self): self.oauth.save_token(self.tmp.name) with open(self.tmp.name, 'rb') as infile: db = pickle.load(infile) self.assertEqual( db['oauth_token'], OAUTH_TOKEN, msg="Saved token ('%s') doesn't match expected value ('%s')" % (db['oauth_token'], OAUTH_TOKEN)) self.assertEqual( db['oauth_token_secret'], OAUTH_TOKEN_SECRET, msg="Saved token secret ('%s') doesn't match expected value ('%s')" % (db['oauth_token_secret'], OAUTH_TOKEN_SECRET)) def test_load_token(self): test_db = { 'oauth_token': OAUTH_TOKEN, 'oauth_token_secret': OAUTH_TOKEN_SECRET } with open(self.tmp.name, 'wb') as outfile: pickle.dump(test_db, outfile, pickle.HIGHEST_PROTOCOL) self.oauth.load_token(self.tmp.name) self.assertEqual( self.oauth.client.token.key, test_db['oauth_token'], msg="Loaded oauth token ('%s') doesn't match expected value ('%s')" % (self.oauth.client.token.key, test_db['oauth_token'])) self.assertEqual( self.oauth.client.token.secret, test_db['oauth_token_secret'], msg= "Loaded oauth token secret ('%s') doesn't match expected value ('%s')" % (self.oauth.client.token.secret, test_db['oauth_token_secret'])) def test_generate_request_params(self): test_params = { 'param1': 'firstParam', 'param2': 'second /Pàram\\', } expected_re = re.compile(r'oauth_nonce=\d{8}&oauth_timestamp=\d{10}&oauth_consumer_key='+\ CONSUMER_KEY+r'&oauth_version=1\.0&oauth_token='+\ OAUTH_TOKEN+r'¶m2=second\+%2FP%C3%A0ram%5C¶m1=firstParam') result = self.oauth.generate_request_params(test_params) try: self.assertRegexpMatches(result, expected_re) except AttributeError: assert expected_re.match( result) is not None, "Result doesn't match reference regex."
def setUp(self): self.oauth = OAuth(CONSUMER_KEY, CONSUMER_SECRET) self.oauth.token = oauth2.Token(OAUTH_TOKEN, OAUTH_TOKEN_SECRET) self.tmp = NamedTemporaryFile()
class OAuthTest(TestCase): def setUp(self): self.oauth = OAuth(CONSUMER_KEY, CONSUMER_SECRET) self.oauth.token = oauth2.Token(OAUTH_TOKEN, OAUTH_TOKEN_SECRET) self.tmp = NamedTemporaryFile() def tearDown(self): self.tmp.close() def test_save_token(self): self.oauth.save_token(self.tmp.name) with open(self.tmp.name, 'rb') as infile: db = pickle.load(infile) self.assertEqual( db['oauth_token'], OAUTH_TOKEN, msg="Saved token ('%s') doesn't match expected value ('%s')" % ( db['oauth_token'], OAUTH_TOKEN ) ) self.assertEqual( db['oauth_token_secret'], OAUTH_TOKEN_SECRET, msg="Saved token secret ('%s') doesn't match expected value ('%s')" % ( db['oauth_token_secret'], OAUTH_TOKEN_SECRET ) ) def test_load_token(self): test_db = { 'oauth_token': OAUTH_TOKEN, 'oauth_token_secret': OAUTH_TOKEN_SECRET } with open(self.tmp.name, 'wb') as outfile: pickle.dump(test_db, outfile, pickle.HIGHEST_PROTOCOL) self.oauth.load_token(self.tmp.name) self.assertEqual( self.oauth.client.token.key, test_db['oauth_token'], msg="Loaded oauth token ('%s') doesn't match expected value ('%s')" % ( self.oauth.client.token.key, test_db['oauth_token'] ) ) self.assertEqual( self.oauth.client.token.secret, test_db['oauth_token_secret'], msg="Loaded oauth token secret ('%s') doesn't match expected value ('%s')" % ( self.oauth.client.token.secret, test_db['oauth_token_secret'] ) ) def test_generate_request_params(self): test_params = { 'param1': 'firstParam', 'param2': 'second /Pàram\\', } expected_re = re.compile(r'oauth_nonce=\d{8}&oauth_timestamp=\d{10}&oauth_consumer_key='+\ CONSUMER_KEY+r'&oauth_version=1\.0&oauth_token='+\ OAUTH_TOKEN+r'¶m2=second\+%2FP%C3%A0ram%5C¶m1=firstParam') result = self.oauth.generate_request_params(test_params) try: self.assertRegexpMatches(result, expected_re) except AttributeError: assert expected_re.match(result) is not None, "Result doesn't match reference regex."