def test_get_view_names(self, mock_is_feature_enabled): mock_is_feature_enabled.return_value = True mock_execute = mock.MagicMock() mock_fetchall = mock.MagicMock( return_value=[["a", "b,", "c"], ["d", "e"]]) database = mock.MagicMock() database.get_sqla_engine.return_value.raw_connection.return_value.cursor.return_value.execute = ( mock_execute) database.get_sqla_engine.return_value.raw_connection.return_value.cursor.return_value.fetchall = ( mock_fetchall) result = PrestoEngineSpec.get_view_names(database, mock.Mock(), None) mock_execute.assert_called_once_with( "SELECT table_name FROM information_schema.views", {}) assert result == ["a", "d"]
def test_presto_get_view_names_return_empty_list(self): self.assertEquals([], PrestoEngineSpec.get_view_names( mock.ANY, mock.ANY, mock.ANY))
def test_presto_get_view_names_return_empty_list( self, ): # pylint: disable=invalid-name self.assertEqual( [], PrestoEngineSpec.get_view_names(mock.ANY, mock.ANY, mock.ANY) )