/*
Copyright 2009 Janne Hietamäki
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package mungbean;
import static mungbean.CollectionUtil.list;
import static mungbean.CollectionUtil.map;
import static mungbean.CollectionUtil.merge;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import jdave.Block;
import jdave.Specification;
import jdave.junit4.JDaveRunner;
import mungbean.protocol.command.Count;
import mungbean.protocol.command.Distinct;
import mungbean.protocol.command.Group;
import mungbean.protocol.command.MapReduce;
import mungbean.protocol.command.MapReduceResult;
import mungbean.protocol.command.admin.IndexOptionsBuilder;
import mungbean.query.Query;
import mungbean.query.QueryBuilder;
import mungbean.query.Update;
import org.junit.runner.RunWith;
@RunWith(JDaveRunner.class)
@SuppressWarnings("unchecked")
public class MongoIntegrationTest extends Specification<Database> {
public class WithDatabase {
final ObjectId id = new ObjectId();
private final QueryBuilder idQuery = new Query().field("_id").is(id);
private final Map<String, Object> doc = newDoc(id, "bar");
private final boolean authenticate = false;
public Database create() {
if (authenticate) {
return new Mungbean(new Settings(), new Server("localhost", 27017, new Authentication("projectx", "joe", "passwordForJoe"))).openDatabase("projectx");
}
return new Mungbean(new Settings(), new Server("localhost", 27017)).openDatabase(new ObjectId().toHex());
}
public void destroy() {
if (!authenticate) {
context.dbAdmin().dropDatabase();
}
}
public void databaseTests() {
DBCollection<Map<String, Object>> collection = context.openCollection("foo");
long initialCount = collection.query(new Count(), new Query());
collection.save(doc);
List<Map<String, Object>> results = collection.query(idQuery);
specify(results.size(), does.equal(1));
specify(results.get(0).get("foo"), does.equal("bar"));
collection.query(new Distinct<List<Object>>("foo"), new Query()).contains("bar");
specify(collection.query(new Count(), new Query()), does.equal(initialCount + 1));
runGroup(collection);
specify(collection.query(new Count(), idQuery), does.equal(1));
collection.delete(idQuery);
specify(collection.query(idQuery).size(), does.equal(0));
specify(context.dbAdmin().getCollectionNames(), containsExactly("foo"));
specify(collection.query(new Count(), new Query()), does.equal(initialCount));
}
public void runGroup(DBCollection<Map<String, Object>> collection) {
HashMap<String, Double> initialValues = new HashMap<String, Double>();
initialValues.put("foo", 0D);
List<Map<String, Object>> result = collection.query(new Group(new String[] { "foo" }, initialValues, "function(obj, prev){ prev.csum=5; }"), new Query());
specify(result.get(0).get("csum"), does.equal(5D));
}
public void violationOfUniqueIndexThrowsAnException() {
final DBCollection<Map<String, Object>> collection = context.openCollection("foo");
collection.collectionAdmin().ensureIndex(new IndexOptionsBuilder().unique().field("foo"));
collection.save(newDoc(new ObjectId(), "bar"));
specify(new Block() {
@Override
public void run() throws Throwable {
collection.save(doc);
}
}, does.raise(MongoException.class));
}
public void advancedQueriesCanBeDoneWithTheDsl() {
final DBCollection<Map<String, Object>> collection = context.openCollection("foo");
for (int a = 0; a < 10; a++) {
collection.save(newDoc(new ObjectId(), a));
}
specify(collection.query(new Distinct<List<Object>>("foo"), new Query().field("foo").greaterThan(5)), containsExactly(6, 7, 8, 9));
List<Map<String, Object>> values = collection.query(new Query().field("foo").orderDescending().greaterThan(3).lessThan(8));
specify(values.size(), does.equal(4));
specify(values.get(0).get("foo"), does.equal(7));
specify(values.get(3).get("foo"), does.equal(4));
}
public void updatesCanBeDoneWithTheDsl() {
final DBCollection<Map<String, Object>> collection = context.openCollection("foo2");
for (int a = 0; a < 10; a++) {
collection.save(newDoc(new ObjectId(), a));
}
collection.update(new Query().field("foo").greaterThan(3), new Update().field("foo").increment(5));
specify(collection.query(new Query().field("foo").is(9)).size(), does.equal(2));
}
public void structuralUpdatesCanBeDone() {
final DBCollection<Map<String, Object>> collection = context.openCollection("foo2");
collection.save(newDoc(new ObjectId(), 1));
collection.update(new Query().field("foo").is(1), new Update().field("list").set(list("1", "2", "3")));
specify(collection.query(new Query().field("list").is(list("1", "2", "3"))).size(), does.equal(1));
}
public void massUpdatesCanBeDoneWithTheDsl() {
final DBCollection<Map<String, Object>> collection = context.openCollection("foo2");
for (int a = 0; a < 10; a++) {
collection.save(newDoc(new ObjectId(), a));
}
collection.update(new Query().field("foo").greaterThan(3), new Update().increment(map("foo", 5)));
specify(collection.query(new Query().field("foo").is(9)).size(), does.equal(2));
}
protected void mapReduceOperationCanBeExecuted() {
final DBCollection<Map<String, Object>> collection = context.openCollection("foo2");
for (int a = 0; a < 10; a++) {
collection.save(newDoc(new ObjectId(), a));
}
MapReduce mapReduce = new MapReduce("function(){emit(this.foo%2==0?\"even\":\"odd\",1); }", "function(k, vals){ var sum=0; for(var i in vals) sum += vals[i]; return sum;}");
mapReduce.setVerbose(true);
MapReduceResult result = collection.query(mapReduce, new Query());
specify(result.wasOk());
}
}
private Map<String, Object> newDoc(final ObjectId id, final Object value) {
return merge(map("foo", value), map("_id", id));
}
}