Python mocked_table Examples

Example #1

def test_create_survey(dynamodb_table):
    from src.data.create_survey import create_survey
    survey_instance = StubSurvey()
    table = mocked_table()
    assert create_survey(
        survey=survey_instance,
        table=table
    ) == survey_instance

Example #2

File: test_create_response.py Project: shinomathew/serverless-surveys

def test_create_response(dynamodb_table):
    from src.data.create_response import create_response
    response_instance = StubResponse()
    table = mocked_table()
    assert create_response(
        response=response_instance,
        table=table
    ) == response_instance

Example #3

File: test_get_customer_handler.py Project: shinomathew/serverless-surveys

def setup_table_item(dynamodb_table):
    table = mocked_table()
    item = {
        'PK': 'CUSTOMER#1',
        'SK': 'PROFILE#1',
        'customer_id': '1',
        'profile_data': {'some': 'data'}
    }
    table.put_item(Item=item)

Example #4

File: test_create_response.py Project: shinomathew/serverless-surveys

def test_create_response_returns_error_info_when_it_fails(dynamodb_table):
    from src.data.create_response import (
        create_response, ResponseCreationException
    )
    table = mocked_table()
    with pytest.raises(ResponseCreationException):
        create_response(
            response='Garbage',
            table=table
        )

Example #5

File: test_get_survey_handler.py Project: shinomathew/serverless-surveys

def setup_table_item(dynamodb_table):
    table = mocked_table()
    item = {
        'PK': 'CUSTOMER#1',
        'SK': 'SURVEY#1',
        'customer_id': '1',
        'survey_id': '1',
        'survey_data': {
            'some': 'data'
        }
    }
    table.put_item(Item=item)

Example #6

File: test_get_customer.py Project: shinomathew/serverless-surveys

def test_get_customer(dynamodb_table):
    from src.data.get_customer import get_customer
    table = mocked_table()
    item = {
        'PK': 'CUSTOMER#TEST_ID',
        'SK': 'PROFILE#TEST_ID',
        'customer_id': 'TEST_ID',
        'profile_data': {
            'some': 'data'
        }
    }
    table.put_item(Item=item)
    customer_instance = StubCustomer()
    assert get_customer(customer=customer_instance,
                        table=table).to_item() == item

Example #7

File: test_get_survey.py Project: shinomathew/serverless-surveys

def test_get_survey(dynamodb_table):
    from src.data.get_survey import get_survey
    table = mocked_table()
    item = {
        'PK': 'CUSTOMER#TEST_ID',
        'SK': 'SURVEY#TEST_ID',
        'customer_id': 'TEST_ID',
        'survey_id': 'TEST_ID',
        'survey_data': {
            'some': 'data'
        }
    }
    table.put_item(Item=item)
    survey_instance = StubSurvey()
    assert get_survey(survey=survey_instance, table=table).to_item() == item

Example #8

def test_get_response(dynamodb_table):
    from src.data.get_response import get_response
    table = mocked_table()
    item = {
        'PK': 'SURVEY#TEST_ID',
        'SK': 'RESPONSE#TEST_ID',
        'survey_id': 'TEST_ID',
        'response_id': 'TEST_ID',
        'response_data': {
            'some': 'data'
        }
    }
    table.put_item(Item=item)
    response_instance = StubResponse()
    assert get_response(response=response_instance,
                        table=table).to_item() == item

Example #9

def test_create_customer_returns_error_info_when_it_fails(dynamodb_table):
    from src.data.create_customer import (create_customer,
                                          CustomerCreationException)
    table = mocked_table()
    with pytest.raises(CustomerCreationException):
        create_customer(customer='Garbage', table=table)

Example #10

def test_create_customer_returns_customer_instance(dynamodb_table):
    from src.data.create_customer import create_customer
    customer_instance = StubCustomer()
    table = mocked_table()
    assert create_customer(customer=customer_instance,
                           table=table) == customer_instance