class TestCidCursor(TestCase):
def setUp(self):
self.cursor = mock.Mock()
self.cursor.execute = mock.Mock(return_value=None)
self.cursor.executemany = mock.Mock(return_value=None)
self.cursor_wrapper = CidCursorWrapper(self.cursor)
@mock.patch('cid.cursor.get_cid')
def test_adds_comment(self, get_cid):
get_cid.return_value = 'testing-cursor'
expected = "/* cid: testing-cursor */\nSELECT 1;"
self.assertEqual(
expected,
self.cursor_wrapper.add_comment("SELECT 1;")
)
@override_settings(CID_SQL_COMMENT_TEMPLATE='correlation_id={cid}')
@mock.patch('cid.cursor.get_cid')
def test_adds_comment_setting_overriden(self, get_cid):
get_cid.return_value = 'testing-cursor'
expected = "/* correlation_id=testing-cursor */\nSELECT 1;"
self.assertEqual(
expected,
self.cursor_wrapper.add_comment("SELECT 1;")
)
@mock.patch('cid.cursor.get_cid')
def test_no_comment_when_cid_is_none(self, get_cid):
get_cid.return_value = None
expected = "SELECT 1;"
self.assertEqual(
expected,
self.cursor_wrapper.add_comment("SELECT 1;")
)
@mock.patch('cid.cursor.CidCursorWrapper.add_comment')
def test_execute_calls_add_comment(self, add_comment):
sql = "SELECT 1;"
self.cursor_wrapper.execute(sql)
add_comment.assert_called_with(sql)
@mock.patch('cid.cursor.CidCursorWrapper.add_comment')
def test_executemany_calls_add_comment(self, add_comment):
sql = "SELECT 1;"
self.cursor_wrapper.executemany(sql, [])
add_comment.assert_called_with(sql)
@mock.patch('cid.cursor.get_cid')
def test_escape_cid(self, get_cid):
get_cid.return_value = '/* a correlation id with funny characters */'
expected = '/* cid: \\/\\* a correlation id with funny characters \\*\\/ */\nSELECT 1;'
self.assertEqual(self.cursor_wrapper.add_comment('SELECT 1;'), expected)
class TestCidCursor(TestCase):
def setUp(self):
self.cursor = mock.Mock()
self.cursor.execute = mock.Mock(return_value=None)
self.cursor.executemany = mock.Mock(return_value=None)
self.cursor_wrapper = CidCursorWrapper(self.cursor)
@mock.patch('cid.cursor.get_cid')
def test_adds_comment(self, get_cid):
get_cid.return_value = 'testing-cursor'
expected = "/* cid: testing-cursor */\nSELECT 1;"
self.assertEqual(
expected,
self.cursor_wrapper.add_comment("SELECT 1;")
)
@override_settings(CID_SQL_COMMENT_TEMPLATE='correlation_id={cid}')
@mock.patch('cid.cursor.get_cid')
def test_adds_comment_setting_overriden(self, get_cid):
get_cid.return_value = 'testing-cursor'
expected = "/* correlation_id=testing-cursor */\nSELECT 1;"
self.assertEqual(
expected,
self.cursor_wrapper.add_comment("SELECT 1;")
)
@mock.patch('cid.cursor.get_cid')
def test_no_comment_when_cid_is_none(self, get_cid):
get_cid.return_value = None
expected = "SELECT 1;"
self.assertEqual(
expected,
self.cursor_wrapper.add_comment("SELECT 1;")
)
@mock.patch('cid.cursor.CidCursorWrapper.add_comment')
def test_execute_calls_add_comment(self, add_comment):
sql = "SELECT 1;"
self.cursor_wrapper.execute(sql)
add_comment.assert_called_with(sql)
@mock.patch('cid.cursor.CidCursorWrapper.add_comment')
def test_executemany_calls_add_comment(self, add_comment):
sql = "SELECT 1;"
self.cursor_wrapper.executemany(sql, [])
add_comment.assert_called_with(sql)
@mock.patch('cid.cursor.get_cid')
def test_escape_cid(self, get_cid):
get_cid.return_value = '/* a correlation id with funny characters */'
expected = '/* cid: \\/\\* a correlation id with funny characters \\*\\/ */\nSELECT 1;'
self.assertEqual(self.cursor_wrapper.add_comment('SELECT 1;'), expected)
class TestCidCursor(TestCase):
def setUp(self):
self.cursor = Mock()
self.cursor.execute = Mock(return_value=None)
self.cursor.executemany = Mock(return_value=None)
self.cursor_wrapper = CidCursorWrapper(self.cursor)
@patch('cid.cursor.get_cid')
def test_adds_comment(self, get_cid):
get_cid.return_value = 'testing-cursor'
expected = "/* cid: testing-cursor */\nSELECT 1;"
self.assertEqual(
expected,
self.cursor_wrapper.add_comment("SELECT 1;")
)
@patch('cid.cursor.get_cid')
def test_no_comment_when_cid_is_none(self, get_cid):
get_cid.return_value = None
expected = "SELECT 1;"
self.assertEqual(
expected,
self.cursor_wrapper.add_comment("SELECT 1;")
)
@patch('cid.cursor.CidCursorWrapper.add_comment')
def test_execute_calls_add_comment(self, add_comment):
sql = "SELECT 1;"
self.cursor_wrapper.execute(sql)
add_comment.assert_called_with(sql)
@patch('cid.cursor.CidCursorWrapper.add_comment')
def test_executemany_calls_add_comment(self, add_comment):
sql = "SELECT 1;"
self.cursor_wrapper.executemany(sql, [])
add_comment.assert_called_with(sql)
def setUp(self):
self.cursor = mock.Mock()
self.cursor.execute = mock.Mock(return_value=None)
self.cursor.executemany = mock.Mock(return_value=None)
self.cursor_wrapper = CidCursorWrapper(self.cursor)
def setUp(self):
self.cursor = mock.Mock()
self.cursor.execute = mock.Mock(return_value=None)
self.cursor.executemany = mock.Mock(return_value=None)
self.cursor_wrapper = CidCursorWrapper(self.cursor)
