def setUp(self): self.con1 = FakeConnection() self.con2 = FakeConnection() self.pc = ParallelConnection([self.con1, self.con2])
class PCTestCase(TestCase): def setUp(self): self.con1 = FakeConnection() self.con2 = FakeConnection() self.pc = ParallelConnection([self.con1, self.con2]) def test_cursor(self): self.pc.cursor('a', b='c') self.con1.assert_called_once_with('cursor', 'a', b='c') self.con2.assert_called_once_with('cursor', 'a', b='c') def test_execute(self): self.pc.cursor() self.pc.execute("hello world", 42) for cursor in self.pc.cursors: cursor.assert_called_once_with("execute", "hello world", 42) def test_fetchone(self): self.pc.cursor() self.pc.execute("hello world", 42) results = self.pc.fetchone() self.assertEqual(results, {1:'a', 2 :'b'}) for cursor in self.pc.cursors: cursor.assert_called_once_with("execute", "hello world", 42) cursor.assert_called_once_with("fetchone") def test_fetchall(self): self.pc.cursor() self.pc.execute("hello world", 42) results = self.pc.fetchall() self.assertEqual(results, [{1:'a', 2:'b'}, {1:'a', 2:'b'}]) for cursor in self.pc.cursors: cursor.assert_called_once_with("execute", "hello world", 42) cursor.assert_called_once_with("fetchall") def test_commit(self): self.pc.commit() self.con1.assert_called_once_with('commit') self.con2.assert_called_once_with('commit') def test_close(self): self.pc.cursor() cursors = self.pc.cursors[:] self.pc.close() for cursor in cursors: cursor.assert_called_once_with('close') self.assertIsNone(self.pc.cursors) def test_mogrify(self): self.pc.cursor() self.pc.mogrify("hello world") self.pc.cursors[0].assert_called_once_with('mogrify', "hello world")