#!/usr/bin/env python3
"""
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.
"""

import os

os.environ["ROOT"] = ""

import time
import subprocess
import os
import logging
import tempfile
import pprint
from xml.dom import minidom

from unittest import TestCase
from mock.mock import MagicMock, call
from mock.mock import patch
from mock.mock import create_autospec
from only_for_platform import (
  get_platform,
  not_for_platform,
  only_for_platform,
  os_distro_value,
  PLATFORM_WINDOWS,
)

from ambari_server.resourceFilesKeeper import ResourceFilesKeeper, KeeperException


class TestResourceFilesKeeper(TestCase):
  TEST_RESOURCES_DIR = ".." + os.sep + "resources"
  TEST_STACKS_DIR = ".." + os.sep + "resources" + os.sep + "stacks"

  # Stack that is not expected to change
  DUMMY_UNCHANGEABLE_STACK = (
    ".."
    + os.sep
    + "resources"
    + os.sep
    + "TestAmbaryServer.samples"
    + os.sep
    + "dummy_stack"
    + os.sep
    + "HIVE"
  )

  DUMMY_UNCHANGEABLE_EXTENSION = (
    ".."
    + os.sep
    + "resources"
    + os.sep
    + "TestAmbaryServer.samples"
    + os.sep
    + "dummy_extension"
    + os.sep
    + "HIVE"
  )

  DUMMY_ACTIVE_STACK = (
    ".."
    + os.sep
    + "resources"
    + os.sep
    + "TestAmbaryServer.samples"
    + os.sep
    + "active_stack"
  )

  DUMMY_INACTIVE_STACK = (
    ".."
    + os.sep
    + "resources"
    + os.sep
    + "TestAmbaryServer.samples"
    + os.sep
    + "inactive_stack"
  )

  DUMMY_UNCHANGEABLE_PACKAGE = os.path.join(
    DUMMY_UNCHANGEABLE_STACK, ResourceFilesKeeper.PACKAGE_DIR
  )

  if get_platform() != PLATFORM_WINDOWS:
    DUMMY_UNCHANGEABLE_PACKAGE_HASH = "290e2adef52648fba2865c192ccdc1dd6858c974"
  else:
    DUMMY_UNCHANGEABLE_PACKAGE_HASH = "2e438f4f9862420ed8930a56b8809b8aca359e87"
  DUMMY_HASH = "dummy_hash"
  YA_HASH = "yet_another_hash"
  SOME_PATH = "some-path"

  DUMMY_UNCHANGEABLE_COMMON_SERVICES = (
    ".."
    + os.sep
    + "resources"
    + os.sep
    + "TestAmbaryServer.samples"
    + os.sep
    + "dummy_common_services"
    + os.sep
    + "HIVE"
    + os.sep
    + "0.11.0.2.0.5.0"
  )

  DUMMY_UNCHANGEABLE_COMMON_SERVICES_PACKAGE = os.path.join(
    DUMMY_UNCHANGEABLE_COMMON_SERVICES, ResourceFilesKeeper.PACKAGE_DIR
  )

  if get_platform() != PLATFORM_WINDOWS:
    UPDATE_DIRECTORY_ARCHIVE_CALL_LIST = (
      "[call('../resources/TestAmbaryServer.samples/"
      "dummy_stack/HIVE/package'),\n "
      "call('../resources/TestAmbaryServer.samples/"
      "dummy_stack/HIVE/package'),\n "
      "call('../resources/TestAmbaryServer.samples/"
      "dummy_stack/HIVE/package'),\n "
      "call('../resources/TestAmbaryServer.samples/"
      "dummy_common_services/HIVE/0.11.0.2.0.5.0/package'),\n "
      "call('../resources/TestAmbaryServer.samples/"
      "dummy_common_services/HIVE/0.11.0.2.0.5.0/package'),\n "
      "call('../resources/TestAmbaryServer.samples/dummy_extension/HIVE/package'),\n "
      "call('../resources/stack-hooks'),\n "
      "call('../resources/custom_actions'),\n "
      "call('../resources/host_scripts'),\n "
      "call('../resources/dashboards')]"
    )
  else:
    UPDATE_DIRECTORY_ARCHIVE_CALL_LIST = (
      "[call('..\\\\resources\\\\TestAmbaryServer.samples\\\\dummy_stack\\\\HIVE\\\\package'),\n "
      "call('..\\\\resources\\\\TestAmbaryServer.samples\\\\dummy_stack\\\\HIVE\\\\package'),\n "
      "call('..\\\\resources\\\\TestAmbaryServer.samples\\\\dummy_stack\\\\HIVE\\\\package'),\n "
      "call('..\\\\resources\\\\TestAmbaryServer.samples\\\\dummy_common_services\\\\HIVE\\\\0.11.0.2.0.5.0\\\\package'),\n "
      "call('..\\\\resources\\\\TestAmbaryServer.samples\\\\dummy_common_services\\\\HIVE\\\\0.11.0.2.0.5.0\\\\package'),\n "
      "call('..\\\\resources\\\\TestAmbaryServer.samples\\\\dummy_extension\\\\HIVE\\\\package'),\n "
      "call('..\\\\resources\\\\custom_actions'),\n "
      "call('..\\\\resources\\\\host_scripts'),\n "
      "call('..\\\\resources\\\\dashboards')]"
    )

  def setUp(self):
    logging.basicConfig(level=logging.ERROR)

  @patch.object(ResourceFilesKeeper, "update_directory_archives")
  def test_perform_housekeeping(self, update_directory_archives_mock):
    resource_files_keeper = ResourceFilesKeeper(
      os.sep + "dummy-resources", os.sep + "dummy-path"
    )
    resource_files_keeper.perform_housekeeping()
    update_directory_archives_mock.assertCalled()
    pass

  @patch.object(ResourceFilesKeeper, "list_extensions")
  @patch.object(ResourceFilesKeeper, "update_directory_archive")
  @patch.object(ResourceFilesKeeper, "list_common_services")
  @patch.object(ResourceFilesKeeper, "list_stacks")
  @patch("os.path.abspath")
  def test_update_directory_archives(
    self,
    abspath_mock,
    list_active_stacks_mock,
    list_common_services_mock,
    update_directory_archive_mock,
    list_extensions_mock,
  ):
    list_active_stacks_mock.return_value = [
      self.DUMMY_UNCHANGEABLE_STACK,
      self.DUMMY_UNCHANGEABLE_STACK,
      self.DUMMY_UNCHANGEABLE_STACK,
    ]
    list_common_services_mock.return_value = [
      self.DUMMY_UNCHANGEABLE_COMMON_SERVICES,
      self.DUMMY_UNCHANGEABLE_COMMON_SERVICES,
    ]
    list_extensions_mock.return_value = [self.DUMMY_UNCHANGEABLE_EXTENSION]

    abspath_mock.side_effect = lambda s: s
    resource_files_keeper = ResourceFilesKeeper(
      self.TEST_RESOURCES_DIR, self.TEST_STACKS_DIR
    )
    resource_files_keeper.update_directory_archives()
    self.assertEqual(
      pprint.pformat(update_directory_archive_mock.call_args_list),
      self.UPDATE_DIRECTORY_ARCHIVE_CALL_LIST,
    )
    pass

  @patch("glob.glob")
  @patch("os.path.exists")
  def test_list_stacks(self, exists_mock, glob_mock):
    resource_files_keeper = ResourceFilesKeeper(self.TEST_RESOURCES_DIR, self.SOME_PATH)
    # Test normal execution flow
    glob_mock.return_value = ["stack1", "stack2", "stack3"]
    exists_mock.side_effect = [True, False, True]
    res = resource_files_keeper.list_stacks(self.SOME_PATH)
    self.assertEqual(pprint.pformat(res), "['stack1', 'stack3']")

    # Test exception handling
    glob_mock.side_effect = self.keeper_exc_side_effect
    try:
      resource_files_keeper.list_stacks(self.SOME_PATH)
      self.fail("KeeperException not thrown")
    except KeeperException:
      pass  # Expected
    except Exception as e:
      self.fail("Unexpected exception thrown:" + str(e))

  @patch("glob.glob")
  @patch("os.path.exists")
  def test_list_common_services(self, exists_mock, glob_mock):
    resource_files_keeper = ResourceFilesKeeper(self.TEST_RESOURCES_DIR, self.SOME_PATH)
    # Test normal execution flow
    glob_mock.return_value = ["common_service1", "common_service2", "common_service3"]
    exists_mock.side_effect = [True, False, True]
    res = resource_files_keeper.list_common_services(self.SOME_PATH)
    self.assertEqual(pprint.pformat(res), "['common_service1', 'common_service3']")

    # Test exception handling
    glob_mock.side_effect = self.keeper_exc_side_effect
    try:
      resource_files_keeper.list_common_services(self.SOME_PATH)
      self.fail("KeeperException not thrown")
    except KeeperException:
      pass  # Expected
    except Exception as e:
      self.fail("Unexpected exception thrown:" + str(e))

  @patch("os.path.isfile")
  @patch("os.path.exists")
  @patch("os.listdir")
  @patch.object(ResourceFilesKeeper, "count_hash_sum")
  @patch.object(ResourceFilesKeeper, "read_hash_sum")
  @patch.object(ResourceFilesKeeper, "zip_directory")
  @patch.object(ResourceFilesKeeper, "write_hash_sum")
  def test_update_directory_archive(
    self,
    write_hash_sum_mock,
    zip_directory_mock,
    read_hash_sum_mock,
    count_hash_sum_mock,
    os_listdir_mock,
    os_path_exists_mock,
    os_path_isfile_mock,
  ):
    os_listdir_mock.return_value = ["file1", "dir1"]

    # Test situation when there is no saved directory hash
    read_hash_sum_mock.return_value = None
    count_hash_sum_mock.return_value = self.YA_HASH
    os_path_exists_mock.return_value = True
    resource_files_keeper = ResourceFilesKeeper(self.TEST_RESOURCES_DIR, self.SOME_PATH)
    resource_files_keeper.update_directory_archive(self.SOME_PATH)
    self.assertTrue(read_hash_sum_mock.called)
    self.assertTrue(count_hash_sum_mock.called)
    self.assertTrue(zip_directory_mock.called)
    self.assertTrue(write_hash_sum_mock.called)

    read_hash_sum_mock.reset_mock()
    count_hash_sum_mock.reset_mock()
    zip_directory_mock.reset_mock()
    write_hash_sum_mock.reset_mock()

    # Test situation where there is a .hash file, equal to the current hash,
    # but no archive.zip file
    count_hash_sum_mock.return_value = self.YA_HASH
    read_hash_sum_mock.return_value = self.YA_HASH
    os_path_isfile_mock.return_value = False
    resource_files_keeper = ResourceFilesKeeper(self.TEST_RESOURCES_DIR, self.SOME_PATH)
    resource_files_keeper.update_directory_archive(self.SOME_PATH)
    self.assertTrue(read_hash_sum_mock.called)
    self.assertTrue(count_hash_sum_mock.called)
    self.assertTrue(zip_directory_mock.called)
    self.assertFalse(write_hash_sum_mock.called)

    read_hash_sum_mock.reset_mock()
    count_hash_sum_mock.reset_mock()
    zip_directory_mock.reset_mock()
    write_hash_sum_mock.reset_mock()

    # Test situation when saved directory hash == current hash and old archive does not exist
    read_hash_sum_mock.return_value = self.DUMMY_HASH
    count_hash_sum_mock.return_value = self.YA_HASH
    os_path_isfile_mock.return_value = False
    resource_files_keeper.update_directory_archive(self.SOME_PATH)
    self.assertTrue(read_hash_sum_mock.called)
    self.assertTrue(count_hash_sum_mock.called)
    self.assertTrue(zip_directory_mock.called)
    self.assertTrue(write_hash_sum_mock.called)

    read_hash_sum_mock.reset_mock()
    count_hash_sum_mock.reset_mock()
    zip_directory_mock.reset_mock()
    write_hash_sum_mock.reset_mock()

    # Test situation when saved directory hash == current hash and old archive exists
    read_hash_sum_mock.return_value = self.DUMMY_HASH
    count_hash_sum_mock.return_value = self.DUMMY_HASH
    os_path_exists_mock.return_value = True
    os_path_isfile_mock.return_value = True
    resource_files_keeper.update_directory_archive(self.SOME_PATH)
    self.assertTrue(read_hash_sum_mock.called)
    self.assertTrue(count_hash_sum_mock.called)
    self.assertFalse(zip_directory_mock.called)
    self.assertFalse(write_hash_sum_mock.called)

    read_hash_sum_mock.reset_mock()
    count_hash_sum_mock.reset_mock()
    zip_directory_mock.reset_mock()
    write_hash_sum_mock.reset_mock()

    # Check that no saved hash file is created when zipping failed
    zip_directory_mock.side_effect = self.keeper_exc_side_effect
    read_hash_sum_mock.return_value = self.DUMMY_HASH
    count_hash_sum_mock.return_value = self.YA_HASH
    os_path_exists_mock.return_value = True
    try:
      resource_files_keeper.update_directory_archive(self.SOME_PATH)
      self.fail("KeeperException not thrown")
    except KeeperException:
      pass  # Expected
    except Exception as e:
      self.fail("Unexpected exception thrown:" + str(e))
    self.assertTrue(read_hash_sum_mock.called)
    self.assertTrue(count_hash_sum_mock.called)
    self.assertTrue(zip_directory_mock.called)
    self.assertFalse(write_hash_sum_mock.called)

    read_hash_sum_mock.reset_mock()
    count_hash_sum_mock.reset_mock()
    zip_directory_mock.reset_mock()
    write_hash_sum_mock.reset_mock()

    # Test nozip option
    read_hash_sum_mock.return_value = None
    count_hash_sum_mock.return_value = self.YA_HASH
    os_path_exists_mock.return_value = True
    resource_files_keeper = ResourceFilesKeeper(
      self.TEST_RESOURCES_DIR, self.SOME_PATH, nozip=True
    )
    resource_files_keeper.update_directory_archive(self.SOME_PATH)
    self.assertTrue(read_hash_sum_mock.called)
    self.assertTrue(count_hash_sum_mock.called)
    self.assertFalse(zip_directory_mock.called)
    self.assertTrue(write_hash_sum_mock.called)

    # Test empty directory
    read_hash_sum_mock.reset_mock()
    count_hash_sum_mock.reset_mock()
    zip_directory_mock.reset_mock()
    write_hash_sum_mock.reset_mock()

    # If the input directory is empty, then write_hash_sum() should not be called
    os_listdir_mock.return_value = []  # Empty dir
    zip_directory_mock.side_effect = None
    read_hash_sum_mock.return_value = None  # hash read from .hash file
    os_path_exists_mock.return_value = True
    resource_files_keeper = ResourceFilesKeeper(self.TEST_RESOURCES_DIR, self.SOME_PATH)
    resource_files_keeper.update_directory_archive(self.SOME_PATH)

    self.assertTrue(read_hash_sum_mock.called)
    self.assertTrue(count_hash_sum_mock.called)
    self.assertTrue(zip_directory_mock.called)
    self.assertFalse(write_hash_sum_mock.called)
    pass

    # Test no directory
    read_hash_sum_mock.reset_mock()
    count_hash_sum_mock.reset_mock()
    zip_directory_mock.reset_mock()
    write_hash_sum_mock.reset_mock()

    # If the input directory doesn't exist, then write_hash_sum() should not be called
    zip_directory_mock.side_effect = None
    read_hash_sum_mock.return_value = None  # hash read from .hash file
    os_path_exists_mock.return_value = False
    resource_files_keeper = ResourceFilesKeeper(self.TEST_RESOURCES_DIR, self.SOME_PATH)
    resource_files_keeper.update_directory_archive(self.SOME_PATH)

    self.assertTrue(read_hash_sum_mock.called)
    self.assertTrue(count_hash_sum_mock.called)
    self.assertTrue(zip_directory_mock.called)
    self.assertFalse(write_hash_sum_mock.called)
    pass

  def test_count_hash_sum(self):
    # Test normal flow
    resource_files_keeper = ResourceFilesKeeper(
      self.TEST_RESOURCES_DIR, self.DUMMY_UNCHANGEABLE_PACKAGE
    )
    test_dir = os.path.join(self.DUMMY_UNCHANGEABLE_PACKAGE)
    hash_sum = resource_files_keeper.count_hash_sum(test_dir)
    self.assertEqual(hash_sum, self.DUMMY_UNCHANGEABLE_PACKAGE_HASH)

    # Test exception handling
    with patch("builtins.open") as open_mock:
      open_mock.side_effect = self.exc_side_effect
      try:
        resource_files_keeper.count_hash_sum(test_dir)
        self.fail("KeeperException not thrown")
      except KeeperException:
        pass  # Expected
      except Exception as e:
        self.fail("Unexpected exception thrown:" + str(e))

  def test_read_hash_sum(self):
    resource_files_keeper = ResourceFilesKeeper(
      self.TEST_RESOURCES_DIR, self.DUMMY_UNCHANGEABLE_PACKAGE
    )
    hash_sum = resource_files_keeper.read_hash_sum(self.DUMMY_UNCHANGEABLE_PACKAGE)
    self.assertEqual(hash_sum, "dummy_hash")

    # Test exception handling
    # If file exists, should rethrow exception
    with patch("os.path.isfile") as isfile_mock:
      isfile_mock.return_value = True
      with patch("builtins.open") as open_mock:
        open_mock.side_effect = self.exc_side_effect
        try:
          resource_files_keeper.read_hash_sum("path-to-directory")
          self.fail("KeeperException not thrown")
        except KeeperException:
          pass  # Expected
        except Exception as e:
          self.fail("Unexpected exception thrown:" + str(e))

    # Test exception handling
    # If file does not exist, should ignore exception
    with patch("os.path.isfile") as isfile_mock:
      isfile_mock.return_value = False
      with patch("builtins.open") as open_mock:
        open_mock.side_effect = self.exc_side_effect
        res = resource_files_keeper.read_hash_sum("path-to-directory")
        self.assertEqual(res, None)
    pass

  def test_write_hash_sum(self):
    NEW_HASH = "new_hash"
    resource_files_keeper = ResourceFilesKeeper(
      self.TEST_RESOURCES_DIR, self.DUMMY_UNCHANGEABLE_PACKAGE
    )
    resource_files_keeper.write_hash_sum(self.DUMMY_UNCHANGEABLE_PACKAGE, NEW_HASH)
    hash_sum = resource_files_keeper.read_hash_sum(self.DUMMY_UNCHANGEABLE_PACKAGE)
    self.assertEqual(hash_sum, NEW_HASH)

    # Revert to previous value
    resource_files_keeper.write_hash_sum(
      self.DUMMY_UNCHANGEABLE_PACKAGE, self.DUMMY_HASH
    )
    hash_sum = resource_files_keeper.read_hash_sum(self.DUMMY_UNCHANGEABLE_PACKAGE)
    self.assertEqual(hash_sum, self.DUMMY_HASH)

    # Test exception handling
    with patch("builtins.open") as open_mock:
      open_mock.side_effect = self.exc_side_effect
      try:
        resource_files_keeper.write_hash_sum("path-to-directory", self.DUMMY_HASH)
        self.fail("KeeperException not thrown")
      except KeeperException:
        pass  # Expected
      except Exception as e:
        self.fail("Unexpected exception thrown:" + str(e))

  @patch("os.path.exists")
  def test_zip_directory(self, os_path_exists_mock):
    # Test normal flow
    os_path_exists_mock.return_value = True
    resource_files_keeper = ResourceFilesKeeper(
      self.TEST_RESOURCES_DIR, self.DUMMY_UNCHANGEABLE_PACKAGE
    )
    resource_files_keeper.zip_directory(self.DUMMY_UNCHANGEABLE_PACKAGE)
    arc_file = os.path.join(
      self.DUMMY_UNCHANGEABLE_PACKAGE, ResourceFilesKeeper.ARCHIVE_NAME
    )
    # Arc file should not be empty
    arc_size = os.path.getsize(arc_file)
    self.assertTrue(40000 < arc_size < 50000)
    # After creating zip, count hash sum of dir (should not change)
    hash_val = resource_files_keeper.count_hash_sum(self.DUMMY_UNCHANGEABLE_PACKAGE)
    self.assertEqual(hash_val, self.DUMMY_UNCHANGEABLE_PACKAGE_HASH)
    # Remove arc file
    os.unlink(arc_file)

    # Test exception handling
    with patch("os.path.join") as join_mock:
      join_mock.side_effect = self.exc_side_effect
      try:
        resource_files_keeper.zip_directory("path-to-directory")
        self.fail("KeeperException not thrown")
      except KeeperException:
        pass  # Expected
      except Exception as e:
        self.fail("Unexpected exception thrown:" + str(e))

    # Test skip zipping of an empty directory
    with patch("os.listdir") as os_listdir_mock:
      os_listdir_mock.return_value = False  # Empty dir
      try:
        skip_empty_directory = True
        resource_files_keeper.zip_directory("empty-to-directory", skip_empty_directory)
        self.assertTrue(os_listdir_mock.called)
      except Exception as e:
        self.fail("Unexpected exception thrown: " + str(e))
    pass

    # Test skip zipping of a missing directory
    with patch("os.listdir") as os_listdir_mock:
      os_path_exists_mock.return_value = False
      os_listdir_mock.return_value = False  # Empty dir
      try:
        skip_empty_directory = True
        resource_files_keeper.zip_directory("empty-to-directory", skip_empty_directory)
        self.assertTrue(os_path_exists_mock.called)
      except Exception as e:
        self.fail("Unexpected exception thrown: " + str(e))
    pass

  def test_is_ignored(self):
    resource_files_keeper = ResourceFilesKeeper(
      self.TEST_RESOURCES_DIR, self.DUMMY_UNCHANGEABLE_PACKAGE
    )
    self.assertTrue(resource_files_keeper.is_ignored(".hash"))
    self.assertTrue(resource_files_keeper.is_ignored("archive.zip"))
    self.assertTrue(resource_files_keeper.is_ignored("dummy.pyc"))
    self.assertFalse(resource_files_keeper.is_ignored("dummy.py"))
    self.assertFalse(resource_files_keeper.is_ignored("1.sh"))
    pass

  def exc_side_effect(self, *a):
    raise Exception("horrible_exc")

  def keeper_exc_side_effect(self, *a):
    raise KeeperException("horrible_keeper_exc")