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)