package model;

import org.junit.*;

import model.Modifier;
import model.PrimitiveType;
import model.UmlMethod;
import model.UmlParams;
import model.Visibility;
import java.util.Set;
import java.util.HashSet;

import static org.junit.Assert.*;

/**
 * Test class for {@link UmlMethod}
 * @author echodeltaFR
 * @version 1.3
 */
public class UmlMethodTest {

    private UmlMethod methodTest1;
    private UmlMethod methodTest2;
    private UmlMethod methodTest3;
    private UmlParams paramTest1;
    private UmlParams paramTest2;
    private Set<UmlParams> paramTest1Set;
    private Set<UmlParams> params;
    private Set<Modifier> modifiers;

    @Before public void setUp() {
        paramTest1 = new UmlParams(PrimitiveType.INT, "Paramètre 1");
        paramTest1Set = new HashSet<UmlParams>();
        paramTest1Set.add(paramTest1);
        paramTest2 = new UmlParams(PrimitiveType.BOOLEAN, "Paramètre 2");
        params = new HashSet<UmlParams>();
        params.add(paramTest1);
        params.add(paramTest2);
        modifiers = new HashSet<Modifier>();
        modifiers.add(Modifier.ABSTRACT);
        methodTest1 = new UmlMethod("MethodTest1");
        methodTest2 = new UmlMethod("MethodTest2", null, null, null, null);
        methodTest3 = new UmlMethod("MethodTest3", params, PrimitiveType.INT, Visibility.PRIVATE, modifiers);
    }

    @Test public void testConstructor1() {
        assertEquals("Check the name", "MethodTest1", methodTest1.getName());
        assertTrue("Check the params", methodTest1.getParams().isEmpty());
        assertEquals("Check the return type", null, methodTest1.getReturnType());
        assertEquals("Check the visibility", Visibility.PUBLIC, methodTest1.getVisibility());
        assertTrue("Check the modifiers", methodTest1.getModifiers().isEmpty());
    }

    @Test public void testConstructor2() {
        assertEquals("Check the name", "MethodTest2", methodTest2.getName());
        assertTrue("Check the params", methodTest2.getParams().isEmpty());
        assertEquals("Check the return type", null, methodTest2.getReturnType());
        assertEquals("Check the visibility", Visibility.PUBLIC, methodTest2.getVisibility());
        assertTrue("Check the modifiers", methodTest2.getModifiers().isEmpty());
    }

    @Test public void testConstructor3() {
        assertEquals("Check the name", "MethodTest3", methodTest3.getName());
        assertTrue("Check the params", params.equals(methodTest3.getParams()));
        assertEquals("Check the return type", PrimitiveType.INT, methodTest3.getReturnType());
        assertEquals("Check the visibility", Visibility.PRIVATE, methodTest3.getVisibility());
        assertTrue("Check the modifiers", modifiers.equals(methodTest3.getModifiers()));
    }

    @Test public void testSetName() {
        methodTest3.setName("MethodTest33");
        assertEquals("Check the name setter", "MethodTest33", methodTest3.getName());
    }

    @Test public void testSetReturnType() {
        methodTest3.setReturnType(PrimitiveType.LONG);
        assertEquals("Check the return type setter", PrimitiveType.LONG, methodTest3.getReturnType());
    }

    @Test public void testAddParam() {
        methodTest1.addParam(paramTest1);
        assertTrue("Check the param adding", paramTest1Set.equals(methodTest1.getParams()));
    }

    @Test public void testRemoveParam() {
        methodTest1.removeParam(paramTest1);
        assertTrue("Check the param removing", methodTest1.getParams().isEmpty());
    }

    @Test public void testAddParams() {
        methodTest1.addParams(params);
        assertTrue("Check the params adding", params.equals(methodTest1.getParams()));
    }

    @Test public void testRemoveParams() {
        methodTest1.removeParams(params);
        assertTrue("Check the params removing", methodTest1.getParams().isEmpty());
    }

}