Specs2 driving me nuts ^_^

gistfile1.sbt

package net.tomasherman.specus.server.grid

import org.specs2.mutable.Specification
import org.specs2.specification.Scope
import akka.actor.Channel
import org.specs2.mock._
import net.tomasherman.specus.common.api.net.Packet

/**
 * This file is part of Specus.
 *
 * Specus is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Specus is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 *
 * You should have received a copy of the GNU General Public License
 * along with Specus.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

class NLBFImpl extends NodeLoadBalancerFunctionality{
  def getNext = useNext
  def setNext(value:Int) {useNext= value}
  def getChannels = nodeChannels
}

trait NodeLoadBalancerFuncionalityScope extends Scope with Mockito {
  val loadBalancer = new NLBFImpl
  val packet = mock[Packet]
  val channel1 = mock[Channel[Any]]
  channel1.!(packet).answers({p=>println("fml")})
  val channel2 = mock[Channel[Any]]
  channel2.!(packet).answers({p=>println("fml2")})
  val channel3 = mock[Channel[Any]]
  channel3.!(packet).answers({p=>println("fml3")})
}

class NodeLoadBalancerFuncionalitySpec extends Specification {
  "NodeLoadBalancerFunctionality" should {
    "register channel correctly" in new NodeLoadBalancerFuncionalityScope {
      loadBalancer.getChannels.size must_== 0
      loadBalancer.registerChannel(channel1)
      loadBalancer.getChannels.size must_== 1
      loadBalancer.getChannels(0) must_== channel1
    }

    "unregister channel correctly" in new NodeLoadBalancerFuncionalityScope {
      loadBalancer.registerChannel(channel1)
      loadBalancer.registerChannel(channel2)
      loadBalancer.registerChannel(channel3)
      loadBalancer.unregisterChannel(channel2)
      loadBalancer.getChannels.contains(channel2) must_== false
    }

    "bang next channel correctly" in new NodeLoadBalancerFuncionalityScope with Mockito{
      loadBalancer.registerChannel(channel1)
      loadBalancer.registerChannel(channel2)
      loadBalancer.registerChannel(channel3)
      loadBalancer.bangNext(packet)
      loadBalancer.bangNext(packet)
      loadBalancer.bangNext(packet)
      loadBalancer.bangNext(packet)
      there was one(channel1).!(packet) then one(channel2).!(packet) then one(channel3).!(packet) then one(channel1).!(packet)
    }

    "bang all channels correctly" in new NodeLoadBalancerFuncionalityScope {
      loadBalancer.registerChannel(channel1)
      loadBalancer.registerChannel(channel2)
      loadBalancer.registerChannel(channel3)

      loadBalancer.bangAll(packet)
      there was one(channel1).!(packet)
      there was one(channel2).!(packet)
      there was one(channel3).!(packet)
    }

    "work well when nextUse channel is removed" in new NodeLoadBalancerFuncionalityScope {
      loadBalancer.registerChannel(channel1)
      loadBalancer.registerChannel(channel2)
      loadBalancer.registerChannel(channel3)

      loadBalancer.setNext(2)
      loadBalancer.unregisterChannel(channel3)
      loadBalancer.bangNext(packet)
      there was one(channel1).!(packet)
      there was no(channel2).!(packet)
      there was one(channel3).!(packet)
    }
  }
}