py/test/pytests/mlb_version_unittest.py

#!/usr/bin/python -Bu
# Copyright 2014 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.

"""Unit tests for mlb_version factory test."""

import mox
import unittest

import factory_common  # pylint: disable=unused-import
from cros.factory.device.board import DeviceBoard
from cros.factory.device.info import SystemInfo
from cros.factory.test.pytests import mlb_version
from cros.factory.test.rules import phase


class MLBVersionTestUnittest(unittest.TestCase):
  """Unit tests for mlb_version factory test."""

  def setUp(self):
    self.mox = mox.Mox()
    self.mox.StubOutWithMock(phase, 'GetPhase')

    class FakeArgs(object):
      """A fake factory test args object."""
      expected_version = None

    self.test = mlb_version.MLBVersionTest()
    self.test.dut = self.mox.CreateMock(DeviceBoard)
    self.test.dut.info = self.mox.CreateMock(SystemInfo)
    self.test.args = FakeArgs()

  def tearDown(self):
    self.mox.UnsetStubs()

  def testNoArgs(self):
    # Proto2b board in phase PROTO, EVT, PVT_DOGFOOD.
    phase.GetPhase().AndReturn(phase.PROTO)
    phase.GetPhase().AndReturn(phase.EVT)
    phase.GetPhase().AndReturn(phase.PVT_DOGFOOD)
    # PVT3 board in phase PVT_DOGFOOD, PVT.
    phase.GetPhase().AndReturn(phase.PVT_DOGFOOD)
    phase.GetPhase().AndReturn(phase.PVT)

    self.mox.ReplayAll()
    # Proto2b board in phase PROTO.

    # Proto2b board in phase PROTO. This should pass.
    self.test.dut.info.board_version = 'Proto2B'
    self.test.runTest()
    # Proto2b board in phase EVT. This should fail due to mismatch version
    # prefix.
    self.assertRaisesRegexp(
        AssertionError,
        (r'In phase EVT, expect board version to start with EVT, but got board '
         r'version Proto2B'),
        self.test.runTest)
    # Proto2b board in phase PVT_DOGFOOD. This should fail due to mismatch
    # version prefix.
    self.assertRaisesRegexp(
        AssertionError,
        (r'In phase PVT_DOGFOOD, expect board version to start with '
         r'\(PVT\|MP\), but got board version Proto2B'),
        self.test.runTest)
    self.test.dut.info.board_version = 'PVT3'
    # PVT3 board in phase PVT_DOGFOOD. This should pass.
    self.test.runTest()
    # PVT3 board in phase PVT. This should pass.
    self.test.runTest()

    self.mox.VerifyAll()

  def testWithArgs(self):
    # Expect to see board version 'Proto2B'.
    self.test.args.expected_version = 'Proto2B'

    self.mox.ReplayAll()

    # Test on Proto2b board. This should pass.
    self.test.dut.info.board_version = 'Proto2B'
    self.test.runTest()
    # Test on PVT3 board. This should fail with mismatched board version.
    self.test.dut.info.board_version = 'PVT3'
    self.assertRaisesRegexp(
        AssertionError,
        (r'Board version mismatch\. Expect to see board version Proto2B, but '
         r'the actual board version is PVT3'),
        self.test.runTest)

    self.mox.VerifyAll()


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