def test_missing_id_column(self):
querycolumns = (self.col2, self.col3)
connfactory = MagicMock()
connfactory.get_conn().cursor().fetchall.return_value = self.postgre_type_codes
connfactory.get_conn().cursor().description = querycolumns
query = inputquery.InputQuery(
self.inputsql, self.dbname, self.user, self.password, self.host, self.port, connfactory)
self.assertRaises(RuntimeError, query.get_columns)
def test_column_identification(self):
querycolumns = (self.col1, self.col2, self.col3)
connfactory = MagicMock()
connfactory.get_conn().cursor().fetchall.return_value = self.postgre_type_codes
connfactory.get_conn().cursor().description = querycolumns
query = inputquery.InputQuery(
self.inputsql, self.dbname, self.user, self.password, self.host, self.port, connfactory)
self.assertEqual(
query.get_columns(),
[
{'datatype': 'int', 'name': 'ID'},
{'datatype': 'varchar', 'name': 'name'},
{'datatype': 'varchar', 'name': 'phone'}
]
)
connfactory.get_conn().cursor().fetchall.assert_called_with()
