aboutsummaryrefslogtreecommitdiffstats
path: root/moon_manager/tests/unit_python/conftest.py
blob: 702f3c19fe89a833e7ab1e7188e758fbdc886a5e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
# Software Name: MOON

# Version: 5.4

# SPDX-FileCopyrightText: Copyright (c) 2018-2020 Orange and its contributors
# SPDX-License-Identifier: Apache-2.0

# This software is distributed under the 'Apache License 2.0',
# the text of which is available at 'http://www.apache.org/licenses/LICENSE-2.0.txt'
# or see the "LICENSE" file for more details.


import os
import pytest
import requests_mock
import yaml
import mock_keystone
import mock_nova
import mock_engine
import mock_slaves


__CONF = """
database:
  url: sqlite:////tmp/database_test.db
  driver: moon_manager.plugins.sql
  migration_dir: moon_manager.api.db.migrations

management:
  url: http://127.0.0.1:8000
  user: admin
  password: admin
  token_file: db.json

orchestration:
  driver: moon_manager.plugins.pyorchestrator
  connection: local
  slaves:
    port: 10000...10100
  pipelines:
    port: 20000...20100

information:
  user: admin
  password: p4ssw0rd
  domain: default
  project: admin
  check_token: false
  certificate: false
  url: http://keystone:5000/v3
  subjects:
    drivers:
      moon_manager.plugins.moon_keystone_plugin:
        url: http://keystone:5000/v3
  objects:
    drivers:
      moon_manager.plugins.moon_nova_plugin:
        url: http://keystone:5000/compute/v2.1

  global_attrs:
    driver: moon_manager.plugins.global_attrs
    attributes:
      mode:
        values:
          - build
          - run
        default: run
        url: file:/tmp/mode
        #url: https://127.0.0.1:8080/mode
        #url: mysql+pymysql://moon:p4sswOrd1@db/moon_mode
        #url: sqlite:////tmp/database.db
        #url: driver://moon_manager.plugins.my_plugin

plugins:
  directory: /var/moon/plugins

components:
  manager:
    port: 8080
    bind: 0.0.0.0
    hostname: manager

logging:
  version: 1

  formatters:
    brief:
      format: "%(levelname)s %(name)s %(message)-30s"
    custom:
      format: "%(asctime)-15s %(levelname)s %(name)s %(message)s"

  handlers:
    console:
      class : logging.StreamHandler
      formatter: custom
      level   : INFO
      stream  : ext://sys.stdout
    file:
      class : logging.handlers.RotatingFileHandler
      formatter: custom
      level   : DEBUG
      filename: /tmp/moon.log
      maxBytes: 1048576
      backupCount: 3

  loggers:
    moon:
      level: DEBUG
      handlers: [console, file]
      propagate: no

  root:
    level: ERROR
    handlers: [console]
"""


@pytest.fixture(autouse=True)
def no_requests(monkeypatch):
    """ Modify the response from Requests module
    """
    global manager_api_key
    with requests_mock.Mocker(real_http=True) as m:
        try:
            os.remove("/tmp/database_test.db")
        except FileNotFoundError:
            pass
        try:
            os.remove("/tmp/moon.pwd")
        except FileNotFoundError:
            pass
        print("Configure...")
        from moon_manager.api.configuration import init_database, set_configuration
        set_configuration(yaml.safe_load(__CONF))
        print("Create a new user")
        from moon_utilities.auth_functions import add_user, init_db, get_api_key_for_user
        init_db()
        try:
            user = add_user("admin", "admin")
            manager_api_key = user["api_key"]
        except KeyError:
            print("User already exists")
            manager_api_key = get_api_key_for_user("admin")
        print("Initialize the database")
        init_database()
        from moon_manager import db_driver, orchestration_driver

        db_driver.init()
        orchestration_driver.init()

        mock_keystone.register_keystone(m)
        mock_nova.register_nova(m)
        mock_engine.register_engine(m)
        mock_slaves.register_slaves(m)

        from moon_manager.pip_driver import InformationManager
        for category in InformationManager:
            for manager in InformationManager[category]:
                manager.set_auth()

        yield m
        for category in InformationManager:
            for manager in InformationManager[category]:
                manager.unset_auth()