payload_utils/checker/io_utils_test.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.
"""Tests for io_utils."""

import os
import tempfile
import unittest

from checker import io_utils

from bindings.api.config_bundle_pb2 import ConfigBundle
from bindings.api.program_pb2 import ProgramList, Program


class IoUtilsTest(unittest.TestCase):
  """Tests for io_utils."""

  def setUp(self):
    self.config = ConfigBundle(
        programs=ProgramList(value=[Program(name='TestProgram1')]))
    self.repo_path = tempfile.mkdtemp()

    os.mkdir(os.path.join(self.repo_path, 'generated'))

    self.config_path = os.path.join(self.repo_path, 'generated',
                                    'config.binaryproto')
    with open(self.config_path, 'wb') as f:
      f.write(self.config.SerializeToString())

  def test_read_config(self):
    """Tests the binary proto can be read."""
    self.assertEqual(io_utils.read_config(self.config_path), self.config)

  def test_read_config_repo_path(self):
    """Tests that the binary proto can be read when a repo path is passed."""
    with self.assertWarnsRegex(FutureWarning,
                               'Passing a path to a repo root is deprecated'):
      self.assertEqual(io_utils.read_config(self.repo_path), self.config)