nl.knaw.huygens.timbuctoo.storage.mongo.MongoStorageTestBase.java Source code

Introduction

Here is the source code for nl.knaw.huygens.timbuctoo.storage.mongo.MongoStorageTestBase.java

Source

package nl.knaw.huygens.timbuctoo.storage.mongo;

/*
 * #%L
 * Timbuctoo core
 * =======
 * Copyright (C) 2012 - 2015 Huygens ING
 * =======
 * This program 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.
 * 
 * This program 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 this program.  If not, see
 * <http://www.gnu.org/licenses/gpl-3.0.html>.
 * #L%
 */

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.fail;
import static org.mockito.Matchers.anyString;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import java.io.IOException;
import java.net.UnknownHostException;

import nl.knaw.huygens.timbuctoo.model.Entity;

import org.junit.Before;

import com.mongodb.DB;
import com.mongodb.DBCollection;
import com.mongodb.DBCursor;
import com.mongodb.DBObject;
import com.mongodb.Mongo;
import com.mongodb.MongoException;

public abstract class MongoStorageTestBase {

    public static final String DEFAULT_ID = "TSTD000000000001";

    protected Mongo mongo; // TODO eliminate
    protected DB db; // TODO eliminate
    protected MongoDB mongoDB;
    protected DBCollection dbCollection;
    protected EntityIds entityIds;
    protected MongoQueries queries;

    @Before
    public void setUp() throws UnknownHostException, MongoException {
        mongo = mock(Mongo.class);
        db = mock(DB.class);
        mongoDB = mock(MongoDB.class);
        dbCollection = mock(DBCollection.class);
        entityIds = mock(EntityIds.class);
        queries = new MongoQueries();

        when(db.getCollection(anyString())).thenReturn(dbCollection);
        when(mongoDB.getCollection(anyString())).thenReturn(dbCollection);

        setupStorage();
    }

    protected abstract void setupStorage() throws UnknownHostException, MongoException;

    protected <T extends Entity> void assertEqualDocs(T expected, T actual) {
        try {
            // Use assertEquals instead of assertNull for a clearer message when a test fails.
            assertEquals(null, MongoDiff.diffDocuments(expected, actual));
        } catch (IOException e) {
            fail(e.getMessage());
        }
    }

    protected DBCursor createDBCursorWithOneValue(DBObject dbObject) {
        DBCursor cursor = mock(DBCursor.class);
        when(cursor.hasNext()).thenReturn(true, false);
        when(cursor.next()).thenReturn(dbObject);
        when(cursor.count()).thenReturn(1);
        return cursor;
    }

    protected DBCursor createCursorWithoutValues() {
        DBCursor cursor = mock(DBCursor.class);
        when(cursor.hasNext()).thenReturn(false);
        when(cursor.count()).thenReturn(0);
        return cursor;
    }

}