| # 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 json |
| import os |
| import pathlib |
| import tempfile |
| import unittest |
| |
| from google.protobuf import json_format |
| |
| from checker import io_utils |
| |
| from chromiumos.config.payload.config_bundle_pb2 import ConfigBundle |
| from chromiumos.config.api.program_pb2 import Program |
| |
| |
| class IoUtilsTest(unittest.TestCase): |
| """Tests for io_utils.""" |
| |
| def setUp(self): |
| self.config = ConfigBundle(program_list=[Program(name='TestProgram1')]) |
| repo_path = tempfile.mkdtemp() |
| |
| os.mkdir(os.path.join(repo_path, 'generated')) |
| |
| self.config_path = os.path.join(repo_path, 'generated', 'config.jsonproto') |
| json_output = json_format.MessageToJson( |
| self.config, sort_keys=True, use_integers_for_enums=True) |
| with open(self.config_path, 'w') as f: |
| print(json_output, file=f) |
| |
| self.factory_path = os.path.join(repo_path, 'factory') |
| os.makedirs(os.path.join(self.factory_path, 'generated')) |
| self.model_sku = {"model": {"a": 1}} |
| with open( |
| os.path.join(self.factory_path, 'generated', 'model_sku.json'), |
| 'w') as f: |
| json.dump(self.model_sku, f) |
| |
| def test_read_config(self): |
| """Tests the json proto can be read.""" |
| self.assertEqual(io_utils.read_config(self.config_path), self.config) |
| |
| def test_read_model_sku_json(self): |
| """Tests model_sku.json can be read.""" |
| self.assertDictEqual( |
| io_utils.read_model_sku_json(pathlib.Path(self.factory_path)), |
| self.model_sku) |