package com.netflix.astyanax.cql.test;

import java.nio.ByteBuffer;
import java.util.ArrayList;
import java.util.List;

import org.apache.cassandra.db.marshal.AbstractType;
import org.apache.cassandra.db.marshal.CompositeType;
import org.apache.cassandra.db.marshal.UTF8Type;
import org.apache.log4j.Logger;
import org.junit.AfterClass;
import org.junit.Assert;
import org.junit.BeforeClass;
import org.junit.Test;

import com.netflix.astyanax.SerializerPackage;
import com.netflix.astyanax.model.ColumnFamily;
import com.netflix.astyanax.model.Composite;
import com.netflix.astyanax.serializers.LongSerializer;
import com.netflix.astyanax.serializers.SpecificCompositeSerializer;
import com.netflix.astyanax.serializers.StringSerializer;

public class SerializerPackageTests extends KeyspaceTests {

	private static final Logger LOG = Logger.getLogger(SerializerPackageTests.class);

	public static ColumnFamily<String, Long> CF_SERIALIZER1 = ColumnFamily
			.newColumnFamily(
					"Serializer1", 
					StringSerializer.get(),
					LongSerializer.get());

	@BeforeClass
	public static void init() throws Exception {
		initContext();

		keyspace.prepareQuery(CF_SERIALIZER1)
		.withCql("CREATE TABLE astyanaxunittests.serializer1 (key text, column1 bigint, value text, PRIMARY KEY (key))")
		.execute();

		CF_SERIALIZER1.describe(keyspace);
	}

	@AfterClass
	public static void tearDown() throws Exception {
		keyspace.dropColumnFamily(CF_SERIALIZER1);
	}

	@Test
	public void testSerializer() throws Exception {

		keyspace.prepareQuery(CF_SERIALIZER1)
		.withCql("select * from astyanaxunittests.serializer1")
		.execute();

		SerializerPackage serializer = keyspace.getSerializerPackage("Serializer1", false);

		System.out.println("");
		System.out.println("KeySerializer: " + serializer.getKeySerializer());
		System.out.println("ColumnNameSerializer: " + serializer.getColumnNameSerializer());
		System.out.println("ColumnSerializer: " + serializer.getColumnSerializer());
		System.out.println("DefaultValueSerializer: " + serializer.getDefaultValueSerializer());
		System.out.println("ValueSerializer: " + serializer.getValueSerializer());

		String ss1 = "ss1"; 
		ByteBuffer bb1 = StringSerializer.get().fromString(ss1);
		String ss1Result = serializer.getKeySerializer().getString(bb1);

		System.out.println("ss1Result: " + ss1Result);
		Assert.assertEquals(ss1, ss1Result);

		SpecificCompositeSerializer comp = (SpecificCompositeSerializer) serializer.getColumnNameSerializer();
		System.out.println(comp.getComparators().toString());

		Composite dc = new Composite(ss1);

		List<AbstractType<?>> types = new ArrayList<AbstractType<?>>();
		types.add(UTF8Type.instance);

		CompositeType c1 = CompositeType.getInstance(types);

		SpecificCompositeSerializer ccSerializer = new SpecificCompositeSerializer(c1);
		ByteBuffer bb2 = ccSerializer.toByteBuffer(dc);

		Composite c2 = (Composite) serializer.getColumnNameSerializer().fromByteBuffer(bb2);
		ss1Result =  (String) c2.get(0);

		Assert.assertEquals(ss1, ss1Result);
	}
}