opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/AbstractActorTest.java

   1 /*
   2  * Copyright (c) 2014 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8
   9 package org.opendaylight.controller.cluster.datastore;
  10
  11 import akka.actor.ActorSystem;
  12 import akka.testkit.JavaTestKit;
  13 import org.apache.commons.io.FileUtils;
  14 import org.junit.AfterClass;
  15 import org.junit.BeforeClass;
  16
  17 import java.io.File;
  18 import java.io.IOException;
  19
  20 public abstract class AbstractActorTest {
  21     private static ActorSystem system;
  22
  23     @BeforeClass
  24     public static void setUpClass() throws IOException {
  25
  26         System.setProperty("shard.persistent", "false");
  27         system = ActorSystem.create("test");
  28
  29         deletePersistenceFiles();
  30     }
  31
  32     @AfterClass
  33     public static void tearDownClass() throws IOException {
  34         JavaTestKit.shutdownActorSystem(system);
  35         system = null;
  36
  37         deletePersistenceFiles();
  38     }
  39
  40     protected static void deletePersistenceFiles() throws IOException {
  41         File journal = new File("journal");
  42
  43         if(journal.exists()) {
  44             FileUtils.deleteDirectory(journal);
  45         }
  46
  47         File snapshots = new File("snapshots");
  48
  49         if(snapshots.exists()){
  50             FileUtils.deleteDirectory(snapshots);
  51         }
  52
  53     }
  54
  55     protected ActorSystem getSystem() {
  56         return system;
  57     }
  58
  59 }