Example #1
0
    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 #')),
        ]
Example #2
0
    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 #")),
        ]
Example #3
0
 def test_secondary(self):
     callback = mock.Mock()
     with DBRecorder('second', callback):
         run_query('second', 'SELECT 1')
     callback.assert_called_once_with(DBOp('second', 'SELECT #'))
Example #4
0
 def test_not_equal_sql(self):
     assert (
         DBOp('foo', 'bar') !=
         DBOp('foo', 'baz')
     )
Example #5
0
 def test_not_equal_alias(self):
     assert (
         DBOp('foo', 'bar') !=
         DBOp('baz', 'bar')
     )
Example #6
0
 def test_equal(self):
     assert (
         DBOp('foo', 'bar') ==
         DBOp('foo', 'bar')
     )
Example #7
0
 def test_create(self):
     op = DBOp('myalias', 'SELECT 1')
     assert op.alias == 'myalias'
     assert op.sql == 'SELECT 1'
Example #8
0
 def test_not_equal_traceback(self):
     assert DBOp("foo", "bar", "traceback") != DBOp("foo", "baz", None)
Example #9
0
 def test_secondary(self):
     callback = mock.Mock()
     with DBRecorder("second", callback):
         run_query("second", "SELECT 1")
     callback.assert_called_once_with(DBOp("second", "SELECT #"))
Example #10
0
 def test_not_equal_sql(self):
     assert DBOp("foo", "bar") != DBOp("foo", "baz")
Example #11
0
 def test_not_equal_alias(self):
     assert DBOp("foo", "bar") != DBOp("baz", "bar")
Example #12
0
 def test_equal(self):
     assert DBOp("foo", "bar") == DBOp("foo", "bar")
Example #13
0
 def test_create(self):
     op = DBOp("myalias", "SELECT 1")
     assert op.alias == "myalias"
     assert op.sql == "SELECT 1"
Example #14
0
 def test_create(self):
     op = DBOp("myalias", "SELECT 1", None)
     assert op.alias == "myalias"
     assert op.query == "SELECT 1"
     assert op.traceback is None
Example #15
0
 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))
Example #16
0
 def test_replica(self):
     callback = mock.Mock()
     with DBRecorder('replica', callback):
         run_query('replica', 'SELECT 1')
     callback.assert_called_once_with(DBOp('replica', 'SELECT #'))
Example #17
0
 def test_replica(self):
     callback = mock.Mock()
     with DBRecorder("replica", callback):
         run_query("replica", "SELECT 1")
     callback.assert_called_once_with(DBOp("replica", "SELECT #"))
Example #18
0
 def test_equal(self):
     assert DBOp("foo", "bar", "traceback") == DBOp("foo", "bar",
                                                    "traceback")