def test_get_sleep(self):
client = self._default_client
timestamp = 1471761649
endpoint = BedditClient.build_full_path('/api/v1/user/10000/sleep')
response = Mock()
response.status_code = 200
sleep_object = json.load(
open(os.path.join(BASE_DIR, 'fixtures/sleep.json')))
response.json = lambda: sleep_object
with patch('requests.get', return_value=response) as get:
sleeps = client.get_sleeps(
start=datetime.datetime.utcfromtimestamp(timestamp),
end=datetime.datetime.utcfromtimestamp(timestamp),
limit=10,
reverse=True)
self.assertEqual(len(sleeps), 1)
self.assertEqual(type(sleeps[0]), Sleep)
args, kwargs = get.call_args
self.assertEqual(args[0], endpoint)
self.assertDictEqual(
kwargs['params'], {
'start_date': '2016-08-21',
'end_date': '2016-08-21',
'limit': 10,
'reverse': 'yes'
})
def test_authentication_error(self):
response = Mock()
response.status_code = 400
response.json.side_effect = lambda: {'description': 'auth_error'}
with patch('requests.post', return_value=response):
self.assertRaises(BedditClient.AuthError, BedditClient, 'username',
'password')
def test_reset_password(self):
response = Mock()
response.status_code = 200
with patch('requests.post', return_value=response) as post:
result = BedditClient.reset_password('*****@*****.**')
self.assertTrue(result)
post.assert_called_once_with(
BedditClient.build_full_path('/api/v1/user/password_reset'),
data={'email': '*****@*****.**'})
def test_get_group(self):
client = self._default_client
endpoint = BedditClient.build_full_path('/api/v1/user/10000/group')
response = Mock()
response.status_code = 200
group_object = json.load(
open(os.path.join(BASE_DIR, 'fixtures/group.json')))
response.json = lambda: [group_object]
with patch('requests.get', return_value=response) as get:
groups = client.get_groups()
self.assertEqual(type(groups[0]), Group)
args = get.call_args[0]
self.assertEqual(args[0], endpoint)
def test_update_user(self):
client = self._default_client
endpoint = BedditClient.build_full_path('/api/v1/user/10000')
response = Mock()
response.status_code = 200
user_object = json.load(
open(os.path.join(BASE_DIR, 'fixtures/user.json')))
response.json = lambda: user_object
with patch('requests.put', return_value=response) as put:
user = client.update_user(name='foo')
self.assertEqual(type(user), User)
args = put.call_args[0]
self.assertEqual(args[0], endpoint)
def test_remove_group_invite(self):
client = self._default_client
endpoint = BedditClient.build_full_path(
'/api/v1/group/200/member/10000/remove')
response = Mock()
response.status_code = 200
group_object = json.load(
open(os.path.join(BASE_DIR, 'fixtures/group.json')))
response.json = lambda: [group_object]
with patch('requests.post', return_value=response) as post:
groups = client.remove_group_invite(group_id=200, user_id=10000)
self.assertEqual(type(groups[0]), Group)
args = post.call_args[0]
self.assertEqual(args[0], endpoint)
def test_invite_to_group(self):
client = self._default_client
endpoint = BedditClient.build_full_path('/api/v1/group/new/invite')
response = Mock()
response.status_code = 200
group_object = json.load(
open(os.path.join(BASE_DIR, 'fixtures/group.json')))
response.json = lambda: [group_object]
with patch('requests.post', return_value=response) as post:
groups = client.invite_to_group(email="*****@*****.**")
self.assertEqual(type(groups[0]), Group)
args, kwargs = post.call_args
self.assertEqual(args[0], endpoint)
self.assertDictEqual(kwargs['data'], {'email': '*****@*****.**'})
def _inner(self, *args, **kwargs):
response = Mock()
response.json.side_effect = lambda: {
'user': 10000,
'access_token': 'dummytoken'
}
response.status_code = 200
payload = {
'grant_type': 'password',
'username': '******',
'password': '******'
}
with patch('requests.post', return_value=response) as post:
retval = testcase(self, *args, **kwargs)
post.assert_called_with(
BedditClient.build_full_path('api/v1/auth/authorize'),
data=payload)
return retval
def test_get_session(self):
client = self._default_client
timestamp = 1471761649
endpoint = BedditClient.build_full_path('/api/v1/user/10000/session')
response = Mock()
response.status_code = 200
session_object = json.load(
open(os.path.join(BASE_DIR, 'fixtures/session.json')))
response.json = lambda: session_object
dt = datetime.datetime.fromtimestamp(timestamp)
with patch('requests.get', return_value=response) as get:
sessions = client.get_sessions(updated_after=dt)
self.assertEqual(len(sessions), 1)
self.assertEqual(type(sessions[0]), Session)
args, kwargs = get.call_args
self.assertEqual(args[0], endpoint)
self.assertDictEqual(kwargs['params'],
{'updated_after': timestamp})