/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements.  See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership.  The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License.  You may obtain a copy of the License at
*
*    http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License 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.
*/
package org.apache.cassandra.db;

import static org.apache.cassandra.Util.column;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;

import java.io.IOException;
import java.nio.ByteBuffer;
import java.util.concurrent.ExecutionException;

import org.apache.cassandra.CleanupHelper;
import org.apache.cassandra.Util;
import org.apache.cassandra.db.commitlog.CommitLog;
import org.apache.cassandra.db.filter.QueryFilter;
import org.apache.cassandra.db.filter.QueryPath;
import org.junit.Test;
import org.apache.cassandra.utils.ByteBufferUtil;

/**
 * Test for the truncate operation.
 *
 * @author Ran Tavory (rantav@gmail.com)
 *
 */
public class RecoveryManagerTruncateTest extends CleanupHelper
{
	@Test
	public void testTruncate() throws IOException, ExecutionException, InterruptedException
	{
		Table table = Table.open("Keyspace1");
		ColumnFamilyStore cfs = table.getColumnFamilyStore("Standard1");

		RowMutation rm;
		ColumnFamily cf;

		// trucate clears memtable
		rm = new RowMutation("Keyspace1", ByteBufferUtil.bytes("keymulti"));
		cf = ColumnFamily.create("Keyspace1", "Standard1");
		cf.addColumn(column("col1", "val1", 1L));
		rm.add(cf);
		rm.apply();

		// Make sure data was written
		assertNotNull(getFromTable(table, "Standard1", "keymulti", "col1"));

		// and now truncate it
		cfs.truncate().get();
		CommitLog.recover();

		// and validate truncation.
		assertNull(getFromTable(table, "Standard1", "keymulti", "col1"));

		// truncate clears sstable
		rm = new RowMutation("Keyspace1", ByteBufferUtil.bytes("keymulti"));
		cf = ColumnFamily.create("Keyspace1", "Standard1");
		cf.addColumn(column("col1", "val1", 1L));
		rm.add(cf);
		rm.apply();
		cfs.forceBlockingFlush();
		cfs.truncate().get();
		CommitLog.recover();
		assertNull(getFromTable(table, "Standard1", "keymulti", "col1"));
	}

	private IColumn getFromTable(Table table, String cfName, String keyName, String columnName)
	{
		ColumnFamily cf;
		ColumnFamilyStore cfStore = table.getColumnFamilyStore(cfName);
		if (cfStore == null)
		{
			return null;
		}
		cf = cfStore.getColumnFamily(QueryFilter.getNamesFilter(
		        Util.dk(keyName), new QueryPath(cfName), ByteBufferUtil.bytes(columnName)));
		if (cf == null)
		{
			return null;
		}
		return cf.getColumn(ByteBufferUtil.bytes(columnName));
	}
}