Merge pull request #95 from upmaru/feature/ops-639-use-balancer-addre…

…ss-if-exists

Add test for LXD client creation

lib/uplink/clients/lxd.ex

@@ -69,10 +69,25 @@ defmodule Uplink.Clients.LXD do
   def client do
     %{
       "credential" => credential
-    } = Instellar.get_self()
+    } = params = Instellar.get_self()
+
+    balancer = Map.get(params, "balancer")
+
+    endpoint =
+      case Map.get(params, "balancer") do
+        %{"address" => address, "current_state" => "active"} ->
+          uri = URI.parse(credential["endpoint"])
+
+          uri = %{uri | host: address}
+
+          to_string(uri)
+
+        nil ->
+          credential["endpoint"]
+      end
 
     Lexdee.create_client(
-      credential["endpoint"],
+      endpoint,
       credential["certificate"],
       credential["private_key"]
     )

test/uplink/clients/lxd_test.exs

@@ -0,0 +1,75 @@
+defmodule Uplink.Clients.LXDTest do
+  use ExUnit.Case
+
+  alias Uplink.Cache
+
+  alias Uplink.Clients.LXD
+
+  describe "balancer is nil" do
+    setup do
+      Cache.put(:self, %{
+        "balancer" => nil,
+        "credential" => %{
+          "endpoint" => "http://localhost:8443"
+        }
+      })
+    end
+
+    test "use credential endpoint" do
+      assert %Tesla.Client{pre: pre} = LXD.client()
+
+      {_, _, [base_url]} =
+        Enum.find(pre, fn {middleware, _, _} ->
+          middleware == Tesla.Middleware.BaseUrl
+        end)
+
+      assert base_url == "http://localhost:8443"
+    end
+  end
+
+  describe "client without balancer" do
+    setup do
+      Cache.put(:self, %{
+        "credential" => %{
+          "endpoint" => "http://localhost:8443"
+        }
+      })
+    end
+
+    test "use credential endpoint" do
+      assert %Tesla.Client{pre: pre} = LXD.client()
+
+      {_, _, [base_url]} =
+        Enum.find(pre, fn {middleware, _, _} ->
+          middleware == Tesla.Middleware.BaseUrl
+        end)
+
+      assert base_url == "http://localhost:8443"
+    end
+  end
+
+  describe "client with blaancer" do
+    setup do
+      Cache.put(:self, %{
+        "balancer" => %{
+          "address" => "some.address.com",
+          "current_state" => "active"
+        },
+        "credential" => %{
+          "endpoint" => "http://localhost:8443"
+        }
+      })
+    end
+
+    test "use balancer address" do
+      assert %Tesla.Client{pre: pre} = LXD.client()
+
+      {_, _, [base_url]} =
+        Enum.find(pre, fn {middleware, _, _} ->
+          middleware == Tesla.Middleware.BaseUrl
+        end)
+
+      assert base_url == "http://some.address.com:8443"
+    end
+  end
+end