HeroRepository_UnitTest

package heroRepository;

import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

import static org.junit.Assert.*;

public class HeroRepositoryTest {
    private Item item;
    private Hero hero;
    private HeroRepository hr;

    @Before
    public void initHeroAndItem(){
        this.item = new Item(10,9,6);
        this.hero = new Hero("Pesho", 500, this.item);
        this.hr = new HeroRepository();
    }

    @Test
    public void whenHeroRepositoryConstr_thenCreateNewArrayList(){
        this.hr = new HeroRepository();
        Assert.assertNotNull(hr);
    }

    @Test
    public void givenHero_whenAdd_thenAddHeroToRepository(){
        this.hr.add(this.hero);
        Assert.assertEquals(1, this.hr.getCount());
    }

    @Test (expected = IllegalArgumentException.class)
    public void givenHero_whenAdd_thenThrowIllegalArgumentException(){
        this.hr.add(this.hero);
        Hero anotherHero = new Hero("Pesho", 15, this.item);
        this.hr.add(anotherHero);
    }

    @Test
    public void givenHero_whenRemove_thenRemoveHeroFromRepository(){
        this.hero = new Hero("Pesho", 15, this.item);
        this.hr.add(this.hero);
        this.hr.remove("Pesho");
        Assert.assertEquals(0, this.hr.getCount());
    }

    @Test(expected = NullPointerException.class)
    public void givenHeroDoesNotExists_whenRemoveHero_thenThrowNullPointerException(){
        this.hr.remove("Pesho");
    }

    @Test
    public void testGetHeroWithHighestStrength(){
        this.hr.add(this.hero);
        Item anotherItem = new Item(8,10,9);
        Hero anotherHero = new Hero("Asd", 5, anotherItem);
        this.hr.add(anotherHero);
        Hero test = hr.getHeroWithHighestStrength();
        Assert.assertEquals(10, test.getItem().getStrength());
    }

    @Test (expected = NullPointerException.class)
    public void whenGetHeroWithHighestStrength_thenThrowNullPointerException(){
        this.hr.getHeroWithHighestStrength();
    }

    @Test
    public void testGetHeroWithHighestAgility(){
        this.hr.add(this.hero);
        Item anotherItem = new Item(8,10,9);
        Hero anotherHero = new Hero("Asd", 5, anotherItem);
        this.hr.add(anotherHero);
        Hero test = hr.getHeroWithHighestAgility();
        Assert.assertEquals(10, test.getItem().getAgility());
    }

    @Test (expected = NullPointerException.class)
    public void whenGetHeroWithHighestAgility_thenThrowNullPointerException(){
        this.hr.getHeroWithHighestAgility();
    }

    @Test
    public void testGetHeroWithHighestIntelligence(){
        this.hr.add(this.hero);
        Item anotherItem = new Item(20,20,-5);
        Hero anotherHero = new Hero("Asd", 5, anotherItem);
        this.hr.add(anotherHero);
        Hero test = hr.getHeroWithHighestIntelligence();
        Assert.assertEquals(6, test.getItem().getIntelligence());
    }

    @Test (expected = NullPointerException.class)
    public void whenGetHeroWithHighestIntelligence_thenThrowNullPointerException(){
        this.hr.getHeroWithHighestIntelligence();
    }

    @Test
    public void testGetCount(){
        Assert.assertEquals(0,this.hr.getCount());
    }

    @Test
    public void testToString(){
        this.hr.add(this.hero);
        Item anotherItem = new Item(8,10,9);
        Hero anotherHero = new Hero("Asd", 5, anotherItem);
        this.hr.add(anotherHero);
        String output = String.format("Hero: Pesho – 500%n" +
                "  *  Strength: 10%n" +
                "  *  Agility: 9%n" +
                "  *  Intelligence: 6%n") +
                String.format("Hero: Asd – 5%n" +
                        "  *  Strength: 8%n" +
                        "  *  Agility: 10%n" +
                        "  *  Intelligence: 9%n");
        Assert.assertEquals(output, this.hr.toString());
    }
}