dcae-cli/dcae_cli/util/tests/test_config.py

   1 # ============LICENSE_START=======================================================
   2 # org.onap.dcae
   3 # ================================================================================
   4 # Copyright (c) 2017 AT&T Intellectual Property. All rights reserved.
   5 # ================================================================================
   6 # Licensed under the Apache License, Version 2.0 (the "License");
   7 # you may not use this file except in compliance with the License.
   8 # You may obtain a copy of the License at
   9 #
  10 #      http://www.apache.org/licenses/LICENSE-2.0
  11 #
  12 # Unless required by applicable law or agreed to in writing, software
  13 # distributed under the License is distributed on an "AS IS" BASIS,
  14 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15 # See the License for the specific language governing permissions and
  16 # limitations under the License.
  17 # ============LICENSE_END=========================================================
  18 #
  19 # ECOMP is a trademark and service mark of AT&T Intellectual Property.
  20
  21 # -*- coding: utf-8 -*-
  22 """
  23 Tests the config functionality
  24 """
  25 import os, json
  26 from functools import partial
  27 from mock import patch
  28
  29 import pytest
  30 import click
  31
  32 import dcae_cli
  33 from dcae_cli.util import config, write_pref
  34 from dcae_cli.util.config import get_app_dir, get_config, get_config_path
  35
  36
  37 def test_no_config(monkeypatch, tmpdir):
  38     '''Tests the creation and initialization of a config on a clean install'''
  39     monkeypatch.setattr(click, "get_app_dir", lambda app: str(tmpdir.realpath()))
  40
  41     mock_config = {'user': 'mock-user'}
  42
  43     config_file = tmpdir.join("config.json")
  44     config_file.write(json.dumps(mock_config))
  45
  46     assert get_config() == mock_config
  47
  48
  49 def test_init_config_user(monkeypatch):
  50     good_case = "abc123"
  51     values = [ good_case, "d-e-f", "g*h*i", "j k l" ]
  52
  53     def fake_input(values, message, type="red"):
  54         return values.pop()
  55
  56     monkeypatch.setattr(click, 'prompt', partial(fake_input, values))
  57     assert config._init_config_user() == good_case
  58
  59
  60 def test_init_config(monkeypatch):
  61     monkeypatch.setattr(config, '_init_config_user', lambda: "bigmama")
  62     monkeypatch.setattr(config, '_init_config_server_url',
  63             lambda: "http://some-nexus-in-the-sky.com")
  64     monkeypatch.setattr(dcae_cli.util, 'fetch_file_from_web',
  65             lambda server_url, path: { "db_url": "conn" })
  66     monkeypatch.setattr("dcae_cli._version.__version__", "2.X.X")
  67
  68     expected = {'cli_version': '2.X.X', 'user': 'bigmama', 'db_url': 'conn',
  69             'server_url': 'http://some-nexus-in-the-sky.com'}
  70     assert expected == config._init_config()
  71
  72     # Test using of db fallback
  73
  74     monkeypatch.setattr(dcae_cli.util, 'fetch_file_from_web',
  75             lambda server_url, path: { "db_url": "" })
  76
  77     assert "sqlite" in config._init_config()["db_url"]
  78
  79     monkeypatch.setattr(dcae_cli.util, 'fetch_file_from_web',
  80             lambda server_url, path: {})
  81
  82     assert "sqlite" in config._init_config()["db_url"]
  83
  84     # Simulate error trying to fetch
  85
  86     def fetch_simulate_error(server_url, path):
  87         raise RuntimeError("Simulated error")
  88
  89     monkeypatch.setattr(dcae_cli.util, 'fetch_file_from_web',
  90             fetch_simulate_error)
  91
  92     with pytest.raises(config.ConfigurationInitError):
  93         config._init_config()
  94
  95
  96 def test_should_force_reinit():
  97     bad_config = {}
  98     assert config.should_force_reinit(bad_config) == True
  99
 100     old_config = { "cli_version": "1.0.0" }
 101     assert config.should_force_reinit(old_config) == True
 102
 103     uptodate_config = { "cli_version": "2.0.0" }
 104     assert config.should_force_reinit(uptodate_config) == False
 105
 106
 107 def test_reinit_config(monkeypatch, tmpdir):
 108     monkeypatch.setattr(click, "get_app_dir", lambda app: str(tmpdir.realpath()))
 109
 110     new_config = { "user": "ninny", "db_url": "some-db" }
 111
 112     def init():
 113         return new_config
 114
 115     assert config._reinit_config(init) == new_config
 116
 117     old_config = { "user": "super", "db_url": "other-db", "hidden": "yo" }
 118     write_pref(old_config, get_config_path())
 119
 120     new_config["hidden"] = "yo"
 121     assert config._reinit_config(init) == new_config
 122
 123
 124 if __name__ == '__main__':
 125     '''Test area'''
 126     pytest.main([__file__, ])