1 files changed, 152 insertions, 0 deletions
diff --git a/code/environments/production/modules/apt/spec/classes/apt_update_spec.rb b/code/environments/production/modules/apt/spec/classes/apt_update_spec.rb
new file mode 100644
index 0000000..675bbd2
--- /dev/null
+++ b/code/environments/production/modules/apt/spec/classes/apt_update_spec.rb
@@ -0,0 +1,152 @@
+require 'spec_helper'
+
+describe 'apt::update', type: :class do
+  context "when apt::update['frequency']='always'" do
+    {
+      'a recent run'                                 => Time.now.to_i,
+      'we are due for a run'                         => 1_406_660_561,
+      'the update-success-stamp file does not exist' => -1,
+    }.each_pair do |desc, factval|
+      context "when $::apt_update_last_success indicates #{desc}" do
+        let(:facts) do
+          {
+            os: { family: 'Debian', name: 'Debian', release: { major: '7', full: '7.0' } },
+            lsbdistid: 'Debian',
+            osfamily: 'Debian',
+            apt_update_last_success: factval,
+            lsbdistcodename: 'wheezy',
+            puppetversion: Puppet.version,
+          }
+        end
+        let(:pre_condition) do
+          "class{'::apt': update => {'frequency' => 'always' },}"
+        end
+
+        it 'triggers an apt-get update run' do
+          # set the apt_update exec's refreshonly attribute to false
+          is_expected.to contain_exec('apt_update').with('refreshonly' => false)
+        end
+      end
+    end
+    context 'when $::apt_update_last_success is nil' do
+      let(:facts) do
+        {
+          os: { family: 'Debian', name: 'Debian', release: { major: '7', full: '7.0' } },
+          lsbdistid: 'Debian',
+          osfamily: 'Debian',
+          lsbdistcodename: 'wheezy',
+          puppetversion: Puppet.version,
+        }
+      end
+      let(:pre_condition) { "class{ '::apt': update => {'frequency' => 'always' },}" }
+
+      it 'triggers an apt-get update run' do
+        # set the apt_update exec\'s refreshonly attribute to false
+        is_expected.to contain_exec('apt_update').with('refreshonly' => false)
+      end
+    end
+  end
+  context "when apt::update['frequency']='reluctantly'" do
+    {
+      'a recent run'                                 => Time.now.to_i,
+      'we are due for a run'                         => 1_406_660_561,
+      'the update-success-stamp file does not exist' => -1,
+    }.each_pair do |desc, factval|
+      context "when $::apt_update_last_success indicates #{desc}" do
+        let(:facts) do
+          {
+            os: { family: 'Debian', name: 'Debian', release: { major: '7', full: '7.0' } },
+            lsbdistid: 'Debian',
+            osfamily: 'Debian',
+            apt_update_last_success: factval,
+            lsbdistcodename: 'wheezy',
+            puppetversion: Puppet.version,
+          }
+        end
+        let(:pre_condition) { "class{ '::apt': update => {'frequency' => 'reluctantly' },}" }
+
+        it 'does not trigger an apt-get update run' do
+          # don't change the apt_update exec's refreshonly attribute. (it should be true)
+          is_expected.to contain_exec('apt_update').with('refreshonly' => true)
+        end
+      end
+    end
+    context 'when $::apt_update_last_success is nil' do
+      let(:facts) do
+        {
+          os: { family: 'Debian', name: 'Debian', release: { major: '7', full: '7.0' } },
+          lsbdistid: 'Debian',
+          osfamily: 'Debian',
+          lsbdistcodename: 'wheezy',
+          puppetversion: Puppet.version,
+        }
+      end
+      let(:pre_condition) { "class{ '::apt': update => {'frequency' => 'reluctantly' },}" }
+
+      it 'does not trigger an apt-get update run' do
+        # don't change the apt_update exec's refreshonly attribute. (it should be true)
+        is_expected.to contain_exec('apt_update').with('refreshonly' => true)
+      end
+    end
+  end
+  %w[daily weekly].each do |update_frequency|
+    context "when apt::update['frequency'] has the value of #{update_frequency}" do
+      { 'we are due for a run' => 1_406_660_561, 'the update-success-stamp file does not exist' => -1 }.each_pair do |desc, factval|
+        context "when $::apt_update_last_success indicates #{desc}" do
+          let(:facts) do
+            {
+              os: { family: 'Debian', name: 'Debian', release: { major: '7', full: '7.0' } },
+              lsbdistid: 'Debian',
+              osfamily: 'Debian',
+              apt_update_last_success: factval,
+              lsbdistcodename: 'wheezy',
+              puppetversion: Puppet.version,
+            }
+          end
+          let(:pre_condition) { "class{ '::apt': update => {'frequency' => '#{update_frequency}',} }" }
+
+          it 'triggers an apt-get update run' do
+            # set the apt_update exec\'s refreshonly attribute to false
+            is_expected.to contain_exec('apt_update').with('refreshonly' => false)
+          end
+        end
+      end
+      context 'when the $::apt_update_last_success fact has a recent value' do
+        let(:facts) do
+          {
+            os: { family: 'Debian', name: 'Debian', release: { major: '7', full: '7.0' } },
+            lsbdistid: 'Debian',
+            osfamily: 'Debian',
+            lsbdistcodename: 'wheezy',
+            apt_update_last_success: Time.now.to_i,
+            puppetversion: Puppet.version,
+          }
+        end
+        let(:pre_condition) { "class{ '::apt': update => {'frequency' => '#{update_frequency}',} }" }
+
+        it 'does not trigger an apt-get update run' do
+          # don't change the apt_update exec\'s refreshonly attribute. (it should be true)
+          is_expected.to contain_exec('apt_update').with('refreshonly' => true)
+        end
+      end
+      context 'when $::apt_update_last_success is nil' do
+        let(:facts) do
+          {
+            os: { family: 'Debian', name: 'Debian', release: { major: '7', full: '7.0' } },
+            lsbdistid: 'Debian',
+            osfamily: 'Debian',
+            lsbdistcodename: 'wheezy',
+            apt_update_last_success: nil,
+            puppetversion: Puppet.version,
+          }
+        end
+        let(:pre_condition) { "class{ '::apt': update => {'frequency' => '#{update_frequency}',} }" }
+
+        it 'triggers an apt-get update run' do
+          # set the apt_update exec\'s refreshonly attribute to false
+          is_expected.to contain_exec('apt_update').with('refreshonly' => false)
+        end
+      end
+    end
+  end
+end