py/test/session_unittest.py

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


"""Unittests for test session and invocation."""

import logging
import os
import re
import shutil
import tempfile
import unittest

import mock
from six import assertRegex
from six.moves import xrange

import factory_common  # pylint: disable=unused-import
from cros.factory.test import session

UUID_RE = re.compile(r'^[a-f0-9]{8}-[a-f0-9]{4}-[a-f0-9]{4}-'
                     '[a-f0-9]{4}-[a-f0-9]{12}$')


class SessionTest(unittest.TestCase):
  """Unittest for session.py."""

  def setUp(self):
    for path in [
        session.INSTALLATION_ID_PATH,
        session.DEVICE_ID_PATH,
        session.INIT_COUNT_PATH]:
      try:
        os.unlink(path)
      except OSError:
        pass
    session._installation_id = None  # pylint: disable=protected-access
    session._device_id = None  # pylint: disable=protected-access

    self.tmp = tempfile.mkdtemp()

  def tearDown(self):
    shutil.rmtree(self.tmp)

  def testGetBootID(self):
    assertRegex(self, session.GetBootID(), UUID_RE)

  @mock.patch('session.file_utils.ReadFile', return_value='device_id\n')
  @mock.patch('os.path.exists', return_value=True)
  def testGetDeviceID(self, mock_exists, mock_read_file):
    self.assertEqual('device_id', session.GetDeviceID())
    # Make one more call to ensure the result is cached.
    self.assertEqual('device_id', session.GetDeviceID())
    mock_exists.assert_called_once_with(session.DEVICE_ID_PATH)
    mock_read_file.assert_called_once_with(session.DEVICE_ID_PATH)

  def testGetInstallationID(self):
    installation_id = session.GetInstallationID()
    assertRegex(self, installation_id, UUID_RE)

    # Remove installation_id and make sure we get the same thing
    # back again, re-reading it from disk
    session._installation_id = None  # pylint: disable=protected-access
    self.assertEqual(installation_id, session.GetInstallationID())

    # Remove the installation_id file; now we should get something
    # *different* back.
    session._installation_id = None  # pylint: disable=protected-access
    os.unlink(session.INSTALLATION_ID_PATH)
    session.GetInstallationID.InvalidateCache()
    self.assertNotEqual(installation_id, session.GetInstallationID())

  def testInitCount(self):
    for i in xrange(-1, 5):
      self.assertEqual(i, session.GetInitCount())
      session.IncrementInitCount()
      self.assertEqual(str(i + 1),
                       open(session.INIT_COUNT_PATH).read())


if __name__ == '__main__':
  logging.basicConfig(
      format=('[%(levelname)s] '
              ' %(threadName)s:%(lineno)d %(asctime)s.%(msecs)03d %(message)s'),
      level=logging.INFO,
      datefmt='%Y-%m-%d %H:%M:%S')
  unittest.main()