plantuml/test/net/sourceforge/plantuml/cheneer/command/CommandAssociateTest.java

package net.sourceforge.plantuml.cheneer.command;

import static org.assertj.core.api.Assertions.assertThat;

import java.util.List;

import org.junit.jupiter.api.Test;

import net.sourceforge.plantuml.abel.Link;
import net.sourceforge.plantuml.cheneer.ChenEerDiagram;
import net.sourceforge.plantuml.command.Command;
import net.sourceforge.plantuml.command.CommandExecutionResult;
import net.sourceforge.plantuml.core.UmlSource;
import net.sourceforge.plantuml.klimt.color.NoSuchColorException;
import net.sourceforge.plantuml.regex.IRegex;
import net.sourceforge.plantuml.regex.RegexResult;
import net.sourceforge.plantuml.utils.BlocLines;

public class CommandAssociateTest {

	private final Command<ChenEerDiagram> command = new CommandAssociate();

	private final ChenEerDiagram diagram = new ChenEerDiagram(UmlSource.create(List.of(), false), null);

	@Test
	void test_parse() {
		IRegex regex = CommandAssociate.getRegexConcat();
		RegexResult matcher = regex.matcher("E1 -- E2");

		assertThat(matcher).isNotNull();
		assertThat(matcher.get("NAME1", 0)).isEqualTo("E1");
		assertThat(matcher.get("NAME2", 0)).isEqualTo("E2");
		assertThat(matcher.get("PARTICIPATION", 0)).isEqualTo("-");
		assertThat(matcher.get("CARDINALITY", 0)).isNull();
	}

	@Test
	void test_parseWithCompulsoryParticipation() {
		IRegex regex = CommandAssociate.getRegexConcat();
		RegexResult matcher = regex.matcher("E1 == E2");

		assertThat(matcher).isNotNull();
		assertThat(matcher.get("NAME1", 0)).isEqualTo("E1");
		assertThat(matcher.get("NAME2", 0)).isEqualTo("E2");
		assertThat(matcher.get("PARTICIPATION", 0)).isEqualTo("=");
		assertThat(matcher.get("CARDINALITY", 0)).isNull();
	}

	@Test
	void test_parseWithCardinality() {
		IRegex regex = CommandAssociate.getRegexConcat();
		RegexResult matcher = regex.matcher("E1 -N- E2");

		assertThat(matcher).isNotNull();
		assertThat(matcher.get("NAME1", 0)).isEqualTo("E1");
		assertThat(matcher.get("NAME2", 0)).isEqualTo("E2");
		assertThat(matcher.get("PARTICIPATION", 0)).isEqualTo("-");
		assertThat(matcher.get("CARDINALITY", 0)).isEqualTo("N");
	}

	@Test
	void test_parseCardinalityRange() {
		IRegex regex = CommandAssociate.getRegexConcat();
		RegexResult matcher = regex.matcher("E1 -(1, n)- E2");

		assertThat(matcher).isNotNull();
		assertThat(matcher.get("NAME1", 0)).isEqualTo("E1");
		assertThat(matcher.get("NAME2", 0)).isEqualTo("E2");
		assertThat(matcher.get("PARTICIPATION", 0)).isEqualTo("-");
		assertThat(matcher.get("CARDINALITY", 0)).isEqualTo("(1, n)");
	}

	@Test
	void test_execute() throws NoSuchColorException {
		new CommandCreateEntity().execute(diagram, BlocLines.singleString("entity E1 {"));
		new CommandCreateEntity().execute(diagram, BlocLines.singleString("entity E2 {"));

		BlocLines lines = BlocLines.singleString("E1 -- E2");
		CommandExecutionResult result = command.execute(diagram, lines);

		assertThat(result).matches(CommandExecutionResult::isOk);

		assertThat(diagram.getLinks().size()).isEqualTo(1);
		Link link = diagram.getLinks().get(0);
		assertThat(link.getEntity1().getName()).isEqualTo("E1");
		assertThat(link.getEntity2().getName()).isEqualTo("E2");
		assertThat(link.getType().getStyle().getStroke3().getThickness()).isEqualTo(1);
		assertThat(link.getLabel().toString()).isEqualTo("NULL");
	}

	@Test
	void test_executeWithCompulsoryParticipation() throws NoSuchColorException {
		new CommandCreateEntity().execute(diagram, BlocLines.singleString("entity E1 {"));
		new CommandCreateEntity().execute(diagram, BlocLines.singleString("entity E2 {"));

		BlocLines lines = BlocLines.singleString("E1 == E2");
		CommandExecutionResult result = command.execute(diagram, lines);

		assertThat(result).matches(CommandExecutionResult::isOk);

		assertThat(diagram.getLinks().size()).isEqualTo(1);
		Link link = diagram.getLinks().get(0);
		assertThat(link.getEntity1().getName()).isEqualTo("E1");
		assertThat(link.getEntity2().getName()).isEqualTo("E2");
		assertThat(link.getType().getStyle().getStroke3().getThickness()).isEqualTo(2);
		assertThat(link.getLabel().toString()).isEqualTo("NULL");
	}

	@Test
	void test_executeWithCardinality() throws NoSuchColorException {
		new CommandCreateEntity().execute(diagram, BlocLines.singleString("entity E1 {"));
		new CommandCreateEntity().execute(diagram, BlocLines.singleString("entity E2 {"));

		BlocLines lines = BlocLines.singleString("E1 -N- E2");
		CommandExecutionResult result = command.execute(diagram, lines);

		assertThat(result).matches(CommandExecutionResult::isOk);

		assertThat(diagram.getLinks().size()).isEqualTo(1);
		Link link = diagram.getLinks().get(0);
		assertThat(link.getEntity1().getName()).isEqualTo("E1");
		assertThat(link.getEntity2().getName()).isEqualTo("E2");
		assertThat(link.getType().getStyle().getStroke3().getThickness()).isEqualTo(1);
		assertThat(link.getLabel().toString()).isEqualTo("[N]");
	}
}