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(dcae_cli.util, 'fetch_file_from_nexus',
  63             lambda path: { "db_url": "conn" })
  64     monkeypatch.setattr("dcae_cli._version.__version__", "2.X.X")
  65
  66     expected = {'cli_version': '2.X.X', 'user': 'bigmama', 'db_url': 'conn'}
  67     assert expected == config._init_config()
  68
  69     # Test using of db fallback
  70
  71     monkeypatch.setattr(dcae_cli.util, 'fetch_file_from_nexus',
  72             lambda path: { "db_url": "" })
  73
  74     assert "sqlite" in config._init_config()["db_url"]
  75
  76     monkeypatch.setattr(dcae_cli.util, 'fetch_file_from_nexus',
  77             lambda path: {})
  78
  79     assert "sqlite" in config._init_config()["db_url"]
  80
  81     # Simulate error trying to fetch
  82
  83     def fetch_simulate_error(path):
  84         raise RuntimeError("Simulated error")
  85
  86     monkeypatch.setattr(dcae_cli.util, 'fetch_file_from_nexus',
  87             fetch_simulate_error)
  88
  89     with pytest.raises(config.ConfigurationInitError):
  90         config._init_config()
  91
  92
  93 def test_should_force_reinit():
  94     bad_config = {}
  95     assert config.should_force_reinit(bad_config) == True
  96
  97     old_config = { "cli_version": "1.0.0" }
  98     assert config.should_force_reinit(old_config) == True
  99
 100     uptodate_config = { "cli_version": "2.0.0" }
 101     assert config.should_force_reinit(uptodate_config) == False
 102
 103
 104 def test_reinit_config(monkeypatch, tmpdir):
 105     monkeypatch.setattr(click, "get_app_dir", lambda app: str(tmpdir.realpath()))
 106
 107     new_config = { "user": "ninny", "db_url": "some-db" }
 108
 109     def init():
 110         return new_config
 111
 112     assert config._reinit_config(init) == new_config
 113
 114     old_config = { "user": "super", "db_url": "other-db", "hidden": "yo" }
 115     write_pref(old_config, get_config_path())
 116
 117     new_config["hidden"] = "yo"
 118     assert config._reinit_config(init) == new_config
 119
 120
 121 if __name__ == '__main__':
 122     '''Test area'''
 123     pytest.main([__file__, ])