text: # Licensed to the Apache Software Foundation (ASF) under one
  # or more contributor license agreements.  See the NOTICE file
  # distributed with this work for additional information
  # regarding copyright ownership.  The ASF licenses this file
  # to you under the Apache License, Version 2.0 (the
  # "License"); you may not use this file except in compliance
  # with the License.  You may obtain a copy of the License at
  #
  #   http://www.apache.org/licenses/LICENSE-2.0
  #
  # Unless required by applicable law or agreed to in writing,
  # software distributed under the License is distributed on an
  # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  # KIND, either express or implied.  See the License for the
  # specific language governing permissions and limitations
  # under the License.
  
  # Validates table stored on the LocalFileSystem.
  #
  import pytest
  from subprocess import check_call, call
  
  from tests.common.impala_test_suite import ImpalaTestSuite
  from tests.common.skip import SkipIf
  from tests.common.test_dimensions import create_single_exec_option_dimension
  from tests.util.filesystem_utils import get_secondary_fs_path
  
  @SkipIf.no_secondary_fs
  class TestMultipleFilesystems(ImpalaTestSuite):
    """
    Tests that tables and queries can span multiple filesystems.
    """
  
    @classmethod
    def get_workload(self):
      return 'functional-query'
  
    @classmethod
    def add_test_dimensions(cls):
      super(TestMultipleFilesystems, cls).add_test_dimensions()
      cls.ImpalaTestMatrix.add_dimension(create_single_exec_option_dimension())
  
      cls.ImpalaTestMatrix.add_constraint(lambda v:\
          v.get_value('table_format').file_format == 'text' and \
          v.get_value('table_format').compression_codec == 'none')
  
    def _populate_secondary_fs_partitions(self, db_name):
      # This directory may already exist. So we needn't mind if this call fails.
      call(["hadoop", "fs", "-mkdir", get_secondary_fs_path("/multi_fs_tests/")], shell=False)
      check_call(["hadoop", "fs", "-mkdir",
                  get_secondary_fs_path("/multi_fs_tests/%s.db/" % db_name)], shell=False)
      check_call(["hadoop", "fs", "-cp", "/test-warehouse/alltypes_parquet/",
                  get_secondary_fs_path("/multi_fs_tests/%s.db/" % db_name)], shell=False)
      check_call(["hadoop", "fs", "-cp", "/test-warehouse/tinytable/",
                  get_secondary_fs_path("/multi_fs_tests/%s.db/" % db_name)], shell=False)
  
    @pytest.mark.execute_serially
    def test_multiple_filesystems(self, vector, unique_database):
      try:
        self._populate_secondary_fs_partitions(unique_database)
        self.run_test_case('QueryTest/multiple-filesystems', vector, use_db=unique_database)
      finally:
        # We delete this from the secondary filesystem here because the database was created
        # in HDFS but the queries will create this path in the secondary FS as well. So
        # dropping the database will not delete the directory in the secondary FS.
        check_call(["hadoop", "fs", "-rm", "-r",
            get_secondary_fs_path("/multi_fs_tests/%s.db/" % unique_database)], shell=False)
  