scripts/cros_list_overlays_unittest.py

# -*- coding: utf-8 -*-
# Copyright 2014 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 cros_list_overlays.py"""

from __future__ import print_function

import sys

from chromite.lib import cros_test_lib
from chromite.lib import portage_util
from chromite.scripts import cros_list_overlays


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


class ListOverlaysTest(cros_test_lib.MockTestCase):
  """Tests for main()"""

  def setUp(self):
    self.find_mock = self.PatchObject(portage_util, 'FindOverlays')

  def testSmoke(self):
    """Basic sanity check"""
    cros_list_overlays.main([])

  def testAll(self):
    """Verify --all returns a lot."""
    cros_list_overlays.main(['--all'])

  def testBoard(self):
    """Check --board handling."""
    cros_list_overlays.main(['--board', 'eve'])