com.seleniumtests.reporter.info.ImageLinkInfo Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of core-selenium4 Show documentation

There is a newer version: 5.1.13

package com.seleniumtests.reporter.info;

import com.seleniumtests.core.SeleniumTestsContextManager;
import com.seleniumtests.core.utils.TestNGResultUtils;
import com.seleniumtests.reporter.logger.FileContent;
import org.json.JSONObject;
import org.testng.ITestResult;

import java.io.File;
import java.nio.file.Paths;

public class ImageLinkInfo extends HyperlinkInfo implements FileLinkInfo {

	FileContent imageFileContent;

	public ImageLinkInfo(FileContent imageFileContent) {

		super("Image", imageFileContent.getName());
		this.imageFileContent = imageFileContent;
	}

	/**
	 * Returns the string associated to this info
	 * For HTML, path is relative to the root of test results
	 * @param format
	 * @return
	 */
	@Override
	public String encode(String format) {
		
		if ("html".equals(format)) {
			link = Paths.get(SeleniumTestsContextManager.getGlobalContext().getOutputDirectory()).relativize(imageFileContent.getFile().toPath()).toString();
			return String.format("", link.replace("\\", "/"));
		} else {
			if (format != null) {
				return super.encode(format);
			} else {
				logger.error("format cannot be null");
			}
		}

		return description;
	}

	@Override
	public JSONObject toJson() {
		return new JSONObject().put("type", "imagelink")
				.put("info", description)
				.put("link", imageFileContent.getName())
				.put("id", imageFileContent.getId() == null ? JSONObject.NULL: imageFileContent.getId());
	}

	@Override
	public FileContent getFileContent() {
		return imageFileContent;
	}

}