def test_are_creds_valid_with_invalid_creds(self):
with patch.object(github.Github, 'get_user') as patched_get_user:
mocked_login = PropertyMock()
mocked_login.side_effect = github.BadCredentialsException(401, 'dummy data')
type(patched_get_user.return_value).login = mocked_login
self.assertFalse(self.backend.are_creds_valid())
def test_are_creds_valid_with_invalid_creds(self):
with patch.object(github.Github, 'get_user') as patched_get_user:
mocked_login = PropertyMock()
mocked_login.side_effect = github.BadCredentialsException(
401, 'dummy data')
type(patched_get_user.return_value).login = mocked_login
self.assertFalse(self.backend.are_creds_valid())
def ReturnWorks(illust_id):
r_works = MagicMock()
s = {}
if 0 < illust_id and illust_id < 99999999:
type(r_works).status = "success"
s = self.__GetIllustData(illust_id)
else:
type(r_works).status = "failed"
def ReturnResponse():
r_response = MagicMock()
# イラストデータ設定
type(r_response).type = s["type"]
type(r_response).is_manga = s["is_manga"]
type(r_response).title = s["title"]
r_name_id = MagicMock()
type(r_name_id).name = s["author_name"]
type(r_name_id).id = s["author_id"]
type(r_response).user = r_name_id
def ReturnLarge(url):
r_large = MagicMock()
type(r_large).large = url
return r_large
def ReturnImageurls(url):
r_imageurls = MagicMock()
type(r_imageurls).image_urls = ReturnLarge(url)
return r_imageurls
def ReturnPages():
r_pages = MagicMock()
# 漫画形式のreturn_value設定
type(r_pages).pages = [
ReturnImageurls(url) for url in s["image_urls"]
]
return r_pages
type(r_response).metadata = ReturnPages()
# 一枚絵のreturn_value設定
type(r_response).image_urls = ReturnLarge(s["image_url"])
return r_response
p_response = PropertyMock()
p_response.side_effect = lambda: [ReturnResponse()]
type(r_works).response = p_response
return r_works
def test_get_will_retry():
global sequential_values, retry_policy
retry_policy = MagicMock()
retry_policy.should_retry.return_value = True
retry_policy.seconds_to_sleep.return_value = 0.01
with patch('requests.get') as patched_get:
# When we call assert_equals in this unit test, the side_effect is executed.
# So, the last status_code should be repeated.
sequential_values = [500, 200, 200]
pm = PropertyMock()
pm.side_effect = return_sequential
type(patched_get.return_value).status_code = pm
client = ReliableHttpClient(endpoint, {}, retry_policy)
result = client.get("r", [200])
assert_equals(200, result.status_code)
retry_policy.should_retry.assert_called_once_with(500, False, 0)
retry_policy.seconds_to_sleep.assert_called_once_with(0)
def test_get_will_retry():
global sequential_values, retry_policy
retry_policy = MagicMock()
retry_policy.should_retry.return_value = True
retry_policy.seconds_to_sleep.return_value = 0.01
with patch('requests.get') as patched_get:
# When we call assert_equals in this unit test, the side_effect is executed.
# So, the last status_code should be repeated.
sequential_values = [500, 200, 200]
pm = PropertyMock()
pm.side_effect = return_sequential
type(patched_get.return_value).status_code = pm
client = ReliableHttpClient(endpoint, {}, retry_policy)
result = client.get("r", [200])
assert_equals(200, result.status_code)
retry_policy.should_retry.assert_called_once_with(500, False, 0)
retry_policy.seconds_to_sleep.assert_called_once_with(0)
def test_delete_will_retry():
global sequential_values, retry_policy
retry_policy = MagicMock()
retry_policy.should_retry.return_value = True
retry_policy.seconds_to_sleep.return_value = 0.01
with patch('requests.delete') as patched_delete:
# When we call assert_equals in this unit test, the side_effect is executed.
# So, the last status_code should be repeated.
sequential_values = [500, 200, 200]
pm = PropertyMock()
pm.side_effect = return_sequential
type(patched_delete.return_value).status_code = pm
client = ReliableHttpClient("http://url.com", {}, "username", "password", retry_policy)
result = client.delete("r", [200])
assert_equals(200, result.status_code)
retry_policy.should_retry.assert_called_once_with(500, 0)
retry_policy.seconds_to_sleep.assert_called_once_with(0)
def test_deserialize_not_empty_but_error():
session = MagicMock()
status_property = PropertyMock()
status_property.side_effect = ValueError()
type(session).status = status_property
reader_writer = MagicMock()
reader_writer.read_lines.return_value = """{
"clients": [
{
"name": "py",
"id": "1",
"sqlcontext": true,
"kind": "pyspark",
"connectionstring": "url=https://mysite.com/livy;username=user;password=pass",
"version": "0.0.0"
},
{
"name": "sc",
"id": "2",
"sqlcontext": false,
"kind": "spark",
"connectionstring": "url=https://mysite.com/livy;username=user;password=pass",
"version": "0.0.0"
}
]
}
"""
serializer = ClientManagerStateSerializer(reader_writer)
serializer._create_livy_session = MagicMock(return_value=session)
serializer._create_livy_client = MagicMock()
deserialized = serializer.deserialize_state()
assert len(deserialized) == 0
serializer._create_livy_session.assert_no_called()
serializer._create_livy_session.assert_no_called()
def test_deserialize_not_emtpy_but_error():
client_factory = MagicMock()
session = MagicMock()
status_property = PropertyMock()
status_property.side_effect = ValueError()
type(session).status = status_property
client_factory.create_session.return_value = session
reader_writer = MagicMock()
reader_writer.read_lines.return_value = """{
"clients": [
{
"name": "py",
"id": "1",
"sqlcontext": true,
"language": "python",
"connectionstring": "url=https://mysite.com/livy;username=user;password=pass",
"version": "0.0.0"
},
{
"name": "sc",
"id": "2",
"sqlcontext": false,
"language": "scala",
"connectionstring": "url=https://mysite.com/livy;username=user;password=pass",
"version": "0.0.0"
}
]
}
"""
serializer = ClientManagerStateSerializer(client_factory, reader_writer)
deserialized = serializer.deserialize_state()
assert len(deserialized) == 0
client_factory.create_session.assert_no_called()
client_factory.build_client.assert_no_called()
def test_deserialize_not_empty_but_error():
session = MagicMock()
status_property = PropertyMock()
status_property.side_effect = ValueError()
type(session).status = status_property
reader_writer = MagicMock()
reader_writer.read_lines.return_value = """{
"clients": [
{
"name": "py",
"id": "1",
"sqlcontext": true,
"kind": "pyspark",
"connectionstring": "url=https://mysite.com/livy;username=user;password=pass",
"version": "0.0.0"
},
{
"name": "sc",
"id": "2",
"sqlcontext": false,
"kind": "spark",
"connectionstring": "url=https://mysite.com/livy;username=user;password=pass",
"version": "0.0.0"
}
]
}
"""
serializer = ClientManagerStateSerializer(reader_writer)
serializer._create_livy_session = MagicMock(return_value=session)
serializer._create_livy_client = MagicMock()
deserialized = serializer.deserialize_state()
assert len(deserialized) == 0
serializer._create_livy_session.assert_no_called()
serializer._create_livy_session.assert_no_called()
