X-Git-Url: https://git.adam-barratt.org.uk/?a=blobdiff_plain;f=3rdparty%2Fmodules%2Fpostgresql%2Fspec%2Funit%2Fpuppet%2Fprovider%2Fpostgresql_conn_validator%2Fruby_spec.rb;fp=3rdparty%2Fmodules%2Fpostgresql%2Fspec%2Funit%2Fpuppet%2Fprovider%2Fpostgresql_conn_validator%2Fruby_spec.rb;h=3d678ea03c8e557b7550fc1403a3d0f84a0a610b;hb=a69999e580f8b3abd12446c2d6ad59e517651813;hp=0000000000000000000000000000000000000000;hpb=e7b6b352165009c385c52fcfe5a1055690dbfa4b;p=mirror%2Fdsa-puppet.git

diff --git a/3rdparty/modules/postgresql/spec/unit/puppet/provider/postgresql_conn_validator/ruby_spec.rb b/3rdparty/modules/postgresql/spec/unit/puppet/provider/postgresql_conn_validator/ruby_spec.rb
new file mode 100644
index 000000000..3d678ea03
--- /dev/null
+++ b/3rdparty/modules/postgresql/spec/unit/puppet/provider/postgresql_conn_validator/ruby_spec.rb
@@ -0,0 +1,66 @@
+require 'spec_helper'
+
+describe Puppet::Type.type(:postgresql_conn_validator).provider(:ruby) do
+
+  let(:resource) { Puppet::Type.type(:postgresql_conn_validator).new({
+                                                                   :name => "testname"
+                                                                  }.merge attributes) }
+  let(:provider) { resource.provider }
+
+  let(:attributes) do
+    {
+      :psql_path   => '/usr/bin/psql',
+      :host        => 'db.test.com',
+      :port        => 4444,
+      :db_username => 'testuser',
+      :db_password => 'testpass'
+    }
+  end
+
+  describe '#build_psql_cmd' do
+    it 'contains expected commandline options' do
+      expect(provider.validator.build_psql_cmd).to match /\/usr\/bin\/psql.*--host.*--port.*--username.*/
+    end
+  end
+
+  describe '#parse_connect_settings' do
+    it 'returns array if password is present' do
+      expect(provider.validator.parse_connect_settings).to eq(['PGPASSWORD=testpass'])
+    end
+
+    it 'returns an empty array if password is nil' do
+      attributes.delete(:db_password)
+      expect(provider.validator.parse_connect_settings).to eq([])
+    end
+
+    let(:connect_settings) do
+      {
+        :connect_settings => {
+          :PGPASSWORD => 'testpass',
+          :PGHOST     => 'db.test.com',
+          :PGPORT     => '1234'
+        }
+      }
+    end
+    it 'returns an array of settings' do
+      attributes.delete(:db_password)
+      attributes.merge! connect_settings
+      expect(provider.validator.parse_connect_settings).to eq(['PGPASSWORD=testpass','PGHOST=db.test.com','PGPORT=1234'])
+    end
+  end
+
+  describe '#attempt_connection' do
+    let(:sleep_length) {1}
+    let(:tries) {3}
+    let(:exec) {
+      provider.validator.stub(:execute_command).and_return(true)
+    }
+
+    it 'tries the correct number of times' do
+      expect(provider.validator).to receive(:execute_command).exactly(3).times
+
+      provider.validator.attempt_connection(sleep_length,tries)
+
+    end
+  end
+end