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()