def mock_wheel_table()

in api/conftest.py [0:0]

• 9 lines of code
• 2 McCabe index (conditional complexity)

def mock_wheel_table(mock_dynamodb):
    Wheel = mock_dynamodb.Table(WHEEL_TABLE_NAME)
    add_extended_table_functions(Wheel)
    utils.Wheel = Wheel
    yield Wheel
    wheels = Wheel.scan()['Items']
    with Wheel.batch_writer() as batch:
        for wheel in wheels:
            batch.delete_item(Key={'id': wheel['id']})