count_delete_test.c

/* count_delete.c */

#include "test.h"
#include "mongo.h"
#include <stdio.h>
#include <string.h>
#include <stdlib.h>

int main() {
    mongo conn[1];
    bson b;
    int i;

    const char *db = "test";
    const char *col = "c.simple";
    const char *ns = "test.c.simple";

    INIT_SOCKETS_FOR_WINDOWS;
    CONN_CLIENT_TEST;

    /* if the collection doesn't exist dropping it will fail */
    if ( !mongo_cmd_drop_collection( conn, "test", col, NULL )
            && mongo_count( conn, db, col, NULL ) != 0 ) {
        printf( "failed to drop collection\n" );
        exit( 1 );
    }

    for( i=0; i< 5; i++ ) {
        bson_init( &b );

        bson_append_new_oid( &b, "_id" );
        bson_append_int( &b , "a" , i+1 ); /* 1 to 5 */
        bson_finish( &b );

        mongo_insert( conn , ns , &b, NULL );
        bson_destroy( &b );
    }

    /* query: {a: {$gt: 3}} */
    bson_init( &b );
    {
        bson_append_start_object( &b, "a" );
        bson_append_int( &b, "$gt", 3 );
        bson_append_finish_object( &b );
    }
    bson_finish( &b );

    ASSERT( mongo_count( conn, db, col, NULL ) == 5 );
    ASSERT( mongo_count( conn, db, col, &b ) == 2 );

    mongo_remove( conn, ns, &b, NULL );

    ASSERT( mongo_count( conn, db, col, NULL ) == 3 );
    ASSERT( mongo_count( conn, db, col, &b ) == 0 );

    bson_destroy( &b );
    mongo_cmd_drop_db( conn, db );
    mongo_destroy( conn );
    return 0;
}