Add kvm and nutanix providers (#834)

* Add kvm and nutanix providers * Update tests
trento-project · Sep 21, 2022 · 53afee4 · 53afee4
1 parent ce175e5
commit 53afee4
Show file tree

Hide file tree

Showing 6 changed files with 47 additions and 3 deletions.
diff --git a/lib/trento/application/integration/discovery/payloads/cloud_discovery_payload.ex b/lib/trento/application/integration/discovery/payloads/cloud_discovery_payload.ex
@@ -9,7 +9,9 @@ defmodule Trento.Integration.Discovery.CloudDiscoveryPayload do
   import PolymorphicEmbed, only: [cast_polymorphic_embed: 3]
 
   deftype do
-    field :provider, Ecto.Enum, values: [:aws, :gcp, :azure, :unknown], default: :unknown
+    field :provider, Ecto.Enum,
+      values: [:aws, :gcp, :azure, :kvm, :nutanix, :unknown],
+      default: :unknown
 
     field :metadata, PolymorphicEmbed,
       types: [

diff --git a/lib/trento/domain/host/commands/update_provider.ex b/lib/trento/domain/host/commands/update_provider.ex
@@ -17,7 +17,7 @@ defmodule Trento.Domain.Commands.UpdateProvider do
 
   defcommand do
     field :host_id, Ecto.UUID
-    field :provider, Ecto.Enum, values: [:azure, :aws, :gcp, :unknown]
+    field :provider, Ecto.Enum, values: [:azure, :aws, :gcp, :kvm, :nutanix, :unknown]
 
     field :provider_data, PolymorphicEmbed,
       types: [

diff --git a/lib/trento/domain/host/events/provider_updated.ex b/lib/trento/domain/host/events/provider_updated.ex
@@ -15,7 +15,7 @@ defmodule Trento.Domain.Events.ProviderUpdated do
 
   defevent do
     field :host_id, Ecto.UUID
-    field :provider, Ecto.Enum, values: [:azure, :aws, :gcp, :unknown]
+    field :provider, Ecto.Enum, values: [:azure, :aws, :gcp, :kvm, :nutanix, :unknown]
 
     field :provider_data, PolymorphicEmbed,
       types: [

diff --git a/test/fixtures/discovery/cloud_discovery_kvm.json b/test/fixtures/discovery/cloud_discovery_kvm.json
@@ -0,0 +1,8 @@
+{
+  "discovery_type": "cloud_discovery",
+  "agent_id": "0a055c90-4cb6-54ce-ac9c-ae3fedaf40d4",
+  "payload": {
+    "Provider": "kvm",
+    "Metadata": {}
+  }
+}
diff --git a/test/fixtures/discovery/cloud_discovery_nutanix.json b/test/fixtures/discovery/cloud_discovery_nutanix.json
@@ -0,0 +1,8 @@
+{
+  "discovery_type": "cloud_discovery",
+  "agent_id": "0a055c90-4cb6-54ce-ac9c-ae3fedaf40d4",
+  "payload": {
+    "Provider": "nutanix",
+    "Metadata": {}
+  }
+}
diff --git a/test/trento/application/integration/discovery/policies/host_policy_test.exs b/test/trento/application/integration/discovery/policies/host_policy_test.exs
@@ -119,6 +119,32 @@ defmodule Trento.Integration.Discovery.HostPolicyTest do
              |> HostPolicy.handle()
   end
 
+  test "should return the expected commands when a cloud_discovery payload with an kvm provider is handled" do
+    assert {
+             :ok,
+             %UpdateProvider{
+               host_id: "0a055c90-4cb6-54ce-ac9c-ae3fedaf40d4",
+               provider: :kvm
+             }
+           } ==
+             "cloud_discovery_kvm"
+             |> load_discovery_event_fixture()
+             |> HostPolicy.handle()
+  end
+
+  test "should return the expected commands when a cloud_discovery payload with an nutanix provider is handled" do
+    assert {
+             :ok,
+             %UpdateProvider{
+               host_id: "0a055c90-4cb6-54ce-ac9c-ae3fedaf40d4",
+               provider: :nutanix
+             }
+           } ==
+             "cloud_discovery_nutanix"
+             |> load_discovery_event_fixture()
+             |> HostPolicy.handle()
+  end
+
   test "should return the expected commands when a cloud_discovery payload with an unknown provider is handled" do
     assert {
              :ok,