def test_records_all(self):
callback = mock.Mock()
with AllDBRecorder(callback):
run_query('replica', 'SELECT 1')
run_query('default', 'SELECT 2')
run_query('second', 'SELECT 3')
assert callback.mock_calls == [
mock.call(DBOp('replica', 'SELECT #')),
mock.call(DBOp('default', 'SELECT #')),
mock.call(DBOp('second', 'SELECT #')),
]
def test_records_all(self):
callback = mock.Mock()
with AllDBRecorder(callback):
run_query("replica", "SELECT 1")
run_query("default", "SELECT 2")
run_query("second", "SELECT 3")
assert callback.mock_calls == [
mock.call(DBOp("replica", "SELECT #")),
mock.call(DBOp("default", "SELECT #")),
mock.call(DBOp("second", "SELECT #")),
]
def test_secondary(self):
callback = mock.Mock()
with DBRecorder('second', callback):
run_query('second', 'SELECT 1')
callback.assert_called_once_with(DBOp('second', 'SELECT #'))
def test_not_equal_sql(self):
assert (
DBOp('foo', 'bar') !=
DBOp('foo', 'baz')
)
def test_not_equal_alias(self):
assert (
DBOp('foo', 'bar') !=
DBOp('baz', 'bar')
)
def test_equal(self):
assert (
DBOp('foo', 'bar') ==
DBOp('foo', 'bar')
)
def test_create(self):
op = DBOp('myalias', 'SELECT 1')
assert op.alias == 'myalias'
assert op.sql == 'SELECT 1'
def test_not_equal_traceback(self):
assert DBOp("foo", "bar", "traceback") != DBOp("foo", "baz", None)
def test_secondary(self):
callback = mock.Mock()
with DBRecorder("second", callback):
run_query("second", "SELECT 1")
callback.assert_called_once_with(DBOp("second", "SELECT #"))
def test_not_equal_sql(self):
assert DBOp("foo", "bar") != DBOp("foo", "baz")
def test_not_equal_alias(self):
assert DBOp("foo", "bar") != DBOp("baz", "bar")
def test_equal(self):
assert DBOp("foo", "bar") == DBOp("foo", "bar")
def test_create(self):
op = DBOp("myalias", "SELECT 1")
assert op.alias == "myalias"
assert op.sql == "SELECT 1"
def test_create(self):
op = DBOp("myalias", "SELECT 1", None)
assert op.alias == "myalias"
assert op.query == "SELECT 1"
assert op.traceback is None
def test_default(self, extract_stack):
callback = mock.Mock()
with DBRecorder("default", callback):
run_query("default", "SELECT 1")
callback.assert_called_once_with(DBOp("default", "SELECT #", None))
def test_replica(self):
callback = mock.Mock()
with DBRecorder('replica', callback):
run_query('replica', 'SELECT 1')
callback.assert_called_once_with(DBOp('replica', 'SELECT #'))
def test_replica(self):
callback = mock.Mock()
with DBRecorder("replica", callback):
run_query("replica", "SELECT 1")
callback.assert_called_once_with(DBOp("replica", "SELECT #"))
def test_equal(self):
assert DBOp("foo", "bar", "traceback") == DBOp("foo", "bar",
"traceback")
