I've committed to trunk all the required code and posted about it, hope you find it useful http://prettyprint.me/2010/02/14/running-cassandra-as-an-embedded-service/


On Sun, Jan 24, 2010 at 12:20 PM, Richard Grossman <richiesgr@gmail.com> wrote:
Great Ran,

I think I've missed the .setDaemon to keep the server alive.
Thanks

Richard

On Sun, Jan 24, 2010 at 12:02 PM, Ran Tavory <rantav@gmail.com> wrote:
Here's the code I've just written over the weekend and started using in test:


package com.outbrain.data.cassandra.service;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;

import org.apache.cassandra.config.DatabaseDescriptor;
import org.apache.cassandra.service.CassandraDaemon;
import org.apache.cassandra.utils.FileUtils;
import org.apache.thrift.transport.TTransportException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * An in-memory cassandra storage service that listens to the thrift interface.
 * Useful for unit testing,
 *
 * @author Ran Tavory (ran@outbain.com)
 *
 */
public class InProcessCassandraServer implements Runnable {

  private static final Logger log = LoggerFactory.getLogger(InProcessCassandraServer.class);

  CassandraDaemon cassandraDaemon;

  public void init() {
    try {
      prepare();
    } catch (IOException e) {
      log.error("Cannot prepare cassandra.", e);
    }
    try {
      cassandraDaemon = new CassandraDaemon();
      cassandraDaemon.init(null);
    } catch (TTransportException e) {
      log.error("TTransportException", e);
    } catch (IOException e) {
      log.error("IOException", e);
    }
  }

  @Override
  public void run() {
    cassandraDaemon.start();
  }

  public void stop() {
    cassandraDaemon.stop();
    rmdir("tmp");
  }


  /**
   * Creates all files and directories needed
   * @throws IOException
   */
  private void prepare() throws IOException {
    // delete tmp dir first
    rmdir("tmp");
    // make a tmp dir and copy storag-conf.xml and log4j.properties to it
    copy("/cassandra/storage-conf.xml", "tmp");
    copy("/cassandra/log4j.properties", "tmp");
    System.setProperty("storage-config", "tmp");

    // make cassandra directories.
    for (String s: DatabaseDescriptor.getAllDataFileLocations()) {
      mkdir(s);
    }
    mkdir(DatabaseDescriptor.getBootstrapFileLocation());
    mkdir(DatabaseDescriptor.getLogFileLocation());
  }

  /**
   * Copies a resource from within the jar to a directory.
   *
   * @param resourceName
   * @param directory
   * @throws IOException
   */
  private void copy(String resource, String directory) throws IOException {
    mkdir(directory);
    InputStream is = getClass().getResourceAsStream(resource);
    String fileName = resource.substring(resource.lastIndexOf("/") + 1);
    File file = new File(directory + System.getProperty("file.separator") + fileName);
    OutputStream out = new FileOutputStream(file);
    byte buf[] = new byte[1024];
    int len;
    while ((len = is.read(buf)) > 0) {
      out.write(buf, 0, len);
    }
    out.close();
    is.close();
  }

  /**
   * Creates a directory
   * @param dir
   * @throws IOException
   */
  private void mkdir(String dir) throws IOException {
    FileUtils.createDirectory(dir);
  }

  /**
   * Removes a directory from file system
   * @param dir
   */
  private void rmdir(String dir) {
    FileUtils.deleteDir(new File(dir));
  }
}


And in the test class:

public class XxxTest {

  private static InProcessCassandraServer cassandra;

  @BeforeClass
  public static void setup() throws TTransportException, IOException, InterruptedException {
    cassandra = new InProcessCassandraServer();
    cassandra.init();
    Thread t = new Thread(cassandra);
    t.setDaemon(true);
    t.start();
  }

  @AfterClass
  public static void shutdown() {
    cassandra.stop();
  }
... test
}

Now you can connect to localhost:9160.

Assumptions:
The code assumes you have two files in your classpath: /cassandra/stogage-config.xml and /cassandra/log4j.xml. This is convenient if you use maven, just throw them at /src/test/resources/cassandra/
If you don't work with maven or would like to configure the configuration files differently it should be fairly easy, just change the prepare() method.



On Sun, Jan 24, 2010 at 10:54 AM, Richard Grossman <richiesgr@gmail.com> wrote:
So Is there anybody ? Unit testing is important people ...
Thanks


On Thu, Jan 21, 2010 at 12:09 PM, Richard Grossman <richiesgr@gmail.com> wrote:
Here is the code I use
    class startServer implements Runnable {

        @Override
        public void run() {
            try {
                CassandraDaemon cassandraDaemon = new CassandraDaemon();
                cassandraDaemon.init(null);
                cassandraDaemon.start();
            } catch (TTransportException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            } catch (IOException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            }               
        }       
    }

        Thread thread = new Thread(new startServer());
        thread.start();

<the code to test here>



On Thu, Jan 21, 2010 at 12:08 PM, Richard Grossman <richiesgr@gmail.com> wrote:
Yes I've seen this and also check it but if I start the server then it block the current thread I can continue the test in sequence.
So I've tried to start into separate thread but no chance too it close the server even before I arrive to the code to test.

If you've a trick to start the server in JVM thank

Richard

On Wed, Jan 20, 2010 at 3:47 PM, Jonathan Ellis <jbellis@gmail.com> wrote:
did you look at CassandraDaemon?