ArmorDAOTest.java

package com.lateensoft.pathfinder.toolkit.db.dao.table;

import com.lateensoft.pathfinder.toolkit.dao.DataAccessException;
import com.lateensoft.pathfinder.toolkit.model.character.items.Armor;
import com.lateensoft.pathfinder.toolkit.model.character.items.Size;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.robolectric.Robolectric;
import org.robolectric.RobolectricTestRunner;
import org.robolectric.annotation.Config;

import java.util.List;

import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.Matchers.*;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNull;

@Config(manifest=Config.NONE)
@RunWith(RobolectricTestRunner.class)
public class ArmorDAOTest extends CharacterComponentDAOTest {
    private Armor m_armor1;
    private Armor m_armor2;
    private ArmorDAO dao;

    @Override
    public void setUp() throws Exception {
        super.setUp();
        initAndTestAdd();
    }

    private void initAndTestAdd() throws DataAccessException {
        dao = new ArmorDAO(Robolectric.application);

        m_armor1 = new Armor();
        setValues(m_armor1, m_armor1.getId(), "Heavy armor",
                7.5, true, 1, /*ACP*/-2, 3, 4, 5, "armor", Size.MEDIUM);

        m_armor2 = new Armor();
        setValues(m_armor2, m_armor2.getId(), "Hat",
                1.0, false, 10, /*ACP*/-4, 100, 11, 20, "Hat", Size.SMALL);

        dao.add(getTestCharacterId(), m_armor1);
        dao.add(getTestCharacterId(), m_armor2);
    }

    @Test
    public void testQuery() {
        Armor queried = dao.find(m_armor1.getId());

        assertEquals(queried, m_armor1);
    }

    @Test
    public void testUpdate() throws DataAccessException {
        Armor toUpdate = m_armor2;
        setValues(m_armor2, m_armor2.getId(), "Larger Hat",
                2.0, true, 11, 25, 101, 12, 21, "Hat thing", Size.MEDIUM);

        dao.update(getTestCharacterId(), toUpdate);
        Armor updated = dao.find(toUpdate.getId());
        assertEquals(toUpdate, updated);
    }

    @Test
    public void testDelete() throws DataAccessException {
        dao.remove(m_armor1);
        assertNull(dao.find(m_armor1.getId()));
    }

    @Test
    public void testFindAllForOwner() {
        List<Armor> queriedItems = dao.findAllForOwner(getTestCharacterId());
        assertThat(queriedItems, hasItems(m_armor1, m_armor2));
    }

    @Test
    public void testMaxDex() throws DataAccessException {
        Armor armor1 = new Armor();
        setValues(armor1, armor1.getId(), "Hat",
                1.0, true, 10, 24, 2, 11, 20, "Hat", Size.SMALL);
        dao.add(getTestCharacterId(), armor1);
        Armor armor2 = new Armor();
        setValues(armor2, armor2.getId(), "Hat",
                1.0, true, 10, 24, 42, 11, 20, "Hat", Size.SMALL);
        dao.add(getTestCharacterId(), armor2);

        assertEquals(2, dao.getMaxDexForCharacter(getTestCharacterId()));

        m_armor1.setWorn(false);
        dao.update(getTestCharacterId(), m_armor1);
        armor1.setWorn(false);
        dao.update(getTestCharacterId(), armor1);
        armor2.setWorn(false);
        dao.update(getTestCharacterId(), armor2);
        assertEquals(Integer.MAX_VALUE, dao.getMaxDexForCharacter(getTestCharacterId()));
    }

    @Test
    public void testArmorCheck() throws DataAccessException {
        Armor armor1 = new Armor();
        setValues(armor1, armor1.getId(), "Heavy armor",
                7.5, true, 1, /*ACP*/-5, 3, 4, 5, "armor", Size.MEDIUM);
        dao.add(getTestCharacterId(), armor1);
        assertEquals(-7, dao.getArmorCheckPenaltyForCharacter(getTestCharacterId()));

        dao.remove(m_armor1);
        dao.remove(armor1);
        assertEquals(0, dao.getArmorCheckPenaltyForCharacter(getTestCharacterId()));
    }

    private void setValues(Armor toUpdate, long id, String name,
                           double weight, boolean worn, int ACBonus, int checkPen, int maxDex, int spellFail,
                           int speed, String specProp, Size size) {
        toUpdate.setId(id);
        toUpdate.setName(name);
        toUpdate.setWeight(weight);
        toUpdate.setWorn(worn);
        toUpdate.setACBonus(ACBonus);
        toUpdate.setArmorCheckPenalty(checkPen);
        toUpdate.setMaxDex(maxDex);
        toUpdate.setSpellFail(spellFail);
        toUpdate.setSpeed(speed);
        toUpdate.setSpecialProperties(specProp);
        toUpdate.setSize(size);
    }

}