appengine/swarming/server/bot_archive_test.py

#!/usr/bin/env vpython
# Copyright 2017 The LUCI Authors. All rights reserved.
# Use of this source code is governed under the Apache License, Version 2.0
# that can be found in the LICENSE file.

import json
import os
import unittest

# Import before test_env, to confirm it doesn't depend on GAE.
import bot_archive

import test_env
test_env.setup_test_env()

from proto.config import config_pb2


ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))


_EXPECTED_CONFIG_KEYS = {
    'enable_ts_monitoring',
    'server',
    'server_version',
}


def _read_config():
  config_path = os.path.join(ROOT_DIR, 'swarming_bot', 'config', 'config.json')
  with open(config_path, 'rb') as f:
    return json.load(f) or {}


class Test(unittest.TestCase):
  def test_file(self):
    self.assertEqual(_EXPECTED_CONFIG_KEYS, set(_read_config()))

  def test_make(self):
    settings = config_pb2.SettingsCfg()
    config = json.loads(
        bot_archive._make_config_json('host', 'host_version', settings))
    self.assertEqual(_EXPECTED_CONFIG_KEYS, set(config))


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