TestEntityTestBase.java

/*
 * Copyright (C) 2011-2016 Markus Junginger, greenrobot (http://greenrobot.org)
 *
 * This file is part of greenDAO Generator.
 *
 * greenDAO Generator is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * greenDAO Generator is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with greenDAO Generator.  If not, see <http://www.gnu.org/licenses/>.
 */

package org.greenrobot.greendao.daotest.entity;

import java.util.ArrayList;
import java.util.List;

import org.greenrobot.greendao.test.AbstractDaoTest;
import org.greenrobot.greendao.daotest.TestEntity;
import org.greenrobot.greendao.daotest.TestEntityDao;

public abstract class TestEntityTestBase extends AbstractDaoTest<TestEntityDao, TestEntity, Long> {

    public TestEntityTestBase() {
        super(TestEntityDao.class);
    }

    protected TestEntity createEntity(int simpleInteger, String simpleString) {
        TestEntity entity = new TestEntity();
        entity.setId(null);
        entity.setSimpleStringNotNull("green");
        entity.setSimpleInteger(simpleInteger);
        entity.setSimpleString(simpleString);
        return entity;
    }

    protected ArrayList<TestEntity> insert(int count) {
        ArrayList<TestEntity> list = new ArrayList<TestEntity>();
        for (int i = 0; i < count; i++) {
            TestEntity entity = createEntity(getSimpleInteger(i), getSimpleString(i));
            list.add(entity);
        }
        dao.insertInTx(list);
        return list;
    }

    protected String getSimpleString(int i) {
        return "String" + (i + 100);
    }

    protected int getSimpleInteger(int i) {
        return 100 + i;
    }

    protected void assertIds(List<TestEntity> list, List<TestEntity> list2) {
        for (int i = 0; i < list.size(); i++) {
            TestEntity entity = list.get(i);
            TestEntity lazyEntity = list2.get(i);
            assertIds(entity, lazyEntity);
        }
    }

    protected void assertIds(TestEntity entity, TestEntity entity2) {
        assertEquals(entity.getId(), entity2.getId());
    }

}