scripts/fuzzing/test/cipd_mock.py

#!/usr/bin/env python
# Copyright 2019 The Fuchsia Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import test_env
from lib.cipd import Cipd
from lib.fuzzer import Fuzzer

from device_mock import MockDevice


class MockCipd(Cipd):

  def __init__(self):
    self.fuzzer = Fuzzer(MockDevice(), u'mock-package1', u'mock-target3')
    self.last = None
    super(MockCipd, self).__init__(self.fuzzer)

  def _exec(self, cmd, quiet=False):
    """Overrides Cipd._exec for testing."""
    self.last = self._bin + ' ' + ' '.join(cmd)