lib/chroot_lib_unittest.py

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

"""chroot_lib unit tests."""

from __future__ import print_function

import os
import sys

from chromite.lib import chroot_lib
from chromite.lib import cros_test_lib
from chromite.lib import osutils


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


class ChrootTest(cros_test_lib.TempDirTestCase):
  """Chroot class tests."""

  def testGetEnterArgsEmpty(self):
    """Test empty instance behavior."""
    chroot = chroot_lib.Chroot()
    self.assertFalse(chroot.get_enter_args())

  def testGetEnterArgsAll(self):
    """Test complete instance behavior."""
    path = '/chroot/path'
    cache_dir = '/cache/dir'
    chrome_root = '/chrome/root'
    expected = ['--chroot', path, '--cache-dir', cache_dir,
                '--chrome-root', chrome_root]
    chroot = chroot_lib.Chroot(path=path, cache_dir=cache_dir,
                               chrome_root=chrome_root)

    self.assertCountEqual(expected, chroot.get_enter_args())

  def testEnv(self):
    """Test the env handling."""
    env = {'VAR': 'val'}
    chroot = chroot_lib.Chroot(env=env)
    self.assertEqual(env, chroot.env)

  def testTempdir(self):
    """Test the tempdir functionality."""
    chroot = chroot_lib.Chroot(path=self.tempdir)
    osutils.SafeMakedirs(chroot.tmp)

    self.assertEqual(os.path.join(self.tempdir, 'tmp'), chroot.tmp)

    with chroot.tempdir() as tempdir:
      self.assertStartsWith(tempdir, chroot.tmp)

    self.assertNotExists(tempdir)

  def testExists(self):
    """Test chroot exists."""
    chroot = chroot_lib.Chroot(self.tempdir)
    self.assertTrue(chroot.exists())

    chroot = chroot_lib.Chroot(os.path.join(self.tempdir, 'DOES_NOT_EXIST'))
    self.assertFalse(chroot.exists())

  def testChrootPath(self):
    """Test chroot_path functionality."""
    chroot = chroot_lib.Chroot(self.tempdir)
    path1 = os.path.join(self.tempdir, 'some/path')
    path2 = '/bad/path'

    # Make sure that it gives an absolute path inside the chroot.
    self.assertEqual('/some/path', chroot.chroot_path(path1))
    # Make sure it raises an error for paths not inside the chroot.
    self.assertRaises(chroot_lib.ChrootError, chroot.chroot_path, path2)

  def testFullPath(self):
    """Test full_path functionality."""
    chroot = chroot_lib.Chroot(self.tempdir)
    path1 = 'some/path'
    path2 = '/some/path'

    # Make sure it's building out the path in the chroot.
    self.assertEqual(os.path.join(self.tempdir, path1), chroot.full_path(path1))
    # Make sure it can handle absolute paths.
    self.assertEqual(chroot.full_path(path1), chroot.full_path(path2))

  def testFullPathWithExtraArgs(self):
    """Test full_path functionality with extra args passed."""
    chroot = chroot_lib.Chroot(self.tempdir)
    path1 = 'some/path'
    self.assertEqual(os.path.join(self.tempdir, 'some/path/abc/def/g/h/i'),
                     chroot.full_path(path1, '/abc', 'def', '/g/h/i'))

  def testHasPathSuccess(self):
    """Test has path for a valid path."""
    path = 'some/file.txt'
    tempdir_path = os.path.join(self.tempdir, path)
    osutils.Touch(tempdir_path, makedirs=True)

    chroot = chroot_lib.Chroot(self.tempdir)
    self.assertTrue(chroot.has_path(path))

  def testHasPathInvalidPath(self):
    """Test has path for a non-existent path."""
    chroot = chroot_lib.Chroot(self.tempdir)
    self.assertFalse(chroot.has_path('/does/not/exist'))

  def testHasPathVariadic(self):
    """Test multiple args to has path."""
    path = ['some', 'file.txt']
    tempdir_path = os.path.join(self.tempdir, *path)
    osutils.Touch(tempdir_path, makedirs=True)

    chroot = chroot_lib.Chroot(self.tempdir)
    self.assertTrue(chroot.has_path(*path))

  def testEqual(self):
    """__eq__ method sanity check."""
    path = '/chroot/path'
    cache_dir = '/cache/dir'
    chrome_root = '/chrome/root'
    env = {'USE': 'useflag',
           'FEATURES': 'feature'}
    chroot1 = chroot_lib.Chroot(path=path, cache_dir=cache_dir,
                                chrome_root=chrome_root, env=env)
    chroot2 = chroot_lib.Chroot(path=path, cache_dir=cache_dir,
                                chrome_root=chrome_root, env=env)
    chroot3 = chroot_lib.Chroot(path=path)
    chroot4 = chroot_lib.Chroot(path=path)

    self.assertEqual(chroot1, chroot2)
    self.assertEqual(chroot3, chroot4)
    self.assertNotEqual(chroot1, chroot3)