KissmetricsJsonToEnrichedJsonMapReducerTest.java

/*
 * Copyright (c) 2014-2015 Giving.com, trading as JustGiving or its affiliates. All Rights Reserved. 
 * 
 * Licensed under the Apache License, Version 2.0 (the "License"). 
 * You may not use this file except in compliance with the License. 
 * A copy of the License is located in the "license" file accompanying this file.
 * 
 * This file 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.
 * 
 * @author Richard Freeman
 * 
 */

package com.justgiving.raven.kissmetrics.jsonenricher;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import org.apache.hadoop.io.IntWritable;
import org.apache.hadoop.io.LongWritable;
import org.apache.hadoop.io.NullWritable;
import org.apache.hadoop.io.Text;
import org.apache.hadoop.mrunit.types.Pair;
import org.json.JSONException;
import org.junit.Test;
import org.skyscreamer.jsonassert.JSONAssert;

import com.justgiving.raven.kissmetrics.utils.KissmetricsJsonRowBuilder;

public class KissmetricsJsonToEnrichedJsonMapReducerTest extends KissmetricsJsonToEnrichedJsonTestBase {
	KissmetricsJsonRowBuilder rowbuilder = new KissmetricsJsonRowBuilder();
	
	
	@Test
	public void mapReducer_keyAndjson_enrichedJson() throws IOException, JSONException {
		List<Text> values = new ArrayList<Text>();
		values.add(new Text(rowbuilder.toString()));
				
		//values.add(new IntWritable(1));
		mapReduceDriver.withInput(new LongWritable(1), new Text(rowbuilder.toString()));
		
		rowbuilder.setValue("event_timedate","2014-04-15 16:57:33")
		  .setValue("filename","somefile")
		  .setValue("event", rowbuilder.getValue("_n"))
		  .setValue("user_email", "justgiving@gmail.com")
		  .setValue("user_email_back", "justgiving@gmail.com")
		  .setValue("user_km_id", "3lwlxqlulqe24q/jl4aqlibrtte=")
		  .setValue("km_timestamp", "1397577453")
		  .setValue("bucket","somefile");
		String expectedJSON = rowbuilder.toString();
		
		List<Pair<Text, NullWritable>> output = mapReduceDriver.run();	
		String actualJSON = output.get(0).getFirst().toString();
		
		JSONAssert.assertEquals(expectedJSON, actualJSON, true);	

	}

}