mirror of
https://github.com/2revoemag/PSF-BotServer.git
synced 2026-03-04 20:50:20 +00:00
289 lines
12 KiB
Scala
289 lines
12 KiB
Scala
|
// Copyright (c) 2017 PSForever
|
||
|
|
package objects
|
||
|
|
|
||
|
|
import akka.actor.{ActorRef, ActorSystem, Props}
|
||
|
|
import net.psforever.objects.entity.IdentifiableEntity
|
||
|
|
import net.psforever.objects.guid.NumberPoolHub
|
||
|
|
import net.psforever.objects.guid.actor.{NumberPoolActor, Register, UniqueNumberSystem, Unregister}
|
||
|
|
import net.psforever.objects.guid.selector.RandomSelector
|
||
|
|
import net.psforever.objects.guid.source.LimitedNumberSource
|
||
|
|
|
||
|
|
import scala.concurrent.duration.Duration
|
||
|
|
import scala.util.{Failure, Success}
|
||
|
|
|
||
|
|
class AllocateNumberPoolActors extends ActorTest(ActorSystem("test")) {
|
||
|
|
"AllocateNumberPoolActors" in {
|
||
|
|
val src : LimitedNumberSource = LimitedNumberSource(6000)
|
||
|
|
val guid : NumberPoolHub = new NumberPoolHub(src)
|
||
|
|
guid.AddPool("pool1", (1001 to 2000).toList)
|
||
|
|
guid.AddPool("pool2", (3001 to 4000).toList)
|
||
|
|
guid.AddPool("pool3", (5001 to 6000).toList)
|
||
|
|
val actorMap = UniqueNumberSystemTest.AllocateNumberPoolActors(guid)
|
||
|
|
assert(actorMap.size == 4)
|
||
|
|
assert(actorMap.get("generic").isDefined) //automatically generated
|
||
|
|
assert(actorMap.get("pool1").isDefined)
|
||
|
|
assert(actorMap.get("pool2").isDefined)
|
||
|
|
assert(actorMap.get("pool3").isDefined)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
class UniqueNumberSystemTest extends ActorTest(ActorSystem("test")) {
|
||
|
|
"UniqueNumberSystem" should {
|
||
|
|
"constructor" in {
|
||
|
|
val src : LimitedNumberSource = LimitedNumberSource(6000)
|
||
|
|
val guid : NumberPoolHub = new NumberPoolHub(src)
|
||
|
|
guid.AddPool("pool1", (1001 to 2000).toList)
|
||
|
|
guid.AddPool("pool2", (3001 to 4000).toList)
|
||
|
|
guid.AddPool("pool3", (5001 to 6000).toList)
|
||
|
|
system.actorOf(Props(classOf[UniqueNumberSystem], guid, UniqueNumberSystemTest.AllocateNumberPoolActors(guid)), "uns")
|
||
|
|
//as long as it constructs ...
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
class UniqueNumberSystemTest1 extends ActorTest(ActorSystem("test")) {
|
||
|
|
class EntityTestClass extends IdentifiableEntity
|
||
|
|
|
||
|
|
"UniqueNumberSystem" should {
|
||
|
|
"Register (success)" in {
|
||
|
|
val src : LimitedNumberSource = LimitedNumberSource(6000)
|
||
|
|
val guid : NumberPoolHub = new NumberPoolHub(src)
|
||
|
|
val pool1 = (1001 to 2000).toList
|
||
|
|
val pool2 = (3001 to 4000).toList
|
||
|
|
val pool3 = (5001 to 6000).toList
|
||
|
|
guid.AddPool("pool1", pool1).Selector = new RandomSelector
|
||
|
|
guid.AddPool("pool2", pool2).Selector = new RandomSelector
|
||
|
|
guid.AddPool("pool3", pool3).Selector = new RandomSelector
|
||
|
|
val uns = system.actorOf(Props(classOf[UniqueNumberSystem], guid, UniqueNumberSystemTest.AllocateNumberPoolActors(guid)), "uns")
|
||
|
|
assert(src.CountUsed == 0)
|
||
|
|
//pool1
|
||
|
|
for(_ <- 1 to 100) {
|
||
|
|
val testObj = new EntityTestClass()
|
||
|
|
uns ! Register(testObj, "pool1")
|
||
|
|
val msg = receiveOne(Duration.create(100, "ms"))
|
||
|
|
assert(msg.isInstanceOf[Success[_]])
|
||
|
|
assert(pool1.contains(testObj.GUID.guid))
|
||
|
|
}
|
||
|
|
//pool2
|
||
|
|
for(_ <- 1 to 100) {
|
||
|
|
val testObj = new EntityTestClass()
|
||
|
|
uns ! Register(testObj, "pool2")
|
||
|
|
val msg = receiveOne(Duration.create(100, "ms"))
|
||
|
|
assert(msg.isInstanceOf[Success[_]])
|
||
|
|
assert(pool2.contains(testObj.GUID.guid))
|
||
|
|
}
|
||
|
|
//pool3
|
||
|
|
for(_ <- 1 to 100) {
|
||
|
|
val testObj = new EntityTestClass()
|
||
|
|
uns ! Register(testObj, "pool3")
|
||
|
|
val msg = receiveOne(Duration.create(100, "ms"))
|
||
|
|
assert(msg.isInstanceOf[Success[_]])
|
||
|
|
assert(pool3.contains(testObj.GUID.guid))
|
||
|
|
}
|
||
|
|
assert(src.CountUsed == 300)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
class UniqueNumberSystemTest2 extends ActorTest(ActorSystem("test")) {
|
||
|
|
class EntityTestClass extends IdentifiableEntity
|
||
|
|
|
||
|
|
"UniqueNumberSystem" should {
|
||
|
|
"Register (success; already registered)" in {
|
||
|
|
val src : LimitedNumberSource = LimitedNumberSource(6000)
|
||
|
|
val guid : NumberPoolHub = new NumberPoolHub(src)
|
||
|
|
guid.AddPool("pool1", (1001 to 2000).toList).Selector = new RandomSelector
|
||
|
|
guid.AddPool("pool2", (3001 to 4000).toList).Selector = new RandomSelector
|
||
|
|
guid.AddPool("pool3", (5001 to 6000).toList).Selector = new RandomSelector
|
||
|
|
val uns = system.actorOf(Props(classOf[UniqueNumberSystem], guid, UniqueNumberSystemTest.AllocateNumberPoolActors(guid)), "uns")
|
||
|
|
val testObj = new EntityTestClass()
|
||
|
|
assert(!testObj.HasGUID)
|
||
|
|
assert(src.CountUsed == 0)
|
||
|
|
|
||
|
|
uns ! Register(testObj, "pool1")
|
||
|
|
val msg1 = receiveOne(Duration.create(100, "ms"))
|
||
|
|
assert(msg1.isInstanceOf[Success[_]])
|
||
|
|
assert(testObj.HasGUID)
|
||
|
|
assert(src.CountUsed == 1)
|
||
|
|
|
||
|
|
val id = testObj.GUID.guid
|
||
|
|
uns ! Register(testObj, "pool2") //different pool; makes no difference
|
||
|
|
val msg2 = receiveOne(Duration.create(100, "ms"))
|
||
|
|
assert(msg2.isInstanceOf[Success[_]])
|
||
|
|
assert(testObj.HasGUID)
|
||
|
|
assert(src.CountUsed == 1)
|
||
|
|
assert(testObj.GUID.guid == id) //unchanged
|
||
|
|
}
|
||
|
|
}
|
||
|
|
//a log.warn should have been generated during this test
|
||
|
|
}
|
||
|
|
|
||
|
|
class UniqueNumberSystemTest3 extends ActorTest(ActorSystem("test")) {
|
||
|
|
class EntityTestClass extends IdentifiableEntity
|
||
|
|
|
||
|
|
"UniqueNumberSystem" should {
|
||
|
|
"Register (failure; no pool)" in {
|
||
|
|
val src : LimitedNumberSource = LimitedNumberSource(6000)
|
||
|
|
val guid : NumberPoolHub = new NumberPoolHub(src)
|
||
|
|
guid.AddPool("pool1", (1001 to 2000).toList).Selector = new RandomSelector
|
||
|
|
guid.AddPool("pool2", (3001 to 4000).toList).Selector = new RandomSelector
|
||
|
|
guid.AddPool("pool3", (5001 to 6000).toList).Selector = new RandomSelector
|
||
|
|
val uns = system.actorOf(Props(classOf[UniqueNumberSystem], guid, UniqueNumberSystemTest.AllocateNumberPoolActors(guid)), "uns")
|
||
|
|
val testObj = new EntityTestClass()
|
||
|
|
assert(!testObj.HasGUID)
|
||
|
|
assert(src.CountUsed == 0)
|
||
|
|
|
||
|
|
uns ! Register(testObj, "pool4")
|
||
|
|
val msg1 = receiveOne(Duration.create(100, "ms"))
|
||
|
|
assert(msg1.isInstanceOf[Failure[_]])
|
||
|
|
assert(!testObj.HasGUID)
|
||
|
|
assert(src.CountUsed == 0)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
class UniqueNumberSystemTest4 extends ActorTest(ActorSystem("test")) {
|
||
|
|
class EntityTestClass extends IdentifiableEntity
|
||
|
|
|
||
|
|
"UniqueNumberSystem" should {
|
||
|
|
"Register (failure; empty pool)" in {
|
||
|
|
val src : LimitedNumberSource = LimitedNumberSource(6000)
|
||
|
|
val guid : NumberPoolHub = new NumberPoolHub(src)
|
||
|
|
guid.AddPool("pool1", (1001 to 2000).toList).Selector = new RandomSelector
|
||
|
|
guid.AddPool("pool2", (3001 to 4000).toList).Selector = new RandomSelector
|
||
|
|
guid.AddPool("pool3", (5001 to 6000).toList).Selector = new RandomSelector
|
||
|
|
guid.AddPool("pool4", 50 :: Nil).Selector = new RandomSelector //list of one element; can not add an empty list
|
||
|
|
val uns = system.actorOf(Props(classOf[UniqueNumberSystem], guid, UniqueNumberSystemTest.AllocateNumberPoolActors(guid)), "uns")
|
||
|
|
|
||
|
|
val testObj1 = new EntityTestClass()
|
||
|
|
uns ! Register(testObj1, "pool4")
|
||
|
|
val msg1 = receiveOne(Duration.create(100, "ms"))
|
||
|
|
assert(msg1.isInstanceOf[Success[_]]) //pool4 is now empty
|
||
|
|
|
||
|
|
val testObj2 = new EntityTestClass()
|
||
|
|
uns ! Register(testObj2, "pool4")
|
||
|
|
val msg2 = receiveOne(Duration.create(100, "ms"))
|
||
|
|
assert(msg2.isInstanceOf[Failure[_]])
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
class UniqueNumberSystemTest5 extends ActorTest(ActorSystem("test")) {
|
||
|
|
class EntityTestClass extends IdentifiableEntity
|
||
|
|
|
||
|
|
"UniqueNumberSystem" should {
|
||
|
|
"Unregister (success)" in {
|
||
|
|
val src : LimitedNumberSource = LimitedNumberSource(6000)
|
||
|
|
val guid : NumberPoolHub = new NumberPoolHub(src)
|
||
|
|
val pool2 = (3001 to 4000).toList
|
||
|
|
guid.AddPool("pool1", (1001 to 2000).toList).Selector = new RandomSelector
|
||
|
|
guid.AddPool("pool2", pool2).Selector = new RandomSelector
|
||
|
|
guid.AddPool("pool3", (5001 to 6000).toList).Selector = new RandomSelector
|
||
|
|
val uns = system.actorOf(Props(classOf[UniqueNumberSystem], guid, UniqueNumberSystemTest.AllocateNumberPoolActors(guid)), "uns")
|
||
|
|
val testObj = new EntityTestClass()
|
||
|
|
assert(!testObj.HasGUID)
|
||
|
|
assert(src.CountUsed == 0)
|
||
|
|
|
||
|
|
uns ! Register(testObj, "pool2")
|
||
|
|
val msg1 = receiveOne(Duration.create(100, "ms"))
|
||
|
|
assert(msg1.isInstanceOf[Success[_]])
|
||
|
|
assert(testObj.HasGUID)
|
||
|
|
assert(pool2.contains(testObj.GUID.guid))
|
||
|
|
assert(src.CountUsed == 1)
|
||
|
|
|
||
|
|
uns ! Unregister(testObj)
|
||
|
|
val msg2 = receiveOne(Duration.create(100, "ms"))
|
||
|
|
assert(msg2.isInstanceOf[Success[_]])
|
||
|
|
assert(!testObj.HasGUID)
|
||
|
|
assert(src.CountUsed == 0)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
class UniqueNumberSystemTest6 extends ActorTest(ActorSystem("test")) {
|
||
|
|
class EntityTestClass extends IdentifiableEntity
|
||
|
|
|
||
|
|
"UniqueNumberSystem" should {
|
||
|
|
"Unregister (success; object not registered at all)" in {
|
||
|
|
val src : LimitedNumberSource = LimitedNumberSource(6000)
|
||
|
|
val guid : NumberPoolHub = new NumberPoolHub(src)
|
||
|
|
guid.AddPool("pool1", (1001 to 2000).toList).Selector = new RandomSelector
|
||
|
|
guid.AddPool("pool2", (3001 to 4000).toList).Selector = new RandomSelector
|
||
|
|
guid.AddPool("pool3", (5001 to 6000).toList).Selector = new RandomSelector
|
||
|
|
val uns = system.actorOf(Props(classOf[UniqueNumberSystem], guid, UniqueNumberSystemTest.AllocateNumberPoolActors(guid)), "uns")
|
||
|
|
val testObj = new EntityTestClass()
|
||
|
|
assert(!testObj.HasGUID)
|
||
|
|
assert(src.CountUsed == 0)
|
||
|
|
|
||
|
|
uns ! Unregister(testObj)
|
||
|
|
val msg1 = receiveOne(Duration.create(100, "ms"))
|
||
|
|
assert(msg1.isInstanceOf[Success[_]])
|
||
|
|
assert(!testObj.HasGUID)
|
||
|
|
assert(src.CountUsed == 0)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
class UniqueNumberSystemTest7 extends ActorTest(ActorSystem("test")) {
|
||
|
|
class EntityTestClass extends IdentifiableEntity
|
||
|
|
|
||
|
|
"UniqueNumberSystem" should {
|
||
|
|
"Unregister (failure; number not in system)" in {
|
||
|
|
val src : LimitedNumberSource = LimitedNumberSource(6000)
|
||
|
|
val guid : NumberPoolHub = new NumberPoolHub(src)
|
||
|
|
guid.AddPool("pool1", (1001 to 2000).toList).Selector = new RandomSelector
|
||
|
|
guid.AddPool("pool2", (3001 to 4000).toList).Selector = new RandomSelector
|
||
|
|
guid.AddPool("pool3", (5001 to 6000).toList).Selector = new RandomSelector
|
||
|
|
val uns = system.actorOf(Props(classOf[UniqueNumberSystem], guid, UniqueNumberSystemTest.AllocateNumberPoolActors(guid)), "uns")
|
||
|
|
val testObj = new EntityTestClass()
|
||
|
|
testObj.GUID = net.psforever.packet.game.PlanetSideGUID(6001) //fake registering; number too high
|
||
|
|
assert(testObj.HasGUID)
|
||
|
|
assert(src.CountUsed == 0)
|
||
|
|
|
||
|
|
uns ! Unregister(testObj)
|
||
|
|
val msg1 = receiveOne(Duration.create(100, "ms"))
|
||
|
|
assert(msg1.isInstanceOf[Failure[_]])
|
||
|
|
assert(testObj.HasGUID)
|
||
|
|
assert(src.CountUsed == 0)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
class UniqueNumberSystemTest8 extends ActorTest(ActorSystem("test")) {
|
||
|
|
class EntityTestClass extends IdentifiableEntity
|
||
|
|
|
||
|
|
"UniqueNumberSystem" should {
|
||
|
|
"Unregister (failure; object is not registered to that number)" in {
|
||
|
|
val src : LimitedNumberSource = LimitedNumberSource(6000)
|
||
|
|
val guid : NumberPoolHub = new NumberPoolHub(src)
|
||
|
|
guid.AddPool("pool1", (1001 to 2000).toList).Selector = new RandomSelector
|
||
|
|
guid.AddPool("pool2", (3001 to 4000).toList).Selector = new RandomSelector
|
||
|
|
guid.AddPool("pool3", (5001 to 6000).toList).Selector = new RandomSelector
|
||
|
|
val uns = system.actorOf(Props(classOf[UniqueNumberSystem], guid, UniqueNumberSystemTest.AllocateNumberPoolActors(guid)), "uns")
|
||
|
|
val testObj = new EntityTestClass()
|
||
|
|
testObj.GUID = net.psforever.packet.game.PlanetSideGUID(3500) //fake registering
|
||
|
|
assert(testObj.HasGUID)
|
||
|
|
assert(src.CountUsed == 0)
|
||
|
|
|
||
|
|
uns ! Unregister(testObj)
|
||
|
|
val msg1 = receiveOne(Duration.create(100, "ms"))
|
||
|
|
assert(msg1.isInstanceOf[Failure[_]])
|
||
|
|
assert(testObj.HasGUID)
|
||
|
|
assert(src.CountUsed == 0)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
object UniqueNumberSystemTest {
|
||
|
|
/**
|
||
|
|
* @see `UniqueNumberSystem.AllocateNumberPoolActors(NumberPoolHub)(implicit ActorContext)`
|
||
|
|
*/
|
||
|
|
def AllocateNumberPoolActors(poolSource : NumberPoolHub)(implicit system : ActorSystem) : Map[String, ActorRef] = {
|
||
|
|
poolSource.Pools.map({ case ((pname, pool)) =>
|
||
|
|
pname -> system.actorOf(Props(classOf[NumberPoolActor], pool), pname)
|
||
|
|
}).toMap
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|