/**
 *  Copyright (C) 2009-2011 Typesafe Inc. <http://www.typesafe.com>
 */

package akka.remote

import org.scalatest.WordSpec
import org.scalatest.matchers.MustMatchers
import org.scalatest.BeforeAndAfterAll

import akka.util.Duration

trait MultiJvmSync extends WordSpec with MustMatchers with BeforeAndAfterAll {
  def nodes: Int

  override def beforeAll() = {
    onStart()
    MultiJvmSync.start(getClass.getName, nodes)
  }

  def onStart() {}

  override def afterAll() = {
    MultiJvmSync.end(getClass.getName, nodes)
    onEnd()
  }

  def onEnd() {}

  def barrier(name: String, timeout: Duration = FileBasedBarrier.DefaultTimeout) = {
    MultiJvmSync.barrier(name, nodes, getClass.getName, timeout)
  }
}

object MultiJvmSync {
  val TestMarker = "MultiJvm"
  val StartBarrier = "multi-jvm-start"
  val EndBarrier = "multi-jvm-end"

  def start(className: String, count: Int) = barrier(StartBarrier, count, className)

  def end(className: String, count: Int) = barrier(EndBarrier, count, className)

  def testName(className: String) = {
    val i = className.indexOf(TestMarker)
    if (i >= 0) className.substring(0, i) else className
  }

  def nodeName(className: String) = {
    val i = className.indexOf(TestMarker)
    if (i >= 0) className.substring(i + TestMarker.length) else className
  }

  def barrier(name: String, count: Int, className: String, timeout: Duration = FileBasedBarrier.DefaultTimeout) = {
    new FileBasedBarrier(name, count, testName(className), nodeName(className), timeout).await()
  }
}