[agent6][spec] adding RHEL and Ubuntu repo tests

Author: truthbk

spec/classes/datadog_agent_redhat_spec.rb

@@ -13,6 +13,8 @@
   context 'with manage_repo => true' do
     let(:params){ {:manage_repo => true} }
     it do
+      should contain_yumrepo('datadog-beta')
+        .with_ensure('absent')
       should contain_yumrepo('datadog')
         .with_enabled(1)\
         .with_gpgcheck(1)\
@@ -28,6 +30,55 @@
   end
 
 
+  # it should install the packages
+  it do
+    should contain_package('datadog-agent-base')\
+      .with_ensure('absent')\
+      .that_comes_before('Package[datadog-agent]')
+  end
+  it do
+    should contain_package('datadog-agent')\
+      .with_ensure('latest')
+  end
+
+  # it should be able to start the service and enable the service by default
+  it do
+    should contain_service('datadog-agent')\
+      .that_requires('Package[datadog-agent]')
+  end
+end
+
+describe 'datadog_agent::redhat::agent6' do
+  let(:facts) do
+    {
+      osfamily: 'redhat',
+      operatingsystem: 'Fedora',
+      architecture: 'x86_64'
+    }
+  end
+
+  # it should install the mirror
+  context 'with manage_repo => true' do
+    let(:params){ {:manage_repo => true} }
+    it do
+      should contain_yumrepo('datadog')
+        .with_ensure('absent')
+      should contain_yumrepo('datadog-beta')
+        .with_enabled(1)\
+        .with_gpgcheck(1)\
+          .with_gpgkey('https://yum.datadoghq.com/DATADOG_RPM_KEY.public')\
+        .with_baseurl('https://yum.datadoghq.com/beta/x86_64/')
+    end
+  end
+  context 'with manage_repo => false' do
+    let(:params){ {:manage_repo => false} }
+    it do
+      should_not contain_yumrepo('datadog-beta')
+      should_not contain_yumrepo('datadog-beta')
+    end
+  end
+
+
   # it should install the packages
   it do
     should contain_package('datadog-agent-base')\

spec/classes/datadog_agent_ubuntu_spec.rb

@@ -9,6 +9,8 @@
   end
 
   it do
+    contain_file('/etc/apt/sources.list.d/datadog-beta.list')
+      .with_ensure('absent')
     contain_file('/etc/apt/sources.list.d/datadog.list')\
       .with_content(%r{deb\s+https://apt.datadoghq.com/\s+stable\s+main})
   end
@@ -18,29 +20,77 @@
   it { should contain_datadog_agent__ubuntu__install_key('A2923DFF56EDA6E76E55E492D3A80E30382E94DE') }
   it do
     should contain_file('/etc/apt/sources.list.d/datadog.list')\
-      .that_notifies('Exec[datadog_apt-get_update]')
+      .that_notifies('exec[datadog_apt-get_update]')
   end
   it { should contain_exec('datadog_apt-get_update') }
 
   # it should install the packages
   it do
     should contain_package('apt-transport-https')\
-      .that_comes_before('File[/etc/apt/sources.list.d/datadog.list]')
+      .that_comes_before('file[/etc/apt/sources.list.d/datadog.list]')
   end
   it do
     should contain_package('datadog-agent-base')\
       .with_ensure('absent')\
-      .that_comes_before('Package[datadog-agent]')
+      .that_comes_before('package[datadog-agent]')
   end
   it do
     should contain_package('datadog-agent')\
-      .that_requires('File[/etc/apt/sources.list.d/datadog.list]')\
-      .that_requires('Exec[datadog_apt-get_update]')
+      .that_requires('file[/etc/apt/sources.list.d/datadog.list]')\
+      .that_requires('exec[datadog_apt-get_update]')
   end
 
   # it should be able to start the service and enable the service by default
   it do
     should contain_service('datadog-agent')\
-      .that_requires('Package[datadog-agent]')
+      .that_requires('package[datadog-agent]')
+  end
+end
+
+describe 'datadog_agent::ubuntu::agent6' do
+  let(:facts) do
+    {
+      osfamily: 'debian',
+      operatingsystem: 'Ubuntu'
+    }
+  end
+
+  it do
+    contain_file('/etc/apt/sources.list.d/datadog.list')
+      .with_ensure('absent')
+    contain_file('/etc/apt/sources.list.d/datadog-beta.list')\
+      .with_content(%r{deb\s+https://apt.datadoghq.com/\s+beta\s+main})
+  end
+
+  # it should install the mirror
+  it { should contain_datadog_agent__ubuntu__install_key('935F5A436A5A6E8788F0765B226AE980C7A7DA52') }
+  it { should contain_datadog_agent__ubuntu__install_key('A2923DFF56EDA6E76E55E492D3A80E30382E94DE') }
+
+  it do
+    should contain_file('/etc/apt/sources.list.d/datadog-beta.list')\
+      .that_notifies('exec[datadog_apt-get_update]')
+  end
+  it { should contain_exec('datadog_apt-get_update') }
+
+  # it should install the packages
+  it do
+    should contain_package('apt-transport-https')\
+      .that_comes_before('file[/etc/apt/sources.list.d/datadog-beta.list]')
+  end
+  it do
+    should contain_package('datadog-agent-base')\
+      .with_ensure('absent')\
+      .that_comes_before('package[datadog-agent]')
+  end
+  it do
+    should contain_package('datadog-agent')\
+      .that_requires('file[/etc/apt/sources.list.d/datadog-beta.list]')\
+      .that_requires('exec[datadog_apt-get_update]')
+  end
+
+  # it should be able to start the service and enable the service by default
+  it do
+    should contain_service('datadog-agent')\
+      .that_requires('package[datadog-agent]')
   end
 end