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")