py/probe/functions/flash_chip_unittest.py

#!/usr/bin/env python3
# Copyright 2016 The Chromium OS 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 subprocess
import unittest
from unittest import mock

from cros.factory.probe.functions import flash_chip


class FlashChipFunctionTest(unittest.TestCase):

  @mock.patch('cros.factory.utils.process_utils.CheckOutput',
              return_value='vendor="Google" name="Chip"')
  def testNormal(self, MockCheckOutput):
    expected = {'vendor': 'Google', 'name': 'Chip'}
    results = flash_chip.FlashChipFunction(chip='main')()
    self.assertEqual(results, [expected])
    MockCheckOutput.assert_called_with(
        ['flashrom', '-p', 'host', '--flash-name'])

  @mock.patch('cros.factory.utils.process_utils.CheckOutput',
              side_effect=subprocess.CalledProcessError(1, 'command'))
  def testNoOutput(self, MockCheckOutput):
    results = flash_chip.FlashChipFunction(chip='ec')()
    self.assertEqual(results, [])
    MockCheckOutput.assert_called_with(['flashrom', '-p', 'ec', '--flash-name'])


if __name__ == '__main__':
  unittest.main()