# Copyright 2013 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

from future import Future
from patcher import Patcher

class TestPatcher(Patcher):
  def __init__(self, version, patched_files, patch_data, assert_binary=False):
    self._version = version
    self._patched_files = patched_files
    self._patch_data = patch_data
    self._assert_binary = assert_binary

    self.get_version_count = 0
    self.get_patched_files_count = 0
    self.apply_count = 0

  def GetVersion(self):
    self.get_version_count += 1
    return self._version

  def GetPatchedFiles(self, version=None):
    self.get_patched_files_count += 1
    return self._patched_files

  def Apply(self, paths, file_system, binary, version=None):
    if self._assert_binary:
      assert binary
    self.apply_count += 1
    try:
      return Future(value=dict((path, self._patch_data[path])
                               for path in paths))
    except KeyError:
      raise FileNotFoundError('One of %s is deleted in the patch.' % paths)

  def GetIdentity(self):
    return self.__class__.__name__