blob: 5177f8c031ee195549afc2af9e2310f07531cc73 [file] [log] [blame]
# 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.
"""Tests for the blueprint library."""
from __future__ import print_function
from chromite.lib import cros_test_lib
class BlueprintLibTest(cros_test_lib.MockTempDirTestCase,
cros_test_lib.BlueprintTestCase,
cros_test_lib.WorkspaceTestCase):
"""Unittest for blueprint_lib.py"""
def setUp(self):
self.SetupFakeWorkspace()
def testBlueprint(self):
bricks = ['//foo', '//bar', '//baz']
blueprint = self.CreateBlueprint(bricks=bricks, bsp='//bsp',
main_package='virtual/target-os')
self.assertEqual(blueprint.GetBricks(), bricks)
self.assertEqual(blueprint.GetBSP(), '//bsp')
self.assertEqual(blueprint.GetMainPackage(), 'virtual/target-os')
def testBlueprintNoBricks(self):
blueprint = self.CreateBlueprint(bsp='//bsp2',
main_package='virtual/target-os-dev')
self.assertEqual(blueprint.GetBricks(), [])
self.assertEqual(blueprint.GetBSP(), '//bsp2')
self.assertEqual(blueprint.GetMainPackage(), 'virtual/target-os-dev')