How to write Pytest for AWS Athena service

I am writing pytest for my lambda which is using AThena, created pytest.fixture using below code:

@pytest.fixture()
def moto_boto_athena():
    @mock_athena
    def boto_resource_athena():
        res = boto3.client('athena')
        return res
    return boto_resource_athena

My Lambda is having below code:

response = get_athena_client().start_query_execution(
        QueryString=query_string,
        QueryExecutionContext={
            'Database': athena_database
        },
        ResultConfiguration={
            'OutputLocation': athena_output_bucket,
        }
    )

    query_status = None
    #Check the status of QueryExecution
    while query_status == 'QUEUED' or query_status == 'RUNNING' or query_status is None:
        query_status = get_athena_client(). \
            get_query_execution(QueryExecutionId=response['QueryExecutionId'])['QueryExecution']['Status']['State']
        if query_status == 'FAILED' or query_status == 'CANCELLED':
            raise Exception

But as soon as I start my test execution, getting below exception:

raise NotImplementedError(
>           "The {0} action has not been implemented".format(action)
        )
E       NotImplementedError: The start_query_execution action has not been implemented

..\..\..\..\..\AppData\Roaming\Python\Python36\site-packages\moto\core\responses.py:393: NotImplementedError

Please guide.