def test_set_method_ignores_conflict(self):
update_fields = ({'name': 'test'})
where_fields = ({'id': 1})
update_q = UpdateQuery(self.stub_model,
update_fields=update_fields,
where_fields=where_fields)
update_q.set(name='test')
self.assertEqual('update stub_model set name=test where id=1;',
update_q.sql)
def test_commit(self, job_spawner):
update_fields = ({'name': 'test'})
where_fields = ({'id': 1})
update_q = UpdateQuery(self.stub_model,
update_fields=update_fields,
where_fields=where_fields)
update_q.commit()
job_spawner.assert_called_with(update_q.sql,
self.stub_model.__databases__,
commit=True)
def test_set_method_changes_update_list_correctly(self):
update_fields = ({'name': 'test'})
where_fields = ({'id': 1})
update_q = UpdateQuery(self.stub_model,
update_fields=update_fields,
where_fields=where_fields)
update_q.set(age=24)
self.assertEqual(
'update stub_model set name=test and age=24 where id=1;',
update_q.sql)
def test_update_query_without_where_fields(self):
try:
update_q = UpdateQuery(self.stub_model, update_fields=[{'id': 1}])
except AssertionError as no_param_err:
self.assertEqual(
"Please specify in which fields you want to update. e.g where_fields=[{id:1}]",
no_param_err.args[0])
def test_update_query_with_without_update_fields(self):
try:
update_q = UpdateQuery(self.stub_model)
except AssertionError as no_param_err:
self.assertEqual(
"Please define the fields you want to update. e.g update_fields=[{name:mehmet}]",
no_param_err.args[0])
def test_update_query_with_kwargs_n_args(self):
update_fields = ({'name': 'test'})
where_fields = ({'id': 1})
update_q = UpdateQuery(self.stub_model,
update_fields=update_fields,
where_fields=where_fields)
self.assertEqual('update stub_model set name=test where id=1;',
update_q.sql)
def test_sql_property(self):
update_fields = ({'name': 'test'})
where_fields = ({'id': 1})
update_q = UpdateQuery(self.stub_model,
update_fields=update_fields,
where_fields=where_fields)
self.assertEqual(
update_q.sql,
update_q.base_sql.format(tablename=update_q.model.__tablename__,
update_columns=' and '.join(
update_q.update_list)))