def test_postal_code_handles_errors_properly(self):
SQLite3Mock.fetchall_returns = [sqlite3.Error('Error')]
result = self.zcp.get_postal_code_data('21901')
self.assertIsNone(result)
self.assertEqual(
SQLite3Mock.execute_calls,
[
('PRAGMA temp_store = 2', None),
('SELECT * FROM city WHERE postal_code = ?', ('21901',))
]
)
SQLite3Mock.reset()
SQLite3Mock.fetchall_returns = [
sqlite3.Error('Error'),
('a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l'),
]
result = self.zcp.get_postal_code_data('21901-2000')
self.assertIsNone(result)
self.assertEqual(
SQLite3Mock.execute_calls,
[
('PRAGMA temp_store = 2', None),
('SELECT * FROM city WHERE postal_code = ?', ('21901-2000',)),
('SELECT * FROM city WHERE postal_code = ?', ('21901',))
]
)
def test_postal_code_handles_errors_properly(self):
SQLite3Mock.fetchall_returns = [sqlite3.Error('Error')]
result = self.zcp.get_postal_code_data('21901')
self.assertIsNone(result)
self.assertEqual(SQLite3Mock.execute_calls,
[('PRAGMA temp_store = 2', None),
('SELECT * FROM city WHERE postal_code = ?',
('21901', ))])
SQLite3Mock.reset()
SQLite3Mock.fetchall_returns = [
sqlite3.Error('Error'),
('a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l'),
]
result = self.zcp.get_postal_code_data('21901-2000')
self.assertIsNone(result)
self.assertEqual(SQLite3Mock.execute_calls,
[('PRAGMA temp_store = 2', None),
('SELECT * FROM city WHERE postal_code = ?',
('21901-2000', )),
('SELECT * FROM city WHERE postal_code = ?',
('21901', ))])
def tearDown(self):
SQLite3Mock.reset()
registry.flush()
self.ap = None
def tearDown(self):
SQLite3Mock.reset()
registry.flush()
self.zcp = None
