def test_close_fail_general(self, mock_create_engine, mock_update,
mock_alchemy_funks):
with mock.patch.dict(update_contributor.os.environ,
{"Database_Location": "Djibouti"}):
with mock.patch("update_contributor.Session") as mock_sesh:
mock_session = AlchemyMagicMock()
mock_sesh.return_value = mock_session
mock_session.close.side_effect = Exception("Bad Me")
mock_update.return_value.values.return_value\
.returning.return_value\
.on_conflict_do_nothing.return_value = "bob"
x = update_contributor.lambda_handler(
{
"additional_comments": "6", # "Hello",
"contributor_comments":
"666", # "Contributor says hi!",
"survey_period": "201712", # "201712",
"survey_code": "066", # "066",
"ru_reference": "77700000001"
},
"")
assert (x["statusCode"] == 500)
assert ("Database Session Closed Badly." in x['body']['Error'])
assert ("General Error" in str(x['body']['Error']))
def test_input_data_exception(self):
with open('tests/fixtures/test_data.txt') as infile:
test_data = json.load(infile)
with mock.patch.dict(update_contributor.os.environ,
{"Database_Location": "Djibouti"}):
test_data.pop('query_type')
x = update_contributor.lambda_handler("MIKE", '')
assert (x["statusCode"] == 400)
assert ("Invalid" in str(x['body']))
def test_db_connection_exception_general(self, mock_create_engine):
with mock.patch.dict(update_contributor.os.environ,
{"Database_Location": "Djibouti"}):
mock_create_engine.side_effect = Exception("Bad Me")
x = update_contributor.lambda_handler(
{
"additional_comments": "6", # "Hello",
"contributor_comments":
"666", # "Contributor says hello!",
"survey_period": "201712", # "201712",
"survey_code": "066", # "066",
"ru_reference": "77700000001"
},
"")
assert (x["statusCode"] == 500)
assert ("Failed To Connect" in str(x['body']['Error']))
assert ("General Error" in str(x['body']['Error']))
def test_db_connection_exception_driver(self, mock_create_engine):
with mock.patch.dict(update_contributor.os.environ,
{"Database_Location": "MyPostgresDatase"}):
mock_create_engine.side_effect =\
db.exc.NoSuchModuleError('', '', '')
x = update_contributor.lambda_handler(
{
"additional_comments": "6", # "Hello",
"contributor_comments":
"666", # "Contributor says hello!",
"survey_period": "201712", # "201712",
"survey_code": "066", # "066",
"ru_reference": "77700000001"
},
'')
assert (x["statusCode"] == 500)
assert ("Failed To Connect" in x['body']['Error'])
assert ("Driver Error" in str(x['body']['Error']))
def test_lambda_handler_happy_path(self, mock_create_engine, mock_update,
mock_marsh, mock_alchemy_funks):
with mock.patch.dict(update_contributor.os.environ,
{"Database_Location": "Djibouti"}):
mock_update.return_value.values.return_value\
.returning.return_value\
.on_conflict_do_nothing.return_value = "bob"
x = update_contributor.lambda_handler(
{
"additional_comments": "6", # "Hello",
"contributor_comments":
"666", # "Contributor says hello!",
"survey_period": "201712", # "201712",
"survey_code": "066", # "066",
"ru_reference": "77700000001"
},
"")
mock_marsh.return_value = True
assert (x["statusCode"] == 200)
assert ("Successfully" in x['body']['Success'])
def test_update_contributor_fail_general(self, mock_create_engine,
mock_alchemy_funcs):
with mock.patch.dict(update_contributor.os.environ,
{"Database_Location": "Djibouti"}):
with mock.patch("update_contributor.Session.query") as mock_update:
mock_update.side_effect = Exception("Bad Me")
x = update_contributor.lambda_handler(
{
"additional_comments": "6", # "Hello",
"contributor_comments":
"666", # "Contributor says hi!",
"survey_period": "201712", # "201712",
"survey_code": "066", # "066",
"ru_reference": "77700000001"
},
"")
assert (x["statusCode"] == 500)
assert ("Failed To Update Data: contributor_survey_period"
in x['body']['Error'])
assert ("General Error" in str(x['body']['Error']))
def test_environment_variable_exception(self):
x = update_contributor.lambda_handler(
"JAMIE - FORMER SOFTWARE ENGINEER", '')
assert (x["statusCode"] == 500)
assert ("Configuration Error:" in x['body']['Error'])