Skip to content

Commit

Permalink
#5 add test for insert, delete, update, drop collection
Browse files Browse the repository at this point in the history
  • Loading branch information
Thorsten Marx committed Sep 16, 2023
1 parent d8d75f9 commit 7da6203
Show file tree
Hide file tree
Showing 2 changed files with 74 additions and 28 deletions.
8 changes: 0 additions & 8 deletions ...ers/solr/src/main/java/com/github/thmarx/mongo/search/adapters/solr/SolrIndexAdapter.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -31,18 +31,10 @@
import com.github.thmarx.mongo.search.index.messages.UpdateMessage;
import com.github.thmarx.mongo.search.index.utils.PausableThread;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.logging.Level;
import java.util.logging.Logger;
import lombok.extern.slf4j.Slf4j;
import org.apache.solr.client.solrj.SolrClient;
import org.apache.solr.client.solrj.SolrQuery;
import org.apache.solr.client.solrj.SolrServerException;
import org.apache.solr.common.SolrInputDocument;
import org.apache.solr.common.SolrInputField;
import org.bson.Document;

/**
Expand Down
94 changes: 74 additions & 20 deletions ...olr/src/test/java/com/github/thmarx/mongo/search/adapters/solr/SolrMongoSearcherTest.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@
import com.github.thmarx.mongo.search.mapper.FieldMappers;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.model.Filters;
import com.mongodb.client.model.Updates;
import java.io.IOException;
import java.util.List;
import java.util.Map;
Expand All @@ -34,11 +35,13 @@
import org.apache.solr.client.solrj.request.CollectionAdminRequest;
import org.apache.solr.client.solrj.response.QueryResponse;
import org.apache.solr.client.solrj.response.SolrPingResponse;
import org.apache.solr.common.SolrDocument;
import org.apache.solr.common.SolrInputDocument;
import org.apache.solr.common.util.NamedList;
import org.assertj.core.api.Assertions;
import org.awaitility.Awaitility;
import org.bson.Document;
import org.bson.conversions.Bson;
import org.bson.types.ObjectId;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.BeforeMethod;
Expand Down Expand Up @@ -93,43 +96,92 @@ public void afterMethod() throws Exception, SolrServerException {
}

@Test
public void test_solr() throws IOException, InterruptedException, SolrServerException {
public void test_insert() throws IOException, InterruptedException, SolrServerException {

System.out.println("TEST");
Thread.sleep(2000);

SolrPingResponse response = solrClient.ping("dummy");
System.out.println("response: " + response.jsonStr());
assertCollectionSize(COLLECTION_DOKUMENTE, 0);

CollectionAdminRequest col = CollectionAdminRequest.createCollection(COLLECTION_DOKUMENTE, 1, 1);

insertDocument(COLLECTION_DOKUMENTE, Map.of("name", "thorsten"));

NamedList<Object> request = solrClient.request(col);
System.out.println(request);
Awaitility.await().atMost(10, TimeUnit.SECONDS).until(() -> getSize(COLLECTION_DOKUMENTE) > 0);

SolrInputDocument doc = new SolrInputDocument();
doc.addField("id", UUID.randomUUID().toString());
doc.addField("name", "thorsten");
solrClient.add(COLLECTION_DOKUMENTE, doc);
}
assertCollectionSize(COLLECTION_DOKUMENTE, 1);

insertDocument(COLLECTION_DOKUMENTE, Map.of("name", "thorsten"));

Awaitility.await().atMost(10, TimeUnit.SECONDS).until(() -> getSize(COLLECTION_DOKUMENTE) > 1);

assertCollectionSize(COLLECTION_DOKUMENTE, 2);
}

@Test
public void test_insert() throws IOException, InterruptedException, SolrServerException {
public void test_clear() throws IOException, InterruptedException, SolrServerException {

Thread.sleep(2000);
assertCollectionSize(COLLECTION_DOKUMENTE, 0);
insertDocument(COLLECTION_DOKUMENTE, Map.of("name", "thorsten"));
insertDocument(COLLECTION_DOKUMENTE, Map.of("name", "thorsten"));
Awaitility.await().atMost(10, TimeUnit.SECONDS).until(() -> getSize(COLLECTION_DOKUMENTE) == 2);
assertCollectionSize(COLLECTION_DOKUMENTE, 2);

database.getCollection(COLLECTION_DOKUMENTE).deleteMany(Filters.empty());

Awaitility.await().atMost(10, TimeUnit.SECONDS).until(() -> getSize(COLLECTION_DOKUMENTE) == 0);
assertCollectionSize(COLLECTION_DOKUMENTE, 0);
}

@Test
public void test_delete_single() throws IOException, InterruptedException, SolrServerException {

assertCollectionSize(COLLECTION_DOKUMENTE, 0);
insertDocument(COLLECTION_DOKUMENTE, Map.of("name", "thorsten"));
insertDocument(COLLECTION_DOKUMENTE, Map.of("name", "thorsten"));
Awaitility.await().atMost(10, TimeUnit.SECONDS).until(() -> getSize(COLLECTION_DOKUMENTE) == 2);
assertCollectionSize(COLLECTION_DOKUMENTE, 2);

Awaitility.await().atMost(10, TimeUnit.SECONDS).until(() -> getSize(COLLECTION_DOKUMENTE) > 0);

Document toDelete = database.getCollection(COLLECTION_DOKUMENTE).find(Filters.empty()).first();
deleteDocument(COLLECTION_DOKUMENTE, toDelete.getObjectId("_id").toString());

Awaitility.await().atMost(10, TimeUnit.SECONDS).until(() -> getSize(COLLECTION_DOKUMENTE) == 1);
assertCollectionSize(COLLECTION_DOKUMENTE, 1);
}

@Test
public void test_update() throws IOException, InterruptedException, SolrServerException {

assertCollectionSize(COLLECTION_DOKUMENTE, 0);
insertDocument(COLLECTION_DOKUMENTE, Map.of("name", "thorsten"));
Awaitility.await().atMost(10, TimeUnit.SECONDS).until(() -> getSize(COLLECTION_DOKUMENTE) == 1);
assertCollectionSize(COLLECTION_DOKUMENTE, 1);

Document toDelete = database.getCollection(COLLECTION_DOKUMENTE).find(Filters.empty()).first();
var uid = toDelete.getObjectId("_id").toString();
Bson update = Updates.combine(
Updates.set("name", "lara")
);
database.getCollection(COLLECTION_DOKUMENTE)
.updateOne(Filters.eq("_id", toDelete.getObjectId("_id")), update);

Awaitility.await().atMost(10, TimeUnit.SECONDS).until(() -> getSize(COLLECTION_DOKUMENTE) > 1);
Awaitility.await().atMost(10, TimeUnit.SECONDS).until(() -> {
var doc = getFromIndex(COLLECTION_DOKUMENTE, uid);
return doc.getFirstValue("name").equals("lara");
});
assertCollectionSize(COLLECTION_DOKUMENTE, 1);
}

@Test
public void test_drop_collection() throws IOException, InterruptedException, SolrServerException {

assertCollectionSize(COLLECTION_DOKUMENTE, 0);
insertDocument(COLLECTION_DOKUMENTE, Map.of("name", "thorsten"));
insertDocument(COLLECTION_DOKUMENTE, Map.of("name", "thorsten"));
Awaitility.await().atMost(10, TimeUnit.SECONDS).until(() -> getSize(COLLECTION_DOKUMENTE) == 2);
assertCollectionSize(COLLECTION_DOKUMENTE, 2);

database.getCollection(COLLECTION_DOKUMENTE).drop();

Awaitility.await().atMost(10, TimeUnit.SECONDS).until(() -> getSize(COLLECTION_DOKUMENTE) == 0);
assertCollectionSize(COLLECTION_DOKUMENTE, 0);
}

private void insertDocument(final String collectionName, final Map attributes) {
Expand All @@ -143,11 +195,13 @@ private void deleteDocument(final String collectionName, final String uid) {
collection.deleteOne(Filters.eq("_id", new ObjectId(uid)));
}

private SolrDocument getFromIndex(final String collection, final String uid) throws IOException, SolrServerException {
return solrClient.getById(collection, uid);
}

private long getSize(final String collection) throws IOException, SolrServerException {
SolrQuery query = new SolrQuery("*:*");
final QueryResponse response = solrClient.query(collection, query);

System.out.println(response.getResults().getNumFound());

return response.getResults().getNumFound();
}
Expand Down

0 comments on commit 7da6203

Please sign in to comment.