HappyDroidTestRunner.java

/*
 * Copyright (c) 2012. HappyDroids LLC, All rights reserved.
 */

package com.happydroids.sparky;

import com.happydroids.HttpTestHelper;
import com.happydroids.server.HappyDroidService;
import com.happydroids.server.TestHappyDroidService;
import org.junit.runners.BlockJUnit4ClassRunner;
import org.junit.runners.model.FrameworkMethod;
import org.junit.runners.model.InitializationError;
import org.junit.runners.model.Statement;

public class HappyDroidTestRunner extends BlockJUnit4ClassRunner {
  public HappyDroidTestRunner(Class<?> klass) throws InitializationError {
    super(klass);
  }

  @Override
  protected Statement methodBlock(FrameworkMethod method) {
    final Statement statement = super.methodBlock(method);
    return new Statement() {
      @Override
      public void evaluate() throws Throwable {
        try {
          beforeTestRun();
          statement.evaluate();
        } finally {
          afterTestRun();
        }
      }
    };
  }

  protected void beforeTestRun() {
    HttpTestHelper.resetInstance();
    HappyDroidService.setInstance(new TestHappyDroidService());
  }

  protected void afterTestRun() {
  }
}