From 08a4c3727a13991ec0fe454ce13749df1947ad45 Mon Sep 17 00:00:00 2001
From: Alexander Olofsson <alexander.olofsson@liu.se>
Date: Fri, 1 Sep 2023 13:56:38 +0200
Subject: [PATCH] Add some simple tests

---
 test/factories/wds_facet.rb               |  3 ++-
 test/models/foreman_wds/wds_facet_test.rb | 32 +++++++++++++++++++++++
 test/models/host/managed_test.rb          |  0
 3 files changed, 34 insertions(+), 1 deletion(-)
 create mode 100644 test/models/foreman_wds/wds_facet_test.rb
 create mode 100644 test/models/host/managed_test.rb

diff --git a/test/factories/wds_facet.rb b/test/factories/wds_facet.rb
index c8b00af..cb0f939 100644
--- a/test/factories/wds_facet.rb
+++ b/test/factories/wds_facet.rb
@@ -3,6 +3,7 @@
 FactoryBot.define do
   factory :wds_facet, class: 'ForemanWds::WdsFacet' do
     host
-    wds_server
+
+    install_image_name { 'install.wim' }
   end
 end
diff --git a/test/models/foreman_wds/wds_facet_test.rb b/test/models/foreman_wds/wds_facet_test.rb
new file mode 100644
index 0000000..4c8a17c
--- /dev/null
+++ b/test/models/foreman_wds/wds_facet_test.rb
@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+require 'test_plugin_helper'
+
+module ForemanWds
+  class WDSFacetTest < ActiveSupport::TestCase
+    let(:wds_server) do
+      FactoryBot.build(:wds_server)
+    end
+    let(:host) do
+      FactoryBot.build(:host, :managed, :with_wds_facet) do
+        host.wds_facet.wds_server = wds_server
+      end
+    end
+
+    context 'without WDS server' do
+      let(:wds_server) { nil }
+
+      it 'does not error' do
+        assert_nil host.wds_facet.boot_image
+        assert_nil host.wds_facet.install_image
+      end
+    end
+
+    context 'with WDS server' do
+      it 'does not error' do
+        assert_nil host.wds_facet.boot_image
+        assert_nil host.wds_facet.install_image
+      end
+    end
+  end
+end
diff --git a/test/models/host/managed_test.rb b/test/models/host/managed_test.rb
new file mode 100644
index 0000000..e69de29
-- 
GitLab