#!/usr/bin/env python3
# Copyright 2020 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""The unittest of factory_config_proto_converter.py"""

import os
import subprocess
import tempfile
import unittest

import factory_config_proto_converter

from chromiumos.config.test import fake_config

THIS_DIR = os.path.dirname(__file__)

PROGRAM_CONFIG_FILE = fake_config.FAKE_PROGRAM_CONFIG
PROJECT_CONFIG_FILE = fake_config.FAKE_PROJECT_CONFIG


def fakeConfig():
  return factory_config_proto_converter._MergeConfigs([
      factory_config_proto_converter._ReadConfig(PROGRAM_CONFIG_FILE),
      factory_config_proto_converter._ReadConfig(PROJECT_CONFIG_FILE)
  ])


class MainTest(unittest.TestCase):

  def testFullTransform(self):
    with tempfile.TemporaryDirectory() as tempdir:
      output_file = os.path.join(tempdir, 'output')
      factory_config_proto_converter.Main(
          project_configs=[PROJECT_CONFIG_FILE],
          program_config=PROGRAM_CONFIG_FILE,
          output=output_file,
      )

      expected_file = os.path.join(THIS_DIR, 'test_data/model_sku.json')
      changed = subprocess.run(['diff', expected_file, output_file
                               ]).returncode != 0

      regen_cmd = ('To regenerate the expected output, run:\n'
                   '\tcd payload_utils && '
                   'python3 -m factory_config_proto_converter '
                   '-c %s '
                   '-p %s '
                   '-o %s ' %
                   (PROJECT_CONFIG_FILE, PROGRAM_CONFIG_FILE, expected_file))

      if changed:
        print(regen_cmd)
        self.fail('Fake project transform does not match')


if __name__ == '__main__':
  unittest.main(module=__name__)