def test_unsupported_method(mocked_anime_db, mocked_anime):
event = {
"requestContext": {
"http": {
"method": "AA"
}
},
"queryStringParameters": {
"mal_id": "123"
}
}
with pytest.raises(mocked_anime.UnsupportedMethod):
handle(event, None)
def test_not_found(self, mocked_anime_db):
mocked_anime_db.table.query.side_effect = mocked_anime_db.NotFoundError
res = handle(self.event, None)
exp = {
"statusCode": 404,
}
assert res == exp
def test_success(self, mocked_anime_db, mocked_anime):
mocked_anime_db.table.query.side_effect = mocked_anime_db.NotFoundError
mocked_anime.sqs_queue.send_message.return_value = True
res = handle(self.event, None)
exp = {
"body": json.dumps({"id": "7e2c8ee2-66cf-598f-aedf-cdfba825613b"}),
"statusCode": 202
}
assert res == exp
def test_invalid_api_name(self):
event = copy.deepcopy(self.event)
event["queryStringParameters"]["api_name"] = "INVALID"
res = handle(event, None)
exp = {
"statusCode": 400,
"body": json.dumps({"error": "Unsupported api_name"})
}
assert res == exp
def test_missing_api_name(self):
event = copy.deepcopy(self.event)
del event["queryStringParameters"]["api_name"]
res = handle(event, None)
exp = {
"statusCode": 400,
"body": json.dumps({"error": "Missing api_name query parameter"})
}
assert res == exp
def test_no_query_params(self):
event = copy.deepcopy(self.event)
event["queryStringParameters"] = {}
res = handle(event, None)
exp = {
"statusCode": 400,
"body": json.dumps({"error": "Please specify query parameters"})
}
assert res == exp
def test_invalid_query_params(self):
event = copy.deepcopy(self.event)
event["queryStringParameters"] = {
"abc": "123"
}
res = handle(event, None)
exp = {
"statusCode": 400,
"body": json.dumps({"error": "Missing api_id query parameter"})
}
assert res == exp
def test_already_exist(self, mocked_anime_db):
mocked_anime_db.table.query.return_value = {
"Items": [
{
"mal_id": "123"
}
]
}
res = handle(self.event, None)
exp = {
"body": json.dumps({"id": "7e2c8ee2-66cf-598f-aedf-cdfba825613b"}),
"statusCode": 202
}
assert res == exp
def test_success(self, mocked_anime_db):
exp_res = {
"id": "123"
}
mocked_anime_db.table.query.return_value = {
"Items": [
exp_res
]
}
res = handle(self.event, None)
exp = {
"statusCode": 200,
"body": json.dumps(exp_res)
}
assert res == exp
def test_empty_body(self, mocked_anime_db):
mocked_anime_db.table.query.return_value = {
"Items": [
{
"mal_id": "123"
}
]
}
event = copy.deepcopy(self.event)
event["body"] = {}
res = handle(event, None)
exp = {
"statusCode": 400,
"body": "Invalid post body"
}
assert res == exp
def test_invalid_api_name(self, mocked_anime_db):
mocked_anime_db.table.query.return_value = {
"Items": [
{
"mal_id": 123
}
]
}
event = copy.deepcopy(self.event)
event["body"] = '{ "api_name": "bad_name", "api_id": "123"}'
res = handle(event, None)
exp = {
"statusCode": 400,
"body": json.dumps({
"message": "Invalid post schema",
"error": "\'bad_name\' is not one of [\'mal\']",
})
}
assert res == exp