Add initial test for pipeline

upmaru · Oct 25, 2024 · e4a6145 · e4a6145
1 parent 41ce9ee
commit e4a6145
Show file tree

Hide file tree

Showing 6 changed files with 492 additions and 3 deletions.
diff --git a/config/test.exs b/config/test.exs
@@ -4,6 +4,10 @@ config :uplink, Uplink.Data, mode: "pro"
 
 config :uplink, Uplink.Monitors, enabled: false
 
+config :uplink, Uplink.Metrics.Pipeline,
+  producer_module: Broadway.DummyProducer,
+  producer_options: []
+
 config :uplink, Uplink.Repo,
   username:
     System.get_env("UPLINK_DB_USERNAME") || System.get_env("POSTGRES_USERNAME"),

diff --git a/lib/uplink/metrics/pipeline.ex b/lib/uplink/metrics/pipeline.ex
@@ -11,11 +11,21 @@ defmodule Uplink.Metrics.Pipeline do
   require Logger
 
   def start_link(_opts \\ []) do
+    configuration = Application.get_env(:uplink, __MODULE__) || []
+
+    producer_module =
+      Keyword.get(configuration, :producer_module, Uplink.Metrics.Producer)
+
+    producer_options =
+      Keyword.get(configuration, :producer_options,
+        poll_interval: :timer.seconds(15)
+      )
+
     Broadway.start_link(__MODULE__,
       name: __MODULE__,
       context: :metrics,
       producer: [
-        module: {Uplink.Metrics.Producer, [poll_interval: :timer.seconds(15)]},
+        module: {producer_module, producer_options},
         concurrency: 1
       ],
       processors: [

diff --git a/lib/uplink/monitors.ex b/lib/uplink/monitors.ex
@@ -67,7 +67,8 @@ defmodule Uplink.Monitors do
   end
 
   def config(key, default) do
-    Application.get_env(:uplink, __MODULE__)
-    |> Keyword.get(key, default)
+    configuration = Application.get_env(:uplink, __MODULE__) || []
+
+    Keyword.get(configuration, key, default)
   end
 end