cassandra/test/org/apache/cassandra/io/SSTableTest.java

package org.apache.cassandra.io;

import org.apache.cassandra.ServerTest;
import org.apache.cassandra.service.StorageService;
import org.apache.cassandra.db.FileStruct;
import org.apache.cassandra.utils.BloomFilter;
import org.apache.commons.collections.CollectionUtils;
import org.testng.annotations.Test;

import java.io.File;
import java.io.IOException;
import java.util.*;

public class SSTableTest extends ServerTest {
    // @Test
    public void testSingleWrite() throws IOException {
        File f = File.createTempFile("sstable", "");
        SSTable ssTable;

        // write test data
        ssTable = new SSTable(f.getParent(), f.getName());
        BloomFilter bf = new BloomFilter(1000, 8);
        Random random = new Random();
        byte[] bytes = new byte[1024];
        random.nextBytes(bytes);

        String key = Integer.toString(1);
        ssTable.append(key, bytes);
        bf.fill(key);
        ssTable.close(bf);

        // TODO this is broken because SST/SequenceFile now assume that only CFs are written

        // verify
        ssTable = new SSTable(f.getPath() + "-Data.db");
        DataInputBuffer bufIn = ssTable.next(key, "Test:C");
        byte[] bytes2 = new byte[1024];
        bufIn.readFully(bytes2);
        assert Arrays.equals(bytes2, bytes);
    }

    // @Test
    public void testManyWrites() throws IOException {
        File f = File.createTempFile("sstable", "");
        SSTable ssTable;

        TreeMap<String, byte[]> map = new TreeMap<String,byte[]>();
        for ( int i = 100; i < 1000; ++i )
        {
            map.put(Integer.toString(i), ("Avinash Lakshman is a good man: " + i).getBytes());
        }

        // write
        ssTable = new SSTable(f.getParent(), f.getName());
        BloomFilter bf = new BloomFilter(1000, 8);
        for (String key: map.navigableKeySet())
        {
            ssTable.append(key, map.get(key));
        }
        ssTable.close(bf);

        // TODO this is broken because SST/SequenceFile now assume that only CFs are written

        // verify
        List<String> keys = new ArrayList(map.keySet());
        Collections.shuffle(keys);
        ssTable = new SSTable(f.getPath() + "-Data.db");
        for (String key: keys)
        {
            DataInputBuffer bufIn = ssTable.next(key, "Test:C");
            byte[] bytes2 = new byte[map.get(key).length];
            bufIn.readFully(bytes2);
            assert Arrays.equals(bytes2, map.get(key));
        }
    }
}