py/device/info_unittest.py

#!/usr/bin/env python2
#
# Copyright 2015 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.


"""Unittest for SystemInfo."""


import logging
import unittest

import mox

import factory_common  # pylint: disable=unused-import
from cros.factory.device import info as info_module
from cros.factory.device import partitions
from cros.factory.device import types

MOCK_RELEASE_IMAGE_LSB_RELEASE = ('GOOGLE_RELEASE=5264.0.0\n'
                                  'CHROMEOS_RELEASE_TRACK=canary-channel\n')


class SystemInfoTest(unittest.TestCase):
  """Unittest for SystemInfo."""

  def setUp(self):
    self.mox = mox.Mox()

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

  def runTest(self):

    dut = self.mox.CreateMock(types.DeviceInterface)
    dut.partitions = self.mox.CreateMock(partitions.Partitions)
    dut.partitions.RELEASE_ROOTFS = self.mox.CreateMock(
        partitions.DiskPartition)
    dut.partitions.RELEASE_ROOTFS.path = '/dev/sda5'
    self.mox.StubOutWithMock(info_module, 'MountDeviceAndReadFile')
    info_module.MountDeviceAndReadFile(
        '/dev/sda5', '/etc/lsb-release', dut=dut).AndReturn(
            MOCK_RELEASE_IMAGE_LSB_RELEASE)

    self.mox.ReplayAll()

    info = info_module.SystemInfo(dut)
    self.assertEqual('5264.0.0', info.release_image_version)
    self.assertEqual('canary-channel', info.release_image_channel)
    # The cached release image version will be used in the second time.
    self.assertEqual('5264.0.0', info.release_image_version)
    self.assertEqual('canary-channel', info.release_image_channel)

    self.mox.VerifyAll()

if __name__ == '__main__':
  logging.basicConfig(
      format='%(asctime)s:%(filename)s:%(lineno)d:%(levelname)s:%(message)s',
      level=logging.DEBUG)
  unittest.main()