tests: misc. API of TRef

oleg-py · May 5, 2019 · 0554e8a · 0554e8a
1 parent 849b091
commit 0554e8a
Show file tree

Hide file tree

Showing 3 changed files with 61 additions and 15 deletions.
diff --git a/shared/src/test/scala/com/olegpy/stm/APITests.scala b/shared/src/test/scala/com/olegpy/stm/APITests.scala
@@ -1,9 +1,8 @@
 package com.olegpy.stm
 
-import cats.effect.{IO, SyncIO}
+import cats.effect.IO
 import utest._
 import cats.syntax.apply._
-import com.olegpy.stm.results._
 
 object APITests extends TestSuite with BaseIOSuite {
   val tests = Tests {
@@ -12,18 +11,6 @@ object APITests extends TestSuite with BaseIOSuite {
         .map { _ ==> number }
     }
 
-    "TRef.apply" - {
-      TRef(number).flatMap(_.get).result
-        .map { _ ==> STMSuccess(number) }
-    }
-
-    "TRef.in" - {
-      for {
-        tr <- TRef.in[SyncIO](number).toIO
-        x  <- tr.get.commit[IO]
-      } yield assert(x == number)
-    }
-
     "for-comprehension with guards" - {
       def transfer(from: TRef[Int], to: TRef[Int], amt: Int): STM[Unit] =
         for {

diff --git a/shared/src/test/scala/com/olegpy/stm/TRefTests.scala b/shared/src/test/scala/com/olegpy/stm/TRefTests.scala
@@ -0,0 +1,60 @@
+package com.olegpy.stm
+
+import cats.data.State
+import cats.implicits._
+import cats.effect.{IO, SyncIO}
+import com.olegpy.stm.results._
+import utest._
+
+
+object TRefTests extends TestSuite with BaseIOSuite {
+  val tests = Tests {
+
+    "TRef.apply" - {
+      TRef(number).flatMap(_.get).result
+        .map { _ ==> STMSuccess(number) }
+    }
+
+    "TRef.in" - {
+      for {
+        tr <- TRef.in[SyncIO](number).toIO
+        x  <- tr.get.commit[IO]
+      } yield assert(x == number)
+    }
+
+    "TRef#modifyState" - {
+      for {
+        tr <- TRef.in[IO](number)
+        stateOp = State.get[Int] <* State.set(0)
+        n  <- tr.modifyState(stateOp).commit[IO]
+        _ = n ==> number
+        x  <- tr.get.commit[IO]
+      } yield x ==> 0
+    }
+
+    "TRef#access" - {
+      for {
+        tr <- TRef.in[IO](0)
+        _  <- tr.access
+          .filter { case (i, _) => i == 0 }
+          .flatMap { case (_, set) => set(number) >> tr.get }
+          .result.map(_ ==> STMSuccess(number))
+      } yield ()
+    }
+
+    "TRef#tryModifyState, TRef#tryModify and TRef#tryUpdate never fail" - {
+      for {
+        tr <- TRef.in[IO](number)
+        stateOp = State.get[Int] <* State.set(0)
+        _ <- (
+          tr.tryModifyState(stateOp),
+          tr.tryModify(_ => (0, 'a')),
+          tr.tryUpdate(_ + 1)
+        ).tupled.commit[IO].flatMap {
+            case (Some(_), Some(_), true) => IO.unit
+            case _ => fail[Unit]
+        }
+      } yield ()
+    }
+  }
+}
diff --git a/shared/src/test/scala/com/olegpy/stm/misc/TDeferredTests.scala b/shared/src/test/scala/com/olegpy/stm/misc/TDeferredTests.scala
@@ -1,7 +1,6 @@
 package com.olegpy.stm.misc
 
 import cats.effect.IO
-import cats.effect.concurrent.Deferred
 import cats.implicits._
 import com.olegpy.stm.BaseIOSuite
 import com.olegpy.stm.results._