Tests for Multiple javascript enrichments

modules/common/src/test/scala/com.snowplowanalytics.snowplow.enrich.common/enrichments/EnrichmentManagerSpec.scala

@@ -903,6 +903,57 @@ class EnrichmentManagerSpec extends Specification with EitherMatchers with CatsE
       }
     }
 
+    "run multiple JavaScript enrichments" >> {
+      val script1 =
+        """
+        function process(event) {
+          event.setApp_id("test_app_id");
+          return [];
+        }"""
+
+      val script2 =
+        """
+        function process(event) {
+          event.setPlatform("test_platform");
+          return [];
+        }"""
+
+      val schemaKey = SchemaKey(
+        "com.snowplowanalytics.snowplow",
+        "javascript_script_config",
+        "jsonschema",
+        SchemaVer.Full(1, 0, 0)
+      )
+      val enrichmentReg = EnrichmentRegistry[IO](
+        javascriptScript = List(
+          JavascriptScriptEnrichment(schemaKey, script1),
+          JavascriptScriptEnrichment(schemaKey, script2)
+        )
+      )
+
+      val parameters = Map(
+        "e" -> "pp",
+        "tv" -> "js-0.13.1",
+        "p" -> "web"
+      ).toOpt
+      val rawEvent = RawEvent(api, parameters, None, source, context)
+      val enriched = EnrichmentManager.enrichEvent[IO](
+        enrichmentReg,
+        client,
+        processor,
+        timestamp,
+        rawEvent,
+        AcceptInvalid.featureFlags,
+        IO.unit,
+        SpecHelpers.registryLookup,
+        atomicFieldLimits
+      )
+      enriched.value.map { e =>
+        (e.map(_.app_id) must beRight("test_app_id")) and
+          (e.map(_.platform) must beRight("test_platform"))
+      }
+    }
+
     "emit an EnrichedEvent with superseded schemas" >> {
       val expectedContexts = jparse(
         """

modules/common/src/test/scala/com.snowplowanalytics.snowplow.enrich.common/enrichments/EnrichmentRegistrySpec.scala

@@ -0,0 +1,173 @@
+/*
+ * Copyright (c) 2012-present Snowplow Analytics Ltd.
+ * All rights reserved.
+ *
+ * This software is made available by Snowplow Analytics, Ltd.,
+ * under the terms of the Snowplow Limited Use License Agreement, Version 1.0
+ * located at https://docs.snowplow.io/limited-use-license-1.0
+ * BY INSTALLING, DOWNLOADING, ACCESSING, USING OR DISTRIBUTING ANY PORTION
+ * OF THE SOFTWARE, YOU AGREE TO THE TERMS OF SUCH LICENSE AGREEMENT.
+ */
+package com.snowplowanalytics.snowplow.enrich.common.enrichments
+
+import org.specs2.Specification
+
+import cats.effect.IO
+import cats.effect.testing.specs2.CatsEffect
+
+import io.circe.Json
+import io.circe.literal._
+
+import com.snowplowanalytics.snowplow.enrich.common.enrichments.registry.EnrichmentConf
+import com.snowplowanalytics.snowplow.enrich.common.utils.ConversionUtils
+import com.snowplowanalytics.snowplow.enrich.common.SpecHelpers
+
+class EnrichmentRegistrySpec extends Specification with CatsEffect {
+  import EnrichmentRegistrySpec._
+
+  def is = s2"""
+  EnrichmentRegistry should parse array of enrichments without any JS enrichment correctly $noJSEnrichment
+  EnrichmentRegistry should parse array of enrichments with single JS enrichment correctly $singleJSEnrichment
+  EnrichmentRegistry should parse array of enrichments with multiple JS enrichments correctly $multipleJSEnrichments
+  """
+
+  def noJSEnrichment =
+    EnrichmentRegistry
+      .parse[IO](
+        enrichmentConfig(),
+        SpecHelpers.client,
+        localMode = false,
+        SpecHelpers.registryLookup
+      )
+      .map { res =>
+        val jsConfs = res.getOrElse(List.empty).filter {
+          case _: EnrichmentConf.JavascriptScriptConf => true
+          case _ => false
+        }
+        jsConfs.size must beEqualTo(0)
+      }
+
+  def singleJSEnrichment = {
+    val jsEnrichments = List(jsEnrichment())
+    EnrichmentRegistry
+      .parse[IO](
+        enrichmentConfig(jsEnrichments),
+        SpecHelpers.client,
+        localMode = false,
+        SpecHelpers.registryLookup
+      )
+      .map { res =>
+        val jsConfs = res.getOrElse(List.empty).filter {
+          case _: EnrichmentConf.JavascriptScriptConf => true
+          case _ => false
+        }
+        jsConfs.size must beEqualTo(1)
+      }
+  }
+
+  def multipleJSEnrichments = {
+    val jsReturns = List("return1", "return2")
+    val jsEnrichments = jsReturns.map(jsEnrichment)
+    EnrichmentRegistry
+      .parse[IO](
+        enrichmentConfig(jsEnrichments),
+        SpecHelpers.client,
+        localMode = false,
+        SpecHelpers.registryLookup
+      )
+      .map { res =>
+        val jsConfs = res.getOrElse(List.empty).flatMap {
+          case e: EnrichmentConf.JavascriptScriptConf => Some(e)
+          case _ => None
+        }
+        jsReturns.zip(jsConfs).forall {
+          case (jsReturn, jsConf) => jsConf.rawFunction should contain(jsReturn)
+        }
+      }
+  }
+}
+
+object EnrichmentRegistrySpec {
+
+  def jsEnrichment(jsReturn: String = "defaultReturn"): Json = {
+    val script = s"""
+      function process(event) {
+        return $jsReturn;
+      }
+    """
+
+    json"""{
+      "schema": "iglu:com.snowplowanalytics.snowplow/javascript_script_config/jsonschema/1-0-0",
+      "data": {
+        "vendor": "com.snowplowanalytics.snowplow",
+        "name": "javascript_script_config",
+        "enabled": true,
+        "parameters": {
+          "script": ${ConversionUtils.encodeBase64Url(script)}
+        }
+      }
+    }"""
+  }
+
+  // Vendor and name are intentionally tweaked in the first enrichment
+  // to test that we are no longer validating them (users were confused about such validation)
+  def enrichmentConfig(additionals: List[Json] = List.empty) = {
+    val enrichmentArr = json"""[
+      {
+        "schema": "iglu:com.snowplowanalytics.snowplow/anon_ip/jsonschema/1-0-0",
+        "data": {
+          "vendor": "com.snowplowanalytics.snowplow_custom",
+          "name": "anon_ip_custom",
+          "enabled": true,
+          "parameters": {
+            "anonOctets": 1
+          }
+        }
+      },
+      {
+        "schema": "iglu:com.snowplowanalytics.snowplow/campaign_attribution/jsonschema/1-0-0",
+        "data": {
+          "vendor": "com.snowplowanalytics.snowplow",
+          "name": "campaign_attribution",
+          "enabled": true,
+          "parameters": {
+            "mapping": "static",
+            "fields": {
+            "mktMedium": ["utm_medium", "medium"],
+            "mktSource": ["utm_source", "source"],
+            "mktTerm": ["utm_term", "legacy_term"],
+            "mktContent": ["utm_content"],
+            "mktCampaign": ["utm_campaign", "cid", "legacy_campaign"]
+            }
+          }
+        }
+      },
+      {
+        "schema": "iglu:com.snowplowanalytics.snowplow/user_agent_utils_config/jsonschema/1-0-0",
+        "data": {
+          "vendor": "com.snowplowanalytics.snowplow",
+          "name": "user_agent_utils_config",
+          "enabled": true,
+            "parameters": {}
+        }
+      },
+      {
+        "schema": "iglu:com.snowplowanalytics.snowplow/referer_parser/jsonschema/2-0-0",
+        "data": {
+          "vendor": "com.snowplowanalytics.snowplow",
+          "name": "referer_parser",
+          "enabled": true,
+          "parameters": {
+            "internalDomains": ["www.subdomain1.snowplowanalytics.com"],
+            "database": "referer-tests.json",
+            "uri": "http://snowplow.com"
+          }
+        }
+      }
+    ]""".asArray.map(_.toList).getOrElse(List.empty) ::: additionals
+    json"""{
+      "schema": "iglu:com.snowplowanalytics.snowplow/enrichments/jsonschema/1-0-0",
+      "data": $enrichmentArr
+    }"""
+  }
+}