WIP set up test stack

Author: vlbee

.gitignore

@@ -17,6 +17,7 @@ project/plugins/project/
 dynamodb-local/
 dynamodb-local-common/
 dynamodb-local-schedule-lambda/
+dynamodb-local-live-activities/
 
 # Scala-IDE specific
 .scala_dependencies

build.sbt

@@ -461,6 +461,7 @@ lazy val reportExtractor = lambda("reportextractor", "reportextractor", Some("co
 lazy val liveactivities = lambda("liveactivities", "liveactivities", Some("com.gu.liveactivities.LambdaLocalRun"))
   .dependsOn(common)
   .dependsOn(apiModels  % "test->test", apiModels  % "compile->compile")
+  .settings(LocalDynamoDBLiveActivities.settings)
   .settings(
     libraryDependencies ++= Seq(
       "com.turo" % "pushy" % "0.13.10",
@@ -472,4 +473,10 @@ lazy val liveactivities = lambda("liveactivities", "liveactivities", Some("com.g
       // Hopefully this workaround can be removed once play-json-extensions either updates to Play 3.0 or is merged into play-json
       ExclusionRule(organization = "com.typesafe.play")
     ),
+    fork := true,
+    startDynamoDBLocal := startDynamoDBLocal.dependsOn(Test / compile).value,
+    Test / test := (Test / test).dependsOn(startDynamoDBLocal).value,
+    Test / testOnly := (Test / testOnly).dependsOn(startDynamoDBLocal).evaluated,
+    Test / testQuick := (Test / testQuick).dependsOn(startDynamoDBLocal).evaluated,
+    Test / testOptions += dynamoDBLocalTestCleanup.value,
   )

liveactivities/src/test/scala/com/gu/liveactivities/DynamodbSpecification.scala

@@ -0,0 +1,59 @@
+package com.gu.liveactivities
+
+import aws.AsyncDynamo
+import com.amazonaws.auth.{
+  AWSCredentials,
+  AWSCredentialsProvider,
+  AWSCredentialsProviderChain
+}
+import com.amazonaws.client.builder.AwsClientBuilder.EndpointConfiguration
+import com.amazonaws.regions.Regions
+import com.amazonaws.services.dynamodbv2.AmazonDynamoDBAsyncClientBuilder
+import com.amazonaws.services.dynamodbv2.model.{
+  CreateTableRequest,
+  DeleteTableRequest
+}
+import org.specs2.mutable.Specification
+import org.specs2.specification.{BeforeAfterAll, Scope}
+
+trait DynamodbSpecification extends Specification with BeforeAfterAll {
+
+  sequential
+
+  val TableName: String
+
+  def createTableRequest: CreateTableRequest
+
+  val TestEndpoint = "http://localhost:8002"
+
+  override def beforeAll(): Unit = {
+    awsClient.createTable(createTableRequest)
+  }
+
+  override def afterAll(): Unit = {
+    awsClient.deleteTable(new DeleteTableRequest(TableName))
+  }
+
+  private def awsClient = {
+    val chain = new AWSCredentialsProviderChain(new AWSCredentialsProvider {
+      override def refresh(): Unit = {}
+
+      override def getCredentials: AWSCredentials = new AWSCredentials {
+        override def getAWSAccessKeyId: String = ""
+
+        override def getAWSSecretKey: String = ""
+      }
+    })
+
+    val client = AmazonDynamoDBAsyncClientBuilder.standard()
+      .withCredentials(chain)
+      .withEndpointConfiguration( new EndpointConfiguration(TestEndpoint, Regions.EU_WEST_1.getName) )
+      .build
+
+    client
+  }
+
+  trait AsyncDynamoScope extends Scope {
+    val asyncClient = new AsyncDynamo(awsClient)
+  }
+}

liveactivities/src/test/scala/com/gu/liveactivities/LiveActivityChannelRepositorySpec.scala

@@ -0,0 +1,70 @@
+package com.gu.liveactivities
+
+
+import com.amazonaws.services.dynamodbv2.model._
+import org.specs2.concurrent.ExecutionEnv
+import org.specs2.mock.Mockito
+import tracking.Repository.RepositoryResult
+
+import scala.concurrent.Future
+import scala.jdk.CollectionConverters._
+
+
+class LiveActivityChannelRepositoryTest (implicit ev: ExecutionEnv) extends DynamodbSpecification with Mockito {
+
+  override val TableName = "test-table"
+
+  "LiveActivityChannelRepository" should {
+    "connect to local DynamoDB and create table" in new RepositoryScope {
+      // This test will pass if the table is created successfully in beforeAll()
+      1 must beEqualTo(1)
+    }
+
+    "save a new channel mapping for an id if it does not exist" in new RepositoryScope with ExampleData {
+      repository.saveMapping(footballMapping).flatMap { _ =>
+        repository.getMappingByActivityId(footballMapping.liveActivityId)
+      } must beEqualTo(Right(Some(footballMapping))).await
+    }
+
+    "Error if trying to save a new channel mapping for an id that already exists" in new RepositoryScope with ExampleData {
+      repository.saveMapping(footballMapping).flatMap { _ =>
+        repository.getMappingByActivityId(footballMapping.liveActivityId)
+      } must beEqualTo(Right(Some(footballMapping))).await
+    }
+
+//    "delete a channel mapping for an activity id if it exists" in new RepositoryScope with ExampleData {
+//    }
+//
+//    "get a channel mapping for an activity id if it exists" in new RepositoryScope with ExampleData {
+//    }
+  }
+
+  trait RepositoryScope extends AsyncDynamoScope {
+    val repository = new LiveActivityChannelRepository(asyncClient, TableName)
+  }
+
+  trait ExampleData {
+    val footballData = FootballLiveActivity(
+      homeTeam = "HomeTeamName",
+      awayTeam = "AwayTeamName",
+      articleUrl = "https://www.theguardian.com/football/test-article-id"
+    )
+
+    val footballMapping = LiveActivityMapping(
+      liveActivityId = "football-1234567",
+      channelId = "test-channel-id",
+      data = Some(footballData)
+    )
+  }
+
+  override def createTableRequest: CreateTableRequest = {
+    val IdField = "liveActivityId"
+
+    new CreateTableRequest(TableName, List(new KeySchemaElement(IdField, KeyType.HASH)).asJava)
+      .withAttributeDefinitions(List(
+        new AttributeDefinition(IdField, ScalarAttributeType.S)
+      ).asJava)
+      .withProvisionedThroughput(new ProvisionedThroughput(5L, 5L))
+  }
+
+}

project/LocalDynamoDBLiveActivities.scala

@@ -0,0 +1,14 @@
+import com.localytics.sbt.dynamodb.DynamoDBLocalKeys
+import com.localytics.sbt.dynamodb.DynamoDBLocalKeys._
+import sbt._
+
+object LocalDynamoDBLiveActivities {
+
+  val settings: Seq[Setting[_]] = DynamoDBLocalKeys.baseDynamoDBSettings ++ Seq(
+    dynamoDBLocalDownloadDir := file("dynamodb-local-live-activities"),
+    dynamoDBLocalInMemory := true,
+    dynamoDBLocalVersion := "latest",
+//    dynamoDBLocalSharedDB := true, // Add this line
+    dynamoDBLocalPort := 8002
+  )
+}