def setUp(self):
self.gh = api.Github()
self.handler = Handler(self.gh)
def setUp(self):
self.gh = api.Github()
self.handler = Handler(self.gh)
class TestHandler(TestCase):
def setUp(self):
self.gh = api.Github()
self.handler = Handler(self.gh)
def test_get_converter(self):
self.assertIsInstance(self.handler._get_converter(), Modelizer)
kwargs = {"converter": Rawlizer}
self.assertIsInstance(self.handler._get_converter(kwargs), Rawlizer)
self.assertEquals(kwargs, {})
self.handler.converter = Modelizer
self.assertIsInstance(self.handler._get_converter(), Modelizer)
def test_bool(self):
with patch.object(api.Github, "head") as head:
response = head.return_value
response.status_code = 204
bool1 = self.handler._bool("test")
head.side_effect = NotFound()
bool2 = self.handler._bool("test")
head.assert_called_with("test")
self.assertTrue(bool1)
self.assertFalse(bool2)
with patch.object(api.Github, "put") as put:
response = put.return_value
response.status_code = 204
booll = self.handler._put("test")
put.assert_called_with("test", method="put")
self.assertTrue(booll)
with patch.object(api.Github, "delete") as delete:
response = delete.return_value
response.content = self.gh._parser.dumps({"data": "test"})
response.status_code = 204
bool1 = self.handler._bool("test", method="delete")
bool2 = self.handler._bool("test", method="delete", data={"some": "data"})
self.assertTrue(bool1)
self.assertTrue(bool2)
@patch.object(api.Github, "_request")
def test_get_resources(self, request):
# Simulating per_page=2 with STUB (it returns two resources)
response = request.return_value
response.status_code = 200
response.headers = {"link": GET_LINK}
response.content = self.gh._parser.dumps(GET_RESOURCES)
resources = self.handler._get_resources("users", model=User)
self.assertFalse(request.called)
resources = list(resources)
self.assertTrue(request.call_count, 5)
request_args = ("GET", "users")
self.assertEquals(
request.call_args_list,
[
(request_args, {"page": 1}),
(request_args, {"page": 2}),
(request_args, {"page": 3}),
(request_args, {"page": 4}),
(request_args, {"page": 5}),
],
)
self.assertEquals(len(resources), 10)
self.assertEquals(resources[0].login, "octocat")
request.reset_mock()
resources = self.handler._get_resources("users", model=User, limit=5)
resources = list(resources)
self.assertEquals(request.call_count, 3)
self.assertEquals(len(resources), 5)
request.reset_mock()
resources = self.handler._get_resources("users", model=User, limit=4)
resources = list(resources)
self.assertEquals(request.call_count, 2)
self.assertEquals(len(resources), 4)
request.reset_mock()
resources = self.handler._get_resources("users", model=User, limit=-5)
resources = list(resources)
self.assertEquals(request.call_count, 3)
self.assertEquals(len(resources), 5)
@patch.object(api.Github, "get")
def test_get_resource(self, get):
# Converter test + api(get) test. Half trivial
get.return_value = {"login": "******"}
model = self.handler._get_resource("test", model=User)
self.assertEquals(model.login, "octocat")
@patch.object(api.Github, "post")
def test_post_resource(self, post):
post.return_value = {"data": "posted"}
data = {"data": "to_post"}
user_new = self.handler._post_resource("test", data=data, model=User)
post.assert_called_with("test", data=data)
class TestHandler(TestCase):
def setUp(self):
self.gh = api.Github()
self.handler = Handler(self.gh)
def test_get_converter(self):
self.assertIsInstance(self.handler._get_converter(), Modelizer)
kwargs = {'converter': Rawlizer}
self.assertIsInstance(self.handler._get_converter(kwargs), Rawlizer)
self.assertEquals(kwargs, {})
self.handler.converter = Modelizer
self.assertIsInstance(self.handler._get_converter(), Modelizer)
def test_bool(self):
with patch.object(api.Github, 'head') as head:
response = head.return_value
response.status_code = 204
bool1 = self.handler._bool('test')
head.side_effect = NotFound()
bool2 = self.handler._bool('test')
head.assert_called_with('test')
self.assertTrue(bool1)
self.assertFalse(bool2)
with patch.object(api.Github, 'put') as put:
response = put.return_value
response.status_code = 204
booll = self.handler._put('test')
put.assert_called_with('test', method='put')
self.assertTrue(booll)
with patch.object(api.Github, 'delete') as delete:
response = delete.return_value
response.content = self.gh._parser.dumps({'data': 'test'})
response.status_code = 204
bool1 = self.handler._bool('test', method='delete')
bool2 = self.handler._bool('test',
method='delete',
data={'some': 'data'})
self.assertTrue(bool1)
self.assertTrue(bool2)
@patch.object(api.Github, '_request')
def test_get_resources(self, request):
# Simulating per_page=2 with STUB (it returns two resources)
response = request.return_value
response.status_code = 200
response.headers = {'link': GET_LINK}
response.content = self.gh._parser.dumps(GET_RESOURCES)
resources = self.handler._get_resources('users', model=User)
self.assertFalse(request.called)
resources = list(resources)
self.assertTrue(request.call_count, 5)
request_args = ('GET', 'users')
self.assertEquals(request.call_args_list, [(request_args, {
'page': 1
}), (request_args, {
'page': 2
}), (request_args, {
'page': 3
}), (request_args, {
'page': 4
}), (request_args, {
'page': 5
})])
self.assertEquals(len(resources), 10)
self.assertEquals(resources[0].login, 'octocat')
request.reset_mock()
resources = self.handler._get_resources('users', model=User, limit=5)
resources = list(resources)
self.assertEquals(request.call_count, 3)
self.assertEquals(len(resources), 5)
request.reset_mock()
resources = self.handler._get_resources('users', model=User, limit=4)
resources = list(resources)
self.assertEquals(request.call_count, 2)
self.assertEquals(len(resources), 4)
request.reset_mock()
resources = self.handler._get_resources('users', model=User, limit=-5)
resources = list(resources)
self.assertEquals(request.call_count, 3)
self.assertEquals(len(resources), 5)
@patch.object(api.Github, 'get')
def test_get_resource(self, get):
# Converter test + api(get) test. Half trivial
get.return_value = {'login': '******'}
model = self.handler._get_resource('test', model=User)
self.assertEquals(model.login, 'octocat')
@patch.object(api.Github, 'post')
def test_post_resource(self, post):
post.return_value = {'data': 'posted'}
data = {'data': 'to_post'}
user_new = self.handler._post_resource('test', data=data, model=User)
post.assert_called_with('test', data=data)
