Add more tests, introduce TestFetching.withLookup

Author: rtyley

core/src/test/scala/com/gu/etagcaching/FreshnessPolicyTest.scala

@@ -3,6 +3,7 @@ package com.gu.etagcaching
 import com.github.blemale.scaffeine.{AsyncLoadingCache, Scaffeine}
 import com.gu.etagcaching.FreshnessPolicy.{AlwaysWaitForRefreshedValue, TolerateOldValueWhileRefreshing}
 import com.gu.etagcaching.fetching.{ETaggedData, Fetching, MissingOrETagged}
+import com.gu.etagcaching.testkit.TestFetching
 import org.scalatest.OptionValues
 import org.scalatest.concurrent.ScalaFutures
 import org.scalatest.concurrent.TimeLimits.failAfter

core/src/test/scala/com/gu/etagcaching/LoadingTest.scala

@@ -1,45 +1,97 @@
 package com.gu.etagcaching
 
-import com.gu.etagcaching.FreshnessPolicy.TolerateOldValueWhileRefreshing
+import com.gu.etagcaching.FreshnessPolicy.AlwaysWaitForRefreshedValue
 import com.gu.etagcaching.Loading.Update
-import com.gu.etagcaching.fetching.Fetching
-import org.scalatest.OptionValues
+import com.gu.etagcaching.LoadingTest.TestApparatus
+import com.gu.etagcaching.fetching.{ETaggedData, Fetching}
+import com.gu.etagcaching.testkit.{CountingParser, TestFetching}
+import org.scalatest.OptionValues.convertOptionToValuable
+import org.scalatest.concurrent.ScalaFutures.convertScalaFuture
 import org.scalatest.concurrent.{Eventually, ScalaFutures}
 import org.scalatest.flatspec.AnyFlatSpec
 import org.scalatest.matchers.should.Matchers
+import org.scalatest.matchers.should.Matchers.convertToAnyShouldWrapper
+import org.scalatest.{Inside, OptionValues}
 
+import java.time.DayOfWeek
+import java.time.DayOfWeek.{MONDAY, SATURDAY, THURSDAY}
+import java.util.Locale
+import java.util.Locale.{FRANCE, GERMANY, UK}
 import scala.collection.mutable
 import scala.concurrent.ExecutionContext.Implicits.global
-import scala.concurrent.duration._
 
