Python handler Examples

Example #1

File: test_handler.py Project: maxgala/aspire-backend

def test_delete_404(apigw_delete_event, mocker):
    ret = delete.handler(apigw_delete_event("tag1"), "")
    assert ret["statusCode"] == 404

    ret = delete.handler(apigw_delete_event("tag2"), "")
    assert ret["statusCode"] == 404

    ret = delete.handler(apigw_delete_event("tag3"), "")
    assert ret["statusCode"] == 404

    ret = delete.handler(apigw_delete_event("other tag"), "")
    assert ret["statusCode"] == 404

Example #2

    def test_delete_404(self):
        tag = "tag1"
        with mock.patch("DeleteById.lambda_function.Session") as mock_session:
            mock_query = mock_session.return_value.query
            mock_get = mock_query.return_value.get
            mock_get.return_value = None
            ret = delete.handler(apigw_delete_event(tag), "")

        self.assertEqual(ret["statusCode"], 404, self.msg_status_code.format(404, ret["statusCode"]))

Example #3

    def test_delete_200(self):
        tag = "tag1"
        with mock.patch("DeleteById.lambda_function.Session") as mock_session:
            mock_delete = mock_session.return_value.delete
            mock_delete.side_effect = tags_db.remove(tag)
            ret = delete.handler(apigw_delete_event(tag), "")

        self.assertEqual(ret["statusCode"], 200, self.msg_status_code.format(200, ret["statusCode"]))
        self.assertTrue(tag not in tags_db)

Example #4

    def test_delete_400(self):
        tag = ""
        with mock.patch("DeleteById.lambda_function.Session") as mock_session:
            ret = delete.handler(apigw_delete_event(tag), "")

        self.assertEqual(ret["statusCode"], 400, self.msg_status_code.format(400, ret["statusCode"]))

Example #5

File: test_handler.py Project: maxgala/aspire-backend

def test_delete_400(apigw_delete_event, mocker):
    ret = delete.handler(apigw_delete_event(""), "")

    assert ret["statusCode"] == 400