cbuildbot/topology_unittest.py

# -*- coding: utf-8 -*-
# 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.

"""Unit tests for topology module."""

from __future__ import print_function

import sys

from chromite.cbuildbot import topology
from chromite.lib import cros_test_lib


assert sys.version_info >= (3, 6), 'This module requires Python 3.6+'


class TopologyTest(cros_test_lib.TestCase):
  """Unit test of topology module."""

  def setUp(self):
    # Mutually isolate these tests and make them independent of
    # TOPOLOGY_DEFAULTS
    topology.topology = topology.LockedDefaultDict()

  def testNotFetched(self):
    with self.assertRaises(topology.LockedDictAccessException):
      topology.topology.get('/foo')


def FakeFetchTopology(keyvals=None):
  """Setup topology without the need for a DB

  args:
    keyvals: optional dictionary to populate topology
  """
  keyvals = keyvals if keyvals is not None else {}

  topology.FetchTopology()
  topology.topology.update(keyvals)
  topology.topology.unlock()


class FakeFetchTopologyTest(cros_test_lib.TestCase):
  """Test FakeFetchTopologyunittest helper function"""

  def setUp(self):
    _resetTopology()

  def testFakeTopology(self):
    data = {1:'one', 2:'two', 3:'three'}
    FakeFetchTopology(data)
    self.assertDictContainsSubset(data, topology.topology)

  def testFakeTopologyEmpty(self):
    FakeFetchTopology()
    # pylint: disable=protected-access
    self.assertFalse(topology.topology._locked)


def _resetTopology():
  """Remove effects of unittests on topology"""
  topology.topology = topology.LockedDefaultDict()
  topology.topology.update(topology.TOPOLOGY_DEFAULTS)