Getting Started: Writing your Unit Test using TestDriver

The first thing we need to implement is a class derived from RavenServerLocator

public abstract class RavenServerLocator {
    /**
     * Allows to fetch the server path
     * @return server path
     */
    public abstract String getServerPath();

    /**
     * Allows to fetch the command used to invoke the server.
     * @return command
     */
    public abstract String getCommand();

    /**
     * Allows to fetch the command arguments.
     * @return command line arguments
     */
    public String[] getCommandArguments() {
        return new String[0];
    }
}

Example

public class MyRavenDBLocator extends RavenServerLocator {
    @Override
    public String getServerPath() {
        return "/opt/RavenDB/Server/Raven.Server";
    }

    @Override
    public String getCommand() {
        return getServerPath();
    }
}

Now that we learned how to implement a server locator we can define a class that derives from Raven's TestDriver. Lets start with reviewing the TestDriver's methods and properties and later we will get into implementation (complete code sample of a RavenTestDriver can be found at the bottom of the page).

Properties and Methods

Pre-Initializing the IDocumentStore allows you to mutate the conventions used by the document store.

Example

//This allows us to modify the conventions of the store we get from 'getDocumentStore'
@Override
protected void customizeStore(DocumentStore store) {
    store.getConventions().setMaxNumberOfRequestsPerSession(50);
}

Example

@Test
public void myFirstTest() throws Exception {
    try (IDocumentStore store = getDocumentStore()) {
        store.executeIndex(new TestDocumentByName());

        try (IDocumentSession session = store.openSession()) {
            TestDocument document1 = new TestDocument();
            document1.setName("Hello world!");

            TestDocument document2 = new TestDocument();
            document2.setName("Goodbye...");

            session.store(document1);
            session.store(document2);
            session.saveChanges();
        }

        waitForIndexing(store);  //If we want to query documents sometime we need to wait for the indexes to catch up
        waitForUserToContinueTheTest(store); //Sometimes we want to debug the test itself, this redirect us to the studio

        try (IDocumentSession session = store.openSession()) {
            List<TestDocument> query = session.query(TestDocument.class, TestDocumentByName.class)
                .whereEquals("name", "hello")
                .toList();

            Assert.assertEquals(1, query.size());
        }
    }
}

In the test we get an IDocumentStore to our test database, deploy an index and insert two documents into it. We then wait for the indexing to complete and launch the Studio so we can verify the documents and index are deployed (we can remove this line once the test is working). At the end of the test we query for TestDocument where their name contains the world 'hello' and assert that we have only one such document.

public class MyRavenDBTestDriver extends RavenTestDriver {
    public MyRavenDBTestDriver(RavenServerLocator locator, RavenServerLocator securedLocator) {
        super(new MyRavenDBLocator(), null);
    }

    //This allows us to modify the conventions of the store we get from 'getDocumentStore'
    @Override
    protected void customizeStore(DocumentStore store) {
        store.getConventions().setMaxNumberOfRequestsPerSession(50);
    }

    @Test
    public void myFirstTest() throws Exception {
        try (IDocumentStore store = getDocumentStore()) {
            store.executeIndex(new RavenDBTestDriver.TestDocumentByName());

            try (IDocumentSession session = store.openSession()) {
                TestDocument document1 = new TestDocument();
                document1.setName("Hello world!");

                TestDocument document2 = new TestDocument();
                document2.setName("Goodbye...");

                session.store(document1);
                session.store(document2);
                session.saveChanges();
            }

            waitForIndexing(store);  //If we want to query documents sometime we need to wait for the indexes to catch up
            waitForUserToContinueTheTest(store); //Sometimes we want to debug the test itself, this redirect us to the studio

            try (IDocumentSession session = store.openSession()) {
                List<RavenDBTestDriver.TestDocument> query = session
                        .query(
                            RavenDBTestDriver.TestDocument.class,
                            RavenDBTestDriver.TestDocumentByName.class
                        )
                    .whereEquals("name", "hello")
                    .toList();

                Assert.assertEquals(1, query.size());
            }
        }
    }
}

public class MyRavenDBLocator extends RavenServerLocator {
    @Override
    public String getServerPath() {
        return "/opt/RavenDB/Server/Raven.Server";
    }

    @Override
    public String getCommand() {
        return getServerPath();
    }
}

public static class TestDocumentByName extends AbstractIndexCreationTask {
    public TestDocumentByName() {
        map = "from doc in docs select new { doc.name }";
        index("name", FieldIndexing.SEARCH);
    }
}

public class TestDocument {
    private String name;

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }
}