def METHOD_NAME(self, mock_query):
    mock_query.return_value = ResultSet(
        rows=[("name",)], column_list=["column_name"]
    )
    result = self.engine.get_all_columns_by_tb("some_db", "some_table")
    self.assertEqual(result.rows, ["name"])
    self.assertEqual(result.column_list, ["column_name"])