P

lib/yql/tests/test_yahoo_token.py

@@ -0,0 +1,81 @@
+from unittest import TestCase
+
+from nose.tools import raises
+try:
+    from urlparse import parse_qs, parse_qsl
+except ImportError:
+    from cgi import parse_qs, parse_qsl
+
+import yql
+
+
+class YahooTokenTest(TestCase):
+    def test_create_yahoo_token(self):
+        token = yql.YahooToken('test-key', 'test-secret')
+        self.assertEqual(token.key, 'test-key')
+        self.assertEqual(token.secret, 'test-secret')
+
+    def test_y_token_to_string(self):
+        token = yql.YahooToken('test-key', 'test-secret')
+        token_to_string = token.to_string()
+        string_data = dict(parse_qsl(token_to_string))
+        self.assertEqual(string_data.get('oauth_token'), 'test-key')
+        self.assertEqual(string_data.get('oauth_token_secret'), 'test-secret')
+
+    def test_y_token_to_string2(self):
+        token = yql.YahooToken('test-key', 'test-secret')
+
+        token.timestamp = '1111'
+        token.session_handle = 'poop'
+        token.callback_confirmed = 'basilfawlty'
+
+        token_to_string = token.to_string()
+        string_data = dict(parse_qsl(token_to_string))
+        self.assertEqual(string_data.get('oauth_token'), 'test-key')
+        self.assertEqual(string_data.get('oauth_token_secret'), 'test-secret')
+        self.assertEqual(string_data.get('token_creation_timestamp'), '1111')
+        self.assertEqual(string_data.get('oauth_callback_confirmed'), 'basilfawlty')
+        self.assertEqual(string_data.get('oauth_session_handle'), 'poop')
+
+    def test_y_token_from_string(self):
+        token_string = "oauth_token=foo&oauth_token_secret=bar&"\
+                       "oauth_session_handle=baz&token_creation_timestamp=1111"
+        token_from_string = yql.YahooToken.from_string(token_string)
+        self.assertEqual(token_from_string.key, 'foo')
+        self.assertEqual(token_from_string.secret, 'bar')
+        self.assertEqual(token_from_string.session_handle, 'baz')
+        self.assertEqual(token_from_string.timestamp, '1111')
+
+    @raises(ValueError)
+    def test_y_token_raises_value_error(self):
+        yql.YahooToken.from_string('')
+
+    @raises(ValueError)
+    def test_y_token_raises_value_error2(self):
+        yql.YahooToken.from_string('foo')
+
+    @raises(ValueError)
+    def test_y_token_raises_value_error3(self):
+        yql.YahooToken.from_string('oauth_token=bar')
+
+    @raises(ValueError)
+    def test_y_token_raises_value_error4(self):
+        yql.YahooToken.from_string('oauth_token_secret=bar')
+
+    @raises(AttributeError)
+    def test_y_token_without_timestamp_raises(self):
+        token = yql.YahooToken('test', 'test2')
+        y = yql.ThreeLegged('test', 'test2')
+        y.check_token(token)
+
+    def test_y_token_without_timestamp_raises2(self):
+
+        def refresh_token_replacement(token):
+            return 'replaced'
+
+        y = yql.ThreeLegged('test', 'test2')
+        y.refresh_token = refresh_token_replacement
+
+        token = yql.YahooToken('test', 'test2')
+        token.timestamp = 11111
+        self.assertEqual(y.check_token(token), 'replaced')