From c95bf293dbe23fa6c5fd9e23b35a4e4ce34da415 Mon Sep 17 00:00:00 2001
From: Li Haoyi <haoyi.sg@gmail.com>
Date: Sun, 3 Nov 2019 17:33:31 +0800
Subject: Flesh out `BatchActor.scala` into its own module, `cask.Actor`. Add
 the first unit test for an asynchronous logging actor

---
 cask/actor/test/src-jvm/JvmActorsTest.scala | 124 ++++++++++++++++++++++++++++
 1 file changed, 124 insertions(+)
 create mode 100644 cask/actor/test/src-jvm/JvmActorsTest.scala

(limited to 'cask/actor/test/src-jvm/JvmActorsTest.scala')

diff --git a/cask/actor/test/src-jvm/JvmActorsTest.scala b/cask/actor/test/src-jvm/JvmActorsTest.scala
new file mode 100644
index 0000000..9cfb0d5
--- /dev/null
+++ b/cask/actor/test/src-jvm/JvmActorsTest.scala
@@ -0,0 +1,124 @@
+package cask.actor
+
+import utest._
+object JvmActorsTest extends TestSuite{
+  def tests = Tests{
+    os.remove.all(os.pwd / "out" / "scratch")
+    test("lock"){
+      val rotateSize = 50
+      val logPath = os.pwd / "out" / "scratch" / "log.txt"
+      val oldPath = os.pwd / "out" / "scratch" / "log-old.txt"
+
+      var logSize = 0
+
+      def logLine(s: String): Unit = {
+        val newLogSize = logSize + s.length + 1
+        if (newLogSize <= rotateSize) logSize = newLogSize
+        else {
+          logSize = 0
+          os.move(logPath, oldPath, replaceExisting = true)
+        }
+
+        os.write.append(logPath, s + "\n", createFolders = true)
+      }
+
+      logLine("I am cow")
+      logLine("hear me moo")
+      logLine("I weight twice as much as you")
+      logLine("And I look good on the barbecue")
+      logLine("Yoghurt curds cream cheese and butter")
+      logLine("Comes from liquids from my udder")
+      logLine("I am cow, I am cow")
+      logLine("Hear me moo, moooo")
+
+      os.read(oldPath).trim() ==> "Yoghurt curds cream cheese and butter\nComes from liquids from my udder"
+      os.read(logPath).trim() ==> "I am cow, I am cow\nHear me moo, moooo"
+    }
+
+    test("actor"){
+      class Logger(log: os.Path, old: os.Path, rotateSize: Int)
+                  (implicit ac: Context) extends SimpleActor[String]{
+        def run(s: String) = {
+          val newLogSize = logSize + s.length + 1
+          if (newLogSize <= rotateSize) logSize = newLogSize
+          else {
+            logSize = s.length
+            os.move(log, old, replaceExisting = true)
+          }
+          os.write.append(log, s + "\n", createFolders = true)
+        }
+        private var logSize = 0
+      }
+
+      implicit val ac = new Context.Test()
+
+      val logPath = os.pwd / "out" / "scratch" / "log.txt"
+      val oldPath  = os.pwd / "out" / "scratch" / "log-old.txt"
+
+      val logger = new Logger(logPath, oldPath, rotateSize = 50)
+
+      logger.send("I am cow")
+      logger.send("hear me moo")
+      logger.send("I weight twice as much as you")
+      logger.send("And I look good on the barbecue")
+      logger.send("Yoghurt curds cream cheese and butter")
+      logger.send("Comes from liquids from my udder")
+      logger.send("I am cow, I am cow")
+      logger.send("Hear me moo, moooo")
+
+      ac.waitForInactivity()
+
+      os.read.lines(oldPath) ==> Seq("Comes from liquids from my udder")
+      os.read.lines(logPath) ==> Seq("I am cow, I am cow", "Hear me moo, moooo")
+    }
+
+    test("pipeline"){
+      class Writer(log: os.Path, old: os.Path, rotateSize: Int)
+                  (implicit ac: Context) extends SimpleActor[String]{
+        def run(s: String) = {
+          val newLogSize = logSize + s.length + 1
+          if (newLogSize <= rotateSize) logSize = newLogSize
+          else {
+            logSize = s.length
+            os.move(log, old, replaceExisting = true)
+          }
+          os.write.append(log, s + "\n", createFolders = true)
+        }
+        private var logSize = 0
+      }
+
+      class Logger(dest: Actor[String])(implicit ac: Context) extends SimpleActor[String]{
+        def run(s: String) = dest.send(java.util.Base64.getEncoder.encodeToString(s.getBytes))
+      }
+
+      implicit val ac = new Context.Test()
+
+      val logPath = os.pwd / "out" / "scratch" / "log.txt"
+      val oldPath  = os.pwd / "out" / "scratch" / "log-old.txt"
+
+      val writer = new Writer(logPath, oldPath, rotateSize = 50)
+      val logger = new Logger(writer)
+
+      logger.send("I am cow")
+      logger.send("hear me moo")
+      logger.send("I weight twice as much as you")
+      logger.send("And I look good on the barbecue")
+      logger.send("Yoghurt curds cream cheese and butter")
+      logger.send("Comes from liquids from my udder")
+      logger.send("I am cow, I am cow")
+      logger.send("Hear me moo, moooo")
+
+      ac.waitForInactivity()
+
+      os.read(oldPath) ==> "Q29tZXMgZnJvbSBsaXF1aWRzIGZyb20gbXkgdWRkZXI=\n"
+      os.read(logPath) ==> "SSBhbSBjb3csIEkgYW0gY293\nSGVhciBtZSBtb28sIG1vb29v\n"
+
+      def decodeFile(p: os.Path) = {
+        os.read.lines(p).map(s => new String(java.util.Base64.getDecoder.decode(s)))
+      }
+
+      decodeFile(oldPath) ==> Seq("Comes from liquids from my udder")
+      decodeFile(logPath) ==> Seq("I am cow, I am cow", "Hear me moo, moooo")
+    }
+  }
+}
\ No newline at end of file
-- 
cgit v1.2.3