2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
6 * ================================================================================
7 * Copyright (C) 2017 Amdocs
8 * =============================================================================
9 * Licensed under the Apache License, Version 2.0 (the "License");
10 * you may not use this file except in compliance with the License.
11 * You may obtain a copy of the License at
13 * http://www.apache.org/licenses/LICENSE-2.0
15 * Unless required by applicable law or agreed to in writing, software
16 * distributed under the License is distributed on an "AS IS" BASIS,
17 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
18 * See the License for the specific language governing permissions and
19 * limitations under the License.
21 * ECOMP is a trademark and service mark of AT&T Intellectual Property.
22 * ============LICENSE_END=========================================================
28 * Class to emulate responses from the Saltstack Server that is compliant with the APP-C Saltstack Server
29 * Interface. Used for jUnit tests to verify code is working. In tests it can be used
30 * as a replacement for methods from ConnectionBuilder class
33 package org.onap.ccsdk.sli.adaptors.saltstack.model;
35 import com.att.eelf.configuration.EELFLogger;
36 import com.att.eelf.configuration.EELFManager;
37 import org.apache.commons.lang.StringUtils;
38 import org.json.JSONObject;
41 import java.util.regex.Matcher;
42 import java.util.regex.Pattern;
44 public class SaltstackServerEmulator {
46 private static final String SALTSTATE_FILE_NAME = "fileName";
47 private static final String STATUS_CODE = "StatusCode";
48 private static final String STATUS_MESSAGE = "StatusMessage";
49 private final EELFLogger logger = EELFManager.getInstance().getLogger(SaltstackServerEmulator.class);
50 private String saltStateName = "test_saltState.yaml";
53 * Method that emulates the response from an Saltstack Server
54 * when presented with a request to execute a saltState
55 * Returns an saltstack object result. The response code is always the ssh code 200 (i.e connection successful)
56 * payload is json string as would be sent back by Saltstack Server
58 public SaltstackResult MockReqExec(Map<String, String> params) {
59 SaltstackResult result = new SaltstackResult();
62 if (params.get("Test") == "fail") {
63 result = rejectRequest(result, "Mocked: Fail");
65 String fileName = params.get(SALTSTATE_FILE_NAME);
67 result = acceptRequest(result, "");
69 result = acceptRequest(result, fileName);
71 } catch (Exception e) {
72 logger.error("JSONException caught", e);
73 rejectRequest(result, e.getMessage());
79 * Method to emulate response from an Saltstack
80 * Server when presented with a GET request
81 * Returns an saltstack object result. The response code is always the ssh code 200 (i.e connection successful)
82 * payload is json string as would be sent back by Saltstack Server
84 public SaltstackResult Execute(String agentUrl) {
86 Pattern pattern = Pattern.compile(".*?\\?Id=(.*?)&Type.*");
87 Matcher matcher = pattern.matcher(agentUrl);
88 String id = StringUtils.EMPTY;
89 String vmAddress = "192.168.1.10";
92 id = matcher.group(1);
95 SaltstackResult getResult = new SaltstackResult();
97 JSONObject response = new JSONObject();
98 response.put(STATUS_CODE, 200);
99 response.put(STATUS_MESSAGE, "FINISHED");
101 JSONObject results = new JSONObject();
103 JSONObject vmResults = new JSONObject();
104 vmResults.put(STATUS_CODE, 200);
105 vmResults.put(STATUS_MESSAGE, "SUCCESS");
106 vmResults.put("Id", id);
107 results.put(vmAddress, vmResults);
109 response.put("Results", results);
111 getResult.setStatusCode(200);
112 getResult.setStatusMessage(response.toString());
117 private SaltstackResult rejectRequest(SaltstackResult result, String Message) {
118 result.setStatusCode(SaltstackResultCodes.REJECTED.getValue());
119 result.setStatusMessage("Rejected");
123 private SaltstackResult acceptRequest(SaltstackResult result, String fileName) {
124 result.setStatusCode(SaltstackResultCodes.SUCCESS.getValue());
125 result.setStatusMessage("Success");
126 result.setOutputFileName(fileName);