Fix vhost unit tests

voxpupuli · Dec 20, 2024 · 249cdd4 · 249cdd4
1 parent b03e563
commit 249cdd4
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 3 deletions.
diff --git a/manifests/vhost.pp b/manifests/vhost.pp
@@ -34,7 +34,7 @@
 ) {
   include caddy
 
-  if ($ensure == 'present') and !($source or $content) {
+  if ($ensure != 'absent') and !($source or $content) {
     fail('Either $source or $content must be specified when $ensure is "present"')
   }
 

diff --git a/spec/defines/vhost_spec.rb b/spec/defines/vhost_spec.rb
@@ -25,9 +25,37 @@
         end
 
         context 'with config_dir set' do
-          let(:params) { super().merge(config_dir: '/etc/caddy/conf.d') }
+          let(:params) { super().merge(config_dir: '/etc/caddy/sites-available') }
 
-          it { is_expected.to contain_file('/etc/caddy/conf.d/example.conf') }
+          it { is_expected.to contain_file('/etc/caddy/sites-available/example.conf').with_ensure('file') }
+        end
+
+        context 'with enable_dir set' do
+          let(:params) { super().merge(enable_dir: '/etc/caddy/sites-enabled') }
+
+          it { is_expected.to contain_file('/etc/caddy/config/example.conf').with_ensure('file') }
+
+          it do
+            is_expected.to contain_file('/etc/caddy/sites-enabled/example.conf').
+              with_ensure('link').
+              with_target('/etc/caddy/config/example.conf')
+          end
+
+          %w[present disabled].each do |ens|
+            context "with ensure => #{ens}" do
+              let(:params) { super().merge(ensure: ens) }
+
+              it { is_expected.to contain_file('/etc/caddy/config/example.conf').with_ensure('file') }
+              it { is_expected.to contain_file('/etc/caddy/sites-enabled/example.conf').with_ensure('absent') }
+            end
+          end
+
+          context 'with ensure => absent' do
+            let(:params) { super().merge(ensure: 'absent') }
+
+            it { is_expected.to contain_file('/etc/caddy/config/example.conf').with_ensure('absent') }
+            it { is_expected.to contain_file('/etc/caddy/sites-enabled/example.conf').with_ensure('absent') }
+          end
         end
 
         context 'with ensure => absent' do