def test_create_update_statement_with_where_fields(self):
table = "test"
fields = ["field1", "field2"]
where_fields = [
{
"field_name": "field3",
"operator": ">",
},
{
"field_name": "field4",
"operator": "=",
},
]
statement_fields, statement = UpdateStatementMixin.create_update_statement(
table=table,
fields=fields,
where_fields=where_fields,
statement_string=True,
)
expected_statement_fields = fields + [
field["field_name"] for field in where_fields
]
self.assertTrue(statement_fields == expected_statement_fields)
for word in [
"test", "field1", "field2", "field3", "field4", ">", "=",
"UPDATE", "SET", "WHERE"
]:
self.assertTrue(word in statement)
def test_create_update_statement_without_where_fields(self):
table = "test"
fields = ["field1", "field2"]
statement_fields, statement = UpdateStatementMixin.create_update_statement(
table=table, fields=fields, where_fields=None, statement_string=True
)
expected_statement_fields = fields
self.assertTrue(statement_fields == expected_statement_fields)
for word in ["test", "field1", "field2", "UPDATE", "SET"]:
self.assertTrue(word in statement)
def test_create_update_statement_with_where_fields(self):
table = "test"
fields = ["field1", "field2"]
where_fields = [{"field_name": "field3", "operator": ">"}, {"field_name": "field4", "operator": "="}]
statement_fields, statement = UpdateStatementMixin.create_update_statement(
table=table, fields=fields, where_fields=where_fields, statement_string=True
)
expected_statement_fields = fields + [field["field_name"] for field in where_fields]
self.assertTrue(statement_fields == expected_statement_fields)
for word in ["test", "field1", "field2", "field3", "field4", ">", "=", "UPDATE", "SET", "WHERE"]:
self.assertTrue(word in statement)
def test_create_update_statement_without_where_fields(self):
table = "test"
fields = ["field1", "field2"]
statement_fields, statement = UpdateStatementMixin.create_update_statement(
table=table,
fields=fields,
where_fields=None,
statement_string=True,
)
expected_statement_fields = fields
self.assertTrue(statement_fields == expected_statement_fields)
for word in ["test", "field1", "field2", "UPDATE", "SET"]:
self.assertTrue(word in statement)
