def test_get_update_statement(self, mocked_get_loader_fields, mocked_get_loader_table, mocked_get_sql_where_fields):
mocked_get_loader_fields.return_value = "mock_fields"
mocked_get_loader_table.return_value = "mock_table"
mocked_get_sql_where_fields.return_value = "mock_where_fields"
UpdateStatementMixin.create_update_statement = MagicMock()
update_statement_mixin = UpdateStatementMixin()
update_statement_mixin.get_update_statement()
UpdateStatementMixin.create_update_statement.assert_called_with(
table="mock_table", fields="mock_fields", where_fields="mock_where_fields", statement_string=True
)
def test_get_update_statement(self, mocked_get_loader_fields,
mocked_get_loader_table,
mocked_get_sql_where_fields):
mocked_get_loader_fields.return_value = "mock_fields"
mocked_get_loader_table.return_value = "mock_table"
mocked_get_sql_where_fields.return_value = "mock_where_fields"
UpdateStatementMixin.create_update_statement = MagicMock()
update_statement_mixin = UpdateStatementMixin()
update_statement_mixin.get_update_statement()
UpdateStatementMixin.create_update_statement.assert_called_with(
table="mock_table",
fields="mock_fields",
where_fields="mock_where_fields",
statement_string=True)
