def test_fetch_not_found(): sheet_handler.sheet_model.get = Mock(wraps=mock_fetch_sheet) assert sheet_handler.fetch(valid_input_with_not_valid_sheet_id)['statusCode'] == 404
def test_fetch_all_registers(): sheet_handler.sheet_model.get = Mock(wraps=mock_fetch_sheet) assert sheet_handler.fetch(valid_input_without_sheet_id)['body'] == json.dumps([sheet_for_valid_user])
def test_fetch_single_register(): sheet_handler.sheet_model.get = Mock(wraps=mock_fetch_sheet) assert sheet_handler.fetch(valid_input_with_valid_sheet_id)['body'] == json.dumps(sheet_for_valid_user)
def test_fetch_401_on_not_valid_user(): assert sheet_handler.fetch(empty_input)['statusCode'] == 403
def test_fetch_422_on_not_dict_input(): assert sheet_handler.fetch(string_value_variable)['statusCode'] == 422
def lambda_handler(event, context): return sheet_handler.fetch(event)