@Before public void setUp() throws Exception { mdc = new MongoConnector( DatabaseConfigurationFactory.getDatabaseConfiguration("junit-QueryTest")); mdc.waitForWrites(true); test = new MongoDocument(); test.setAction(Action.ADD); test.putContentField("name", "test"); test.putMetadataField("date", new Date()); test2 = new MongoDocument(); test2.setAction(Action.DELETE); test2.putContentField("name", "test"); test2.putContentField("number", 2); test2.putMetadataField("date", new Date()); random = new MongoDocument(); try { mdc.connect(); } catch (IOException e) { fail("IOException when establishing connection"); } DatabaseDocument<MongoType> d; while ((d = mdc.getDocumentReader().getDocument(new MongoQuery())) != null) { mdc.getDocumentWriter().delete(d); } mdc.getDocumentWriter().insert(test); mdc.getDocumentWriter().insert(test2); mdc.getDocumentWriter().insert(random); }
public long insertCurrentDocuments(int count) throws Exception { long start = System.currentTimeMillis(); for (int i = 0; i < count; i++) { MongoDocument d = new MongoDocument(); d.putContentField(getRandomString(5), getRandomString(20)); mdc.getDocumentWriter().insert(d); } return System.currentTimeMillis() - start; }
@Test public void testRequireNotTouchedByStage() throws JsonException { LocalQuery q = new LocalQuery(); q.requireNotTouchedByStage("xyz"); List<DatabaseDocument<MongoType>> ds = mdc.getDocumentReader().getDocuments(new MongoQuery(q.toJson()), 3); if (ds.size() != 3) { fail("Received incorrect number of documents.."); } Document d = mdc.getDocumentWriter().getAndTag(new MongoQuery(), "xyz"); if (d == null) { fail("Should have gotten a document back..."); } mdc.getDocumentWriter().markTouched(d.getID(), "xyz"); ds = mdc.getDocumentReader().getDocuments(new MongoQuery(q.toJson()), 3); if (ds.size() != 2) { fail("Received incorrect number of documents.."); } }