build_utils_unittest.py

# Copyright 2020 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.

"""Module for build_utils unittests."""

import unittest

import build_utils


class BuildUtilsTestCase(unittest.TestCase):

  def testParseNormalModelName(self):
    full_model = 'octopus_mimrock'
    board = 'octopus'
    model = 'mimrock'
    self.assertEqual(build_utils.parse_full_model(full_model), (board, model))

  def testParseModelNameWithFamily(self):
    full_model = 'veyron_minnie_minnie'
    board = 'veyron_minnie'
    model = 'minnie'
    self.assertEqual(build_utils.parse_full_model(full_model), (board, model))

  def testParseInvalidFullModelRaisesException(self):
    invalid_full_models = ['atlas',
                           'veyron_minnie_veyron_minnie',
                           'too_long_name_a_name']
    for invalid_model in invalid_full_models:
      self.assertRaises(ValueError, build_utils.parse_full_model, invalid_model)

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