Class FindWithModifierTest

src/test/java/org/jongo/FindWithModifierTest.java:32–83

public class FindWithModifierTest extends JongoTestBase {

    private MongoCollection collection;

    @Before
    public void setUp() throws Exception {
        collection = createEmptyCollection("friends");
    }

    @After
    public void tearDown() throws Exception {
        dropCollection("friends");
    }

    @Test
    public void canFindWithHint() throws Exception {
        /* given */
        Friend noName = new Friend(new ObjectId(), null);
        collection.save(noName);

        collection.ensureIndex("{name: 1}", "{sparse: true}");

        /* when */
        // force to use _id index instead of name index which is sparsed
        Iterator<Friend> friends = collection.find().hint("{$natural: 1}").sort("{name: 1}").as(Friend.class);

        /* then */
        assertThat(friends.hasNext()).isTrue();
    }

    @Test
    public void canUseQueryModifier() throws Exception {
        /* given */
        collection.save(new Friend(new ObjectId(), "John"));
        collection.save(new Friend(new ObjectId(), "Robert"));

        /* when */
        Iterator<Friend> friends = collection.find()
                .with(new QueryModifier() {
                    public void modify(DBCursor cursor) {
                        cursor.limit(1);
                    }
                })
                .as(Friend.class);

        /* then */
        assertThat(friends.hasNext()).isTrue();
        friends.next();
        assertThat(friends.hasNext()).isFalse();
    }

}