class TestLite(unittest.TestCase):
def setUp(self):
self.api = Lite(consumer_key="foo", consumer_secret="bar", version="lite")
@mock.patch("contextio.lib.api.Api._request_uri")
def test_get_users_returns_a_list_of_User_resources(self, mock_request):
mock_request.return_value = {"id": "some_id"}
accounts = self.api.get_users()
self.assertEqual(1, len(accounts))
self.assertIsInstance(accounts[0], User)
@mock.patch("contextio.lib.api.Api._request_uri")
def test_post_user_returns_User(self, mock_request):
mock_request.return_value = {"id": "some_id"}
user = self.api.post_user(
password="******", email="*****@*****.**", server="fake.server", username="******",
use_ssl=1, port=993, type="IMAP")
self.assertIsInstance(user, User)
def test_post_user_requires_arguments(self):
with self.assertRaises(errors.ArgumentError):
self.api.post_connect_token(foo="bar")
@mock.patch("contextio.lib.api.Api._request_uri")
def test_post_connect_token_returns_dict(self, mock_request):
mock_request.return_value = {"token": "fake_token"}
connect_token_request = self.api.post_connect_token(callback_url="fake.url")
self.assertEqual(connect_token_request, {"token": "fake_token"})
def setUp(self):
self.api = Lite(consumer_key="foo", consumer_secret="bar", version="lite")
def ContextIO(consumer_key, consumer_secret, **kwargs):
if kwargs.get("api_version") == "lite":
return Lite(consumer_key, consumer_secret, **kwargs)
else:
return V2_0(consumer_key, consumer_secret, **kwargs)
