108a3c1ee4
This PS replaces deprecared module pkg_resources, also fixes the schema validation by adding specific schema draft to choose in order to prevent the processor to fall back to use the latest draft that may potentially cause issues. Also switched to quay.io/airshipit for base ubuntu image Change-Id: I687ef267ee3b027e80815e8852c8edcab5b5b727
103 lines
3.2 KiB
Python
103 lines
3.2 KiB
Python
# Copyright 2018 AT&T Intellectual Property. All other rights reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
"""Test Nodes API"""
|
|
from falcon import testing
|
|
|
|
import pytest
|
|
import json
|
|
import logging
|
|
|
|
from drydock_provisioner import policy
|
|
from drydock_provisioner.control.api import start_api
|
|
import drydock_provisioner.objects as objects
|
|
|
|
import falcon
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
class TestNodesApiUnit(object):
|
|
|
|
def test_post_nodes_resp(self, input_files, falcontest,
|
|
mock_process_node_filter):
|
|
|
|
input_file = input_files.join("deckhand_fullsite.yaml")
|
|
design_ref = "file://%s" % str(input_file)
|
|
|
|
url = '/api/v1.0/nodefilter'
|
|
hdr = self.get_standard_header()
|
|
body = {
|
|
'node_filter': 'filters',
|
|
'design_ref': design_ref,
|
|
}
|
|
|
|
result = falcontest.simulate_post(url,
|
|
headers=hdr,
|
|
body=json.dumps(body))
|
|
|
|
LOG.debug(result.text)
|
|
assert result.status == falcon.HTTP_200
|
|
assert result.text.count('n1') == 1
|
|
assert result.text.count('n2') == 1
|
|
|
|
def test_input_error(self, falcontest):
|
|
url = '/api/v1.0/nodefilter'
|
|
hdr = self.get_standard_header()
|
|
body = {}
|
|
|
|
result = falcontest.simulate_post(url,
|
|
headers=hdr,
|
|
body=json.dumps(body))
|
|
|
|
LOG.debug(result.text)
|
|
assert result.status == falcon.HTTP_400
|
|
|
|
@pytest.fixture()
|
|
def falcontest(self, drydock_state, deckhand_ingester,
|
|
deckhand_orchestrator, mock_get_build_data):
|
|
"""Create a test harness for the Falcon API framework."""
|
|
policy.policy_engine = policy.DrydockPolicy()
|
|
policy.policy_engine.register_policy()
|
|
|
|
return testing.TestClient(
|
|
start_api(state_manager=drydock_state,
|
|
ingester=deckhand_ingester,
|
|
orchestrator=deckhand_orchestrator))
|
|
|
|
def get_standard_header(self):
|
|
hdr = {
|
|
'Content-Type': 'application/json',
|
|
'X-IDENTITY-STATUS': 'Confirmed',
|
|
'X-USER-NAME': 'Test',
|
|
'X-ROLES': 'admin'
|
|
}
|
|
return hdr
|
|
|
|
|
|
@pytest.fixture()
|
|
def mock_process_node_filter(mocker, deckhand_orchestrator):
|
|
n1 = objects.BaremetalNode()
|
|
n1.name = 'n1'
|
|
n1.site = 'test1'
|
|
n2 = objects.BaremetalNode()
|
|
n2.name = 'n2'
|
|
n2.site = 'test2'
|
|
mock_results = [n1, n2]
|
|
|
|
mocker.patch(
|
|
'drydock_provisioner.orchestrator.orchestrator.Orchestrator.process_node_filter',
|
|
return_value=mock_results
|
|
)
|
|
yield
|