MongoStorageTest.java

package com.mongodb.hadoop.pig;

import com.mongodb.BasicDBObjectBuilder;
import com.mongodb.DBObject;
import org.apache.pig.ResourceSchema;
import org.apache.pig.impl.util.Utils;
import org.junit.Test;

import java.util.HashMap;
import java.util.Map;
import java.util.Set;

import static org.junit.Assert.assertEquals;

public class MongoStorageTest {
    @Test
    public void testMap() throws Exception {
        MongoStorage ms = new MongoStorage();
        BasicDBObjectBuilder builder = BasicDBObjectBuilder.start();
        ResourceSchema schema = new ResourceSchema(Utils.getSchemaFromString("m:map[]"));

        Map<String, Object> val = new HashMap<String, Object>();
        val.put("f1", 1);
        val.put("f2", "2");

        ms.writeField(builder, schema.getFields()[0], val);

        DBObject out = builder.get();

        Set<String> outKeySet = out.keySet();

        assertEquals(2, outKeySet.size());
        assertEquals(1, out.get("f1"));
        assertEquals("2", out.get("f2"));
    }

}