com.seleniumtests.reporter.logger.TestMessage Maven / Gradle / Ivy

/**
 * Orignal work: Copyright 2015 www.seleniumtests.com
 * Modified work: Copyright 2016 www.infotel.com
 * 				Copyright 2017-2019 B.Hecquet
 *
 * 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.
 */
package com.seleniumtests.reporter.logger;

import java.util.ArrayList;

import org.json.JSONObject;

public class TestMessage extends TestAction {
	
	private MessageType messageType;


	public enum MessageType {
		ERROR,
		WARNING,	// warning message
		INFO,		// success message
		LOG			// neutral message
	}
	
	public TestMessage(String name, MessageType type) {
		super(name, type == MessageType.ERROR ? true: false, new ArrayList<>());
		messageType = type;
	}

	public MessageType getMessageType() {
		return messageType;
	}
	
	@Override
	public JSONObject toJson() {
		JSONObject actionJson = new JSONObject();
		
		actionJson.put("type", "message");
		actionJson.put("name", encodeString(name, "json"));
		actionJson.put("messageType", messageType.toString());
		
		return actionJson;
	}
	
	@Override
	public TestMessage encode(String format) {
		TestMessage msg = new TestMessage(encodeString(name, format), messageType);
		msg.pwdToReplace = new ArrayList<>(pwdToReplace);
		msg.encoded = true;
		msg.actionException = actionException;
		if (actionException != null) {
			msg.actionExceptionMessage = actionException.getClass().toString() + ": " + encodeString(actionException.getMessage(), format);
		}
		return msg;
	}
}