示例#1
0
 def setUp(self):
    self.mock_request = self.create_patch(
       'yahoo_fantasy.api.OAuth1Service.get_request_token',
       autospec=True,
       side_effect=mock_request_token
    )
    self.mock_auth = self.create_patch(
       'yahoo_fantasy.api.OAuth1Service.get_auth_session',
       autospec=True,
       side_effect=mock_auth_session
    )
    self.api = API('my_key', 'my_secret')
示例#2
0
class APITest(unittest.TestCase):
   def create_patch(self, name, autospec=None, side_effect=None):
      patcher = patch(
         name,
         autospec=autospec,
         side_effect=side_effect
      )
      mocked = patcher.start()
      self.addCleanup(patcher.stop)
      return mocked
      
   def setUp(self):
      self.mock_request = self.create_patch(
         'yahoo_fantasy.api.OAuth1Service.get_request_token',
         autospec=True,
         side_effect=mock_request_token
      )
      self.mock_auth = self.create_patch(
         'yahoo_fantasy.api.OAuth1Service.get_auth_session',
         autospec=True,
         side_effect=mock_auth_session
      )
      self.api = API('my_key', 'my_secret')
   
   def test_creates_oauth_service(self):
      self.assertIsInstance(self.api.service, OAuth1Service)
      self.assertEqual(self.api.service.consumer_key, 'my_key')
      self.assertEqual(self.api.service.consumer_secret, 'my_secret')

   def test_refuses_bad_consumer_key(self):
      with self.assertRaises(AuthenticationError):
         self.api.service.consumer_key = 'bad_key'
         self.api.get_request_token()

   def test_refuses_bad_consumer_secret(self):
      with self.assertRaises(AuthenticationError):
         self.api.service.consumer_secret = 'bad_secret'
         self.api.get_request_token()

   def test_includes_callback_method(self):
      self.api.get_request_token()
      args, kwargs = self.mock_request.call_args
      self.assertEqual(kwargs['params'], {'oauth_callback': 'oob'})

   def test_returns_request_token(self):
      self.api.get_request_token()
      self.assertEqual(self.api.request_token, 'request_token')
      self.assertEqual(self.api.request_secret, 'request_secret')

   def test_generates_authorize_url(self):
      url = self.api.get_authorize_url()
      self.assertEqual(
         url,
         'https://api.login.yahoo.com/oauth/v2/request_auth?oauth_token=request_token'
      )

   def test_creates_session_with_pin(self):
      self.api.get_request_token()
      self.api.create_session('pin')
      self.mock_auth.assert_called_with(
         self.api.service,
         'request_token',
         'request_secret',
         method='POST',
         data={'oauth_verifier': 'pin'}
         )
      self.assertEqual(self.api.pin, 'pin')

   def test_refuses_bad_pin(self):
      self.api.get_request_token()
      with self.assertRaises(AuthenticationError):
         self.api.create_session('bad_pin')