#280: AgentWithServerIntegrationTests should run against local servic…

…e and thus HTTP Dispatcher
AbsaOSS · Oct 2, 2024 · fd2c52f · fd2c52f
1 parent 204f1b1
commit fd2c52f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 4 deletions.
diff --git a/agent/src/test/scala/za/co/absa/atum/agent/AgentWithServerIntegrationTests.scala b/agent/src/test/scala/za/co/absa/atum/agent/AgentWithServerIntegrationTests.scala
@@ -21,6 +21,7 @@ import org.apache.spark.sql.{Row, SparkSession}
 import za.co.absa.atum.agent.model.AtumMeasure.RecordCount
 import za.co.absa.balta.DBTestSuite
 import za.co.absa.balta.classes.JsonBString
+import com.typesafe.config.{Config, ConfigFactory, ConfigValueFactory}
 
 import scala.collection.immutable.ListMap
 
@@ -58,12 +59,21 @@ class AgentWithServerIntegrationTests extends DBTestSuite {
     val rdd = spark.sparkContext.parallelize(testDataForRDD)
     val df = spark.createDataFrame(rdd, testDataSchema)
 
-    // Atum Context Stuff Preparation - Partitioning, Measures, Additional Data, Checkpoint
+    // Atum Agent preparation - Agent configured to work with HTTP Dispatcher and service on localhost
+    val agent = AtumAgent
+    agent.dispatcherFromConfig(
+      ConfigFactory
+        .empty()
+        .withValue("atum.dispatcher.type", ConfigValueFactory.fromAnyRef("http"))
+        .withValue("atum.dispatcher.http.url", ConfigValueFactory.fromAnyRef("http://localhost:8080"))
+    )
+
+    // Atum Context stuff preparation - Partitioning, Measures, Additional Data, Checkpoint
     val domainAtumPartitioning = ListMap(
       "partition1" -> "valueFromTest1",
       "partition2" -> "valueFromTest2"
     )
-    val domainAtumContext = AtumAgent.getOrCreateAtumContext(domainAtumPartitioning)
+    val domainAtumContext = agent.getOrCreateAtumContext(domainAtumPartitioning)
 
     domainAtumContext.addMeasure(RecordCount("*"))
     domainAtumContext.addAdditionalData("author", "Laco")

diff --git a/project/JacocoSetup.scala b/project/JacocoSetup.scala
@@ -49,8 +49,7 @@ object JacocoSetup {
       "za.co.absa.atum.server.Constants*",
       "za.co.absa.atum.server.api.database.DoobieImplicits*",
       "za.co.absa.atum.server.api.database.TransactorProvider*",
-      "za.co.absa.atum.model.dto.*",
-      "za.co.absa.atum.agent.AgentWithServerIntegrationTests"
+      "za.co.absa.atum.model.dto.*"
     )
   }