-class LoadingTest extends AnyFlatSpec with Matchers with ScalaFutures with OptionValues with Eventually {
-  "onUpdate" should "give callbacks that allow logging updates" in {
-    val updates: mutable.Buffer[Update[String, Int]] = mutable.Buffer.empty
+object LoadingTest {
+  /**
+   * Uses a mock (optionally mutable) Map 'dataStore'. Provides an instance of [[Loading]] that fetches from that
+   * datastore, and parses using the provided parser (counting how many times that parsing occurs).
+   */
+  class TestApparatus[K, Response, V](dataStore: scala.collection.Map[K, Response])(parser: Response => V) {
+    private val countingParser = new CountingParser[Response, V](parser)
+    val loading: Loading[K, V] = TestFetching.withStubDataStore(dataStore).thenParsing(countingParser)
 
-    val fetching: Fetching[String, Int] = TestFetching.withIncrementingValues
+    def parseCount(): Long = countingParser.count()
+
+    def parsesCountedDuringConditionalLoadOf(k: K, oldV: ETaggedData[V]): Long = {
+      val before = parseCount()
+      loading.fetchThenParseIfNecessary(k, oldV).futureValue.toOption.value shouldBe parser(dataStore(k))
+      parseCount() - before
+    }
+  }
+}
+
+class LoadingTest extends AnyFlatSpec with Matchers with ScalaFutures with OptionValues with Eventually with Inside {
+
+  "Creating a Loading instance from a Fetching instance" should "be done with 'thenParsing'" in {
+    val fetching: Fetching[Locale, String] =
+      TestFetching.withStubDataStore(Map(FRANCE -> "THURSDAY", GERMANY -> "MONDAY"))
+
+    val loading: Loading[Locale, DayOfWeek] = fetching.thenParsing(DayOfWeek.valueOf)
+
+    loading.fetchAndParse(FRANCE).futureValue.toOption.value shouldBe THURSDAY
+    loading.fetchAndParse(GERMANY).futureValue.toOption.value shouldBe MONDAY
+  }
+
+  "fetchThenParseIfNecessary" should "*only* do parsing if fetching found a change in ETag value" in {
+    val dataStore = mutable.Map(UK -> "SATURDAY")
+    val testApparatus = new TestApparatus(dataStore)(parser = DayOfWeek.valueOf)
+
+    inside(testApparatus.loading.fetchAndParse(UK).futureValue) { case initialLoad: ETaggedData[DayOfWeek] =>
+      testApparatus.parseCount() shouldBe 1
+      initialLoad.result shouldBe SATURDAY
+
+      // No additional parse performed, as UK value's ETag unchanged
+      testApparatus.parsesCountedDuringConditionalLoadOf(UK, initialLoad) shouldBe 0
+
+      dataStore(UK) = "MONDAY"
+
+      // UK's ETag changed, we must parse the new value!
+      testApparatus.parsesCountedDuringConditionalLoadOf(UK, initialLoad) shouldBe 1
+    }
+  }
+
+  "onUpdate" should "provide callbacks that allow logging updates" in {
+    val dataStore = mutable.Map(UK -> "SATURDAY")
+    val testApparatus = new TestApparatus(dataStore)(parser = DayOfWeek.valueOf)
+
+    val updates: mutable.Buffer[Update[Locale, DayOfWeek]] = mutable.Buffer.empty
 
     val cache = new ETagCache(
-      fetching.thenParsing(identity).onUpdate { update =>
-        updates.append(update)
-      },
-      TolerateOldValueWhileRefreshing,
-      _.maximumSize(1).refreshAfterWrite(100.millis)
+      testApparatus.loading.onUpdate(update => updates.append(update)),
+      AlwaysWaitForRefreshedValue,
+      _.maximumSize(1)
     )
 
     val expectedUpdates = Seq(
-      Update("key", None, Some(0)),
-      Update("key", Some(0), Some(1))
+      Update(UK, None, Some(SATURDAY)),
+      Update(UK, Some(SATURDAY), Some(MONDAY))
     )
 
-    cache.get("key").futureValue shouldBe Some(0)
+    cache.get(UK).futureValue shouldBe Some(SATURDAY)
     updates shouldBe expectedUpdates.take(1)
 
-    Thread.sleep(105)
-
-    eventually { cache.get("key").futureValue shouldBe Some(1) }
-    updates.toSeq shouldBe expectedUpdates
+    dataStore(UK) = "MONDAY"
 
-    Thread.sleep(105)
-    updates.toSeq shouldBe expectedUpdates // No updates if we're not requesting the key from the cache
+    cache.get(UK).futureValue shouldBe Some(MONDAY)
+    updates shouldBe expectedUpdates
   }
 }

core/src/test/scala/com/gu/etagcaching/TestFetching.scala

@@ -0,0 +1,15 @@
+package com.gu.etagcaching.testkit
+
+import java.util.concurrent.atomic.AtomicLong
+
+class CountingParser[K,V](parser: K => V) extends (K => V) {
+
+  private val counter = new AtomicLong()
+
+  override def apply(key: K): V = {
+    counter.incrementAndGet()
+    parser(key)
+  }
+
+  def count(): Long = counter.get()
+}

core/src/test/scala/com/gu/etagcaching/testkit/CountingParser.scala

@@ -0,0 +1,46 @@
+package com.gu.etagcaching.testkit
+
+import com.gu.etagcaching.fetching.{ETaggedData, Fetching, Missing, MissingOrETagged}
+
+import java.util.concurrent.atomic.AtomicInteger
+import scala.concurrent.{ExecutionContext, Future}
+
+/**
+ * Creates various test-instances of [[Fetching]] for test fixtures.
+ */
+object TestFetching {
+
+  def withIncrementingValues: Fetching[String, Int] = new Fetching[String, Int] {
+    val counter = new AtomicInteger()
+
+    override def fetch(key: String): Future[MissingOrETagged[Int]] = {
+      val count = counter.getAndIncrement()
+      Future.successful(ETaggedData(count.toString, count))
+    }
+    override def fetchOnlyIfETagChanged(key: String, eTag: String): Future[Option[MissingOrETagged[Int]]] =
+      fetch(key).map(Some(_))(ExecutionContext.parasitic)
+  }
+
+  /**
+   * Create a test [[Fetching]] instance with a (possibly impure) `lookup` function.
+   *
+   * The [[Fetching]] instance will use the object's hashcode to create the object's ETag.
+   *
+   * The `lookup` function can return different values for the same input if we want to simulate
+   * a key's value changing over time.
+   */
+  def withLookup[K, V](lookup: K => Option[V]): Fetching[K, V] = new Fetching[K, V] {
+    override def fetch(key: K): Future[MissingOrETagged[V]] = Future.successful {
+      lookup(key).fold[MissingOrETagged[V]](Missing) { value =>
+        ETaggedData(value.hashCode().toString, value)
+      }
+    }
+    override def fetchOnlyIfETagChanged(key: K, ETag: String): Future[Option[MissingOrETagged[V]]] =
+      fetch(key).map {
+        case ETaggedData(ETag, _) => None
+        case other => Some(other)
+      }(ExecutionContext.parasitic)
+  }
+
+  def withStubDataStore[K, V](dataStore: scala.collection.Map[K, V]): Fetching[K, V] = withLookup(dataStore.get)
+}