fix PartitionedAsyncObjectPoolSpec factory can't return the same int twice to two different partitions

lifey · lifey · commit 01d7c70442b7 · 2015-07-28T23:49:24.000+03:00
diff --git a/db-async-common/src/test/scala/com/github/mauricio/async/db/pool/PartitionedAsyncObjectPoolSpec.scala b/db-async-common/src/test/scala/com/github/mauricio/async/db/pool/PartitionedAsyncObjectPoolSpec.scala
@@ -1,5 +1,7 @@
 package com.github.mauricio.async.db.pool
 
+import java.util.concurrent.atomic.AtomicInteger
+
 import org.specs2.mutable.Specification
 import scala.util.Try
 import scala.concurrent.Await
@@ -17,17 +19,16 @@ class PartitionedAsyncObjectPoolSpec extends SpecificationWithJUnit {
 
     val config =
         PoolConfiguration(100, Long.MaxValue, 100, Int.MaxValue)
-
+    private var current = new AtomicInteger
     val factory = new ObjectFactory[Int] {
         var reject = Set[Int]()
         var failCreate = false
-        private var current = 0
+
         def create =
             if (failCreate)
                 throw new IllegalStateException
             else {
-                current += 1
-                current
+                current.incrementAndGet()
             }
         def destroy(item: Int) = {}
         def validate(item: Int) =
