commit 5d13205e5b4b273c0419f9abf7e34fa496fceef6
author Andrew Lamb <andrewlamb@chromium.org> Fri May 29 14:49:06 2020 -0600
committer Commit Bot <commit-bot@chromium.org> Tue Jun 02 15:31:42 2020 +0000
tree 76ae068ebc1b9ad94d46b451128e8e808a21d78a
parent ff52aa5527361d1a37470c4fd642650f0c9d03a9

Add ConstraintSuite for model_sku.json.

- Start with simple check that "model" key exists.

BUG=chromium:1085429
TEST=CQ

Change-Id: Id70af4d4787a6f7eb7ad7efc4531015d2acf22a1
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/config/+/2222843
Commit-Queue: Andrew Lamb <andrewlamb@chromium.org>
Tested-by: Andrew Lamb <andrewlamb@chromium.org>
Reviewed-by: Sean McAllister <smcallis@google.com>
Reviewed-by: David Burger <dburger@chromium.org>
Reviewed-by: C Shapiro <shapiroc@chromium.org>

payload_utils/checker/io_utils_test.py

diff --git a/payload_utils/checker/io_utils_test.py b/payload_utils/checker/io_utils_test.py
index 0d2727d..35aea1a 100644
--- a/payload_utils/checker/io_utils_test.py
+++ b/payload_utils/checker/io_utils_test.py
@@ -3,7 +3,9 @@
 # found in the LICENSE file.
 """Tests for io_utils."""
 
+import json
 import os
+import pathlib
 import tempfile
 import unittest
 
@@ -31,6 +33,20 @@
     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)