freighthop 0.0.6 → 0.1.0

data/modules/mysql/spec/system/mysql_server_spec.rb

@@ -0,0 +1,85 @@
+require 'spec_helper_system'
+
+describe 'mysql class' do
+  case node.facts['osfamily']
+  when 'RedHat'
+    package_name = 'mysql-server'
+    service_name = 'mysqld'
+    mycnf        = '/etc/my.cnf'
+  when 'Suse'
+    package_name = 'mysql-community-server'
+    service_name = 'mysql'
+    mycnf        = '/etc/my.cnf'
+  when 'Debian'
+    package_name = 'mysql-server'
+    service_name = 'mysql'
+    mycnf        = '/etc/mysql/my.cnf'
+  end
+
+  describe 'running puppet code' do
+    # Using puppet_apply as a helper
+    it 'should work with no errors' do
+      pp = <<-EOS
+        class { 'mysql::server': }
+      EOS
+
+      # Run it twice and test for idempotency
+      puppet_apply(pp) do |r|
+        r.exit_code.should_not == 1
+        r.refresh
+        r.exit_code.should be_zero
+      end
+    end
+
+    describe package(package_name) do
+      it { should be_installed }
+    end
+
+    describe service(service_name) do
+      it { should be_running }
+      it { should be_enabled }
+    end
+  end
+
+  describe 'mycnf' do
+    it 'should contain sensible values' do
+      pp = <<-EOS
+        class { 'mysql::server': }
+      EOS
+      puppet_apply(pp) do |r|
+        r.exit_code.should_not == 1
+      end
+    end
+
+    describe file(mycnf) do
+      it { should contain 'key_buffer = 16M' }
+      it { should contain 'max_binlog_size = 100M' }
+      it { should contain 'query_cache_size = 16M' }
+    end
+  end
+
+  describe 'my.cnf changes' do
+    it 'sets values' do
+      pp = <<-EOS
+        class { 'mysql::server':
+          override_options => { 'mysqld' => 
+            { 'key_buffer'       => '32M',
+              'max_binlog_size'  => '200M',
+              'query_cache_size' => '32M',
+            }
+          }  
+        }
+      EOS
+      puppet_apply(pp) do |r|
+        r.exit_code.should_not == 1
+      end
+    end
+
+    describe file(mycnf) do
+      it { should contain 'key_buffer = 32M' }
+      it { should contain 'max_binlog_size = 200M' }
+      it { should contain 'query_cache_size = 32M' }
+    end
+  end
+
+end

data/modules/mysql/spec/system/types/mysql_grant_spec.rb

@@ -0,0 +1,314 @@
+require 'spec_helper_system'
+
+describe 'mysql_grant' do
+
+  describe 'setup' do
+    it 'setup mysql::server' do
+      pp = <<-EOS
+        class { 'mysql::server': }
+      EOS
+
+      puppet_apply(pp)
+    end
+  end
+
+  describe 'missing privileges for user' do
+    it 'should fail' do
+      pp = <<-EOS
+        mysql_grant { 'test1@tester/test.*':
+          ensure => 'present',
+          table  => 'test.*',
+          user   => 'test1@tester',
+        }
+      EOS
+
+      puppet_apply(pp) do |r|
+        r.stderr.should =~ /privileges parameter is required/
+      end
+    end
+
+    it 'should not find the user' do
+      shell("mysql -NBe \"SHOW GRANTS FOR test1@tester\"") do |r|
+        r.stderr.should =~ /There is no such grant defined for user 'test1' on host 'tester'/
+        r.exit_code.should eq 1
+      end
+    end
+  end
+
+  describe 'missing table for user' do
+    it 'should fail' do
+      pp = <<-EOS
+        mysql_grant { 'atest@tester/test.*':
+          ensure => 'present',
+          user   => 'atest@tester',
+          privileges => ['ALL'],
+        }
+      EOS
+
+      puppet_apply(pp) do |r|
+        r.exit_code.should eq 1
+      end
+    end
+
+    it 'should not find the user' do
+      shell("mysql -NBe \"SHOW GRANTS FOR atest@tester\"") do |r|
+        r.stderr.should =~ /There is no such grant defined for user 'atest' on host 'tester'/
+        r.exit_code.should eq 1
+      end
+    end
+  end
+
+  describe 'adding privileges' do
+    it 'should work without errors' do
+      pp = <<-EOS
+        mysql_grant { 'test2@tester/test.*':
+          ensure     => 'present',
+          table      => 'test.*',
+          user       => 'test2@tester',
+          privileges => ['SELECT', 'UPDATE'],
+        }
+      EOS
+
+      puppet_apply(pp)
+    end
+
+    it 'should find the user' do
+      shell("mysql -NBe \"SHOW GRANTS FOR test2@tester\"") do |r|
+        r.stdout.should =~ /GRANT SELECT, UPDATE.*TO 'test2'@'tester'/
+        r.stderr.should be_empty
+        r.exit_code.should be_zero
+      end
+    end
+  end
+
+  describe 'adding option' do
+    it 'should work without errors' do
+      pp = <<-EOS
+        mysql_grant { 'test3@tester/test.*':
+          ensure  => 'present',
+          table   => 'test.*',
+          user    => 'test3@tester',
+          options => ['GRANT'],
+          privileges => ['SELECT', 'UPDATE'],
+        }
+      EOS
+
+      puppet_apply(pp)
+    end
+
+    it 'should find the user' do
+      shell("mysql -NBe \"SHOW GRANTS FOR test3@tester\"") do |r|
+        r.stdout.should =~ /GRANT SELECT, UPDATE ON `test`.* TO 'test3'@'tester' WITH GRANT OPTION$/
+        r.stderr.should be_empty
+        r.exit_code.should be_zero
+      end
+    end
+  end
+
+  describe 'adding all privileges without table' do
+    it 'should fail' do
+      pp = <<-EOS
+        mysql_grant { 'test4@tester/test.*':
+          ensure     => 'present',
+          user       => 'test4@tester',
+          options    => ['GRANT'],
+          privileges => ['SELECT', 'UPDATE', 'ALL'],
+        }
+      EOS
+
+      puppet_apply(pp) do |r|
+        r.stderr.should =~ /table parameter is required./
+      end
+    end
+
+  end
+
+
+  describe 'adding all privileges' do
+    it 'should only try to apply ALL' do
+      pp = <<-EOS
+        mysql_grant { 'test4@tester/test.*':
+          ensure     => 'present',
+          table      => 'test.*',
+          user       => 'test4@tester',
+          options    => ['GRANT'],
+          privileges => ['SELECT', 'UPDATE', 'ALL'],
+        }
+      EOS
+
+      puppet_apply(pp)
+    end
+
+    it 'should find the user' do
+      shell("mysql -NBe \"SHOW GRANTS FOR test4@tester\"") do |r|
+        r.stdout.should =~ /GRANT ALL PRIVILEGES ON `test`.* TO 'test4'@'tester' WITH GRANT OPTION/
+        r.stderr.should be_empty
+        r.exit_code.should be_zero
+      end
+    end
+  end
+
+  # Test combinations of user@host to ensure all cases work.
+  describe 'short hostname' do
+    it 'should apply' do
+      pp = <<-EOS
+        mysql_grant { 'test@short/test.*':
+          ensure     => 'present',
+          table      => 'test.*',
+          user       => 'test@short',
+          privileges => 'ALL',
+        }
+        mysql_grant { 'test@long.hostname.com/test.*':
+          ensure     => 'present',
+          table      => 'test.*',
+          user       => 'test@long.hostname.com',
+          privileges => 'ALL',
+        }
+        mysql_grant { 'test@192.168.5.6/test.*':
+          ensure     => 'present',
+          table      => 'test.*',
+          user       => 'test@192.168.5.6',
+          privileges => 'ALL',
+        }
+        mysql_grant { 'test@2607:f0d0:1002:0051:0000:0000:0000:0004/test.*':
+          ensure     => 'present',
+          table      => 'test.*',
+          user       => 'test@2607:f0d0:1002:0051:0000:0000:0000:0004',
+          privileges => 'ALL',
+        }
+        mysql_grant { 'test@::1/128/test.*':
+          ensure     => 'present',
+          table      => 'test.*',
+          user       => 'test@::1/128',
+          privileges => 'ALL',
+        }
+      EOS
+
+      puppet_apply(pp)
+    end
+
+    it 'finds short hostname' do
+      shell("mysql -NBe \"SHOW GRANTS FOR test@short\"") do |r|
+        r.stdout.should =~ /GRANT ALL PRIVILEGES ON `test`.* TO 'test'@'short'/
+        r.stderr.should be_empty
+        r.exit_code.should be_zero
+      end
+    end
+    it 'finds long hostname' do
+      shell("mysql -NBe \"SHOW GRANTS FOR 'test'@'long.hostname.com'\"") do |r|
+        r.stdout.should =~ /GRANT ALL PRIVILEGES ON `test`.* TO 'test'@'long.hostname.com'/
+        r.stderr.should be_empty
+        r.exit_code.should be_zero
+      end
+    end
+    it 'finds ipv4' do
+      shell("mysql -NBe \"SHOW GRANTS FOR 'test'@'192.168.5.6'\"") do |r|
+        r.stdout.should =~ /GRANT ALL PRIVILEGES ON `test`.* TO 'test'@'192.168.5.6'/
+        r.stderr.should be_empty
+        r.exit_code.should be_zero
+      end
+    end
+    it 'finds ipv6' do
+      shell("mysql -NBe \"SHOW GRANTS FOR 'test'@'2607:f0d0:1002:0051:0000:0000:0000:0004'\"") do |r|
+        r.stdout.should =~ /GRANT ALL PRIVILEGES ON `test`.* TO 'test'@'2607:f0d0:1002:0051:0000:0000:0000:0004'/
+        r.stderr.should be_empty
+        r.exit_code.should be_zero
+      end
+    end
+    it 'finds short ipv6' do
+      shell("mysql -NBe \"SHOW GRANTS FOR 'test'@'::1/128'\"") do |r|
+        r.stdout.should =~ /GRANT ALL PRIVILEGES ON `test`.* TO 'test'@'::1\/128'/
+        r.stderr.should be_empty
+        r.exit_code.should be_zero
+      end
+    end
+  end
+
+  describe 'complex test' do
+    it 'setup mysql::server' do
+      pp = <<-EOS
+      $dbSubnet = '10.10.10.%'
+
+      mysql_database { 'foo':
+        ensure => present,
+      }
+
+      exec { 'mysql-create-table':
+        command     => '/usr/bin/mysql -NBe "CREATE TABLE foo.bar (name VARCHAR(20))"',
+        environment => "HOME=${::root_home}",
+        unless      => '/usr/bin/mysql -NBe "SELECT 1 FROM foo.bar LIMIT 1;"',
+        require     => Mysql_database['foo'],
+      }
+
+      Mysql_grant {
+          ensure     => present,
+          options    => ['GRANT'],
+          privileges => ['ALL'],
+          table      => '*.*',
+          require    => [ Mysql_database['foo'], Exec['mysql-create-table'] ],
+      }
+
+      mysql_grant { "user1@${dbSubnet}/*.*":
+          user       => "user1@${dbSubnet}",
+      }
+      mysql_grant { "user2@${dbSubnet}/foo.bar":
+          privileges => ['SELECT', 'INSERT', 'UPDATE'],
+          user       => "user2@${dbSubnet}",
+          table      => 'foo.bar',
+      }
+      mysql_grant { "user3@${dbSubnet}/foo.*":
+          privileges => ['SELECT', 'INSERT', 'UPDATE'],
+          user       => "user3@${dbSubnet}",
+          table      => 'foo.*',
+      }
+      mysql_grant { 'web@%/*.*':
+          user       => 'web@%',
+      }
+      mysql_grant { "web@${dbSubnet}/*.*":
+          user       => "web@${dbSubnet}",
+      }
+      mysql_grant { "web@${fqdn}/*.*":
+          user       => "web@${fqdn}",
+      }
+      mysql_grant { 'web@localhost/*.*':
+          user       => 'web@localhost',
+      }
+      EOS
+
+      puppet_apply(pp) do |r|
+        r.exit_code.should_not == 1
+        r.refresh
+        r.exit_code.should be_zero
+      end
+    end
+  end
+
+  describe 'lower case privileges' do
+    it 'create ALL privs' do
+      pp = <<-EOS
+      mysql_grant { 'lowercase@localhost/*.*':
+          user       => 'lowercase@localhost',
+          privileges => 'ALL',
+          table      => '*.*',
+      }
+      EOS
+
+      puppet_apply(pp)
+    end
+
+    it 'create lowercase all privs' do
+      pp = <<-EOS
+      mysql_grant { 'lowercase@localhost/*.*':
+          user       => 'lowercase@localhost',
+          privileges => 'all',
+          table      => '*.*',
+      }
+      EOS
+
+      puppet_apply(pp) do |r|
+        r.exit_code.should be_zero
+      end
+    end
+  end
+
+end

data/modules/mysql/spec/system/types/mysql_user_spec.rb

@@ -0,0 +1,35 @@
+require 'spec_helper_system'
+
+describe 'mysql_user' do
+
+  describe 'setup' do
+    it 'should work with no errors' do
+      pp = <<-EOS
+        class { 'mysql::server': }
+      EOS
+
+      puppet_apply(pp)
+    end
+  end
+
+  describe 'adding user' do
+    it 'should work without errors' do
+      pp = <<-EOS
+        mysql_user { 'ashp@localhost':
+          password_hash => '6f8c114b58f2ce9e',
+        }
+      EOS
+
+      puppet_apply(pp)
+    end
+
+    it 'should find the user' do
+      shell("mysql -NBe \"select '1' from mysql.user where CONCAT(user, '@', host) = 'ashp@localhost'\"") do |r|
+        r.stdout.should =~ /^1$/
+        r.stderr.should be_empty
+        r.exit_code.should be_zero
+      end
+    end
+  end
+
+end

data/modules/mysql/spec/unit/mysql_password_spec.rb

@@ -0,0 +1,27 @@
+require 'spec_helper'
+
+describe 'the mysql_password function' do
+  before :all do
+    Puppet::Parser::Functions.autoloader.loadall
+  end
+
+  let(:scope) { PuppetlabsSpec::PuppetInternals.scope }
+
+  it 'should exist' do
+    Puppet::Parser::Functions.function('mysql_password').should == 'function_mysql_password'
+  end
+
+  it 'should raise a ParseError if there is less than 1 arguments' do
+    lambda { scope.function_mysql_password([]) }.should( raise_error(Puppet::ParseError))
+  end
+
+  it 'should raise a ParseError if there is more than 1 arguments' do
+    lambda { scope.function_mysql_password(%w(foo bar)) }.should( raise_error(Puppet::ParseError))
+  end
+
+  it 'should convert password into a hash' do
+    result = scope.function_mysql_password(%w(password))
+    result.should(eq('*2470C0C06DEE42FD1618BB99005ADCA2EC9D1E19'))
+  end
+
+end

data/modules/mysql/spec/unit/puppet/functions/mysql_deepmerge_spec.rb

@@ -0,0 +1,77 @@
+#! /usr/bin/env ruby -S rspec
+
+require 'spec_helper'
+
+describe Puppet::Parser::Functions.function(:mysql_deepmerge) do
+  let(:scope) { PuppetlabsSpec::PuppetInternals.scope }
+
+  describe 'when calling mysql_deepmerge from puppet' do
+    it "should not compile when no arguments are passed" do
+      pending("Fails on 2.6.x, see bug #15912") if Puppet.version =~ /^2\.6\./
+      Puppet[:code] = '$x = mysql_deepmerge()'
+      expect {
+        scope.compiler.compile
+      }.to raise_error(Puppet::ParseError, /wrong number of arguments/)
+    end
+
+    it "should not compile when 1 argument is passed" do
+      pending("Fails on 2.6.x, see bug #15912") if Puppet.version =~ /^2\.6\./
+      Puppet[:code] = "$my_hash={'one' => 1}\n$x = mysql_deepmerge($my_hash)"
+      expect {
+        scope.compiler.compile
+      }.to raise_error(Puppet::ParseError, /wrong number of arguments/)
+    end
+  end
+
+  describe 'when calling mysql_deepmerge on the scope instance' do
+    it 'should require all parameters are hashes' do
+      expect { new_hash = scope.function_mysql_deepmerge([{}, '2'])}.to raise_error(Puppet::ParseError, /unexpected argument type String/)
+      expect { new_hash = scope.function_mysql_deepmerge([{}, 2])}.to raise_error(Puppet::ParseError, /unexpected argument type Fixnum/)
+    end
+
+    it 'should accept empty strings as puppet undef' do
+      expect { new_hash = scope.function_mysql_deepmerge([{}, ''])}.not_to raise_error
+    end
+
+    it 'should be able to mysql_deepmerge two hashes' do
+      new_hash = scope.function_mysql_deepmerge([{'one' => '1', 'two' => '1'}, {'two' => '2', 'three' => '2'}])
+      new_hash['one'].should   == '1'
+      new_hash['two'].should   == '2'
+      new_hash['three'].should == '2'
+    end
+
+    it 'should mysql_deepmerge multiple hashes' do
+      hash = scope.function_mysql_deepmerge([{'one' => 1}, {'one' => '2'}, {'one' => '3'}])
+      hash['one'].should == '3'
+    end
+
+    it 'should accept empty hashes' do
+      scope.function_mysql_deepmerge([{},{},{}]).should == {}
+    end
+
+    it 'should mysql_deepmerge subhashes' do
+      hash = scope.function_mysql_deepmerge([{'one' => 1}, {'two' => 2, 'three' => { 'four' => 4 } }])
+      hash['one'].should == 1
+      hash['two'].should == 2
+      hash['three'].should == { 'four' => 4 }
+    end
+
+    it 'should append to subhashes' do
+      hash = scope.function_mysql_deepmerge([{'one' => { 'two' => 2 } }, { 'one' => { 'three' => 3 } }])
+      hash['one'].should == { 'two' => 2, 'three' => 3 }
+    end
+
+    it 'should append to subhashes 2' do
+      hash = scope.function_mysql_deepmerge([{'one' => 1, 'two' => 2, 'three' => { 'four' => 4 } }, {'two' => 'dos', 'three' => { 'five' => 5 } }])
+      hash['one'].should == 1
+      hash['two'].should == 'dos'
+      hash['three'].should == { 'four' => 4, 'five' => 5 }
+    end
+
+    it 'should append to subhashes 3' do
+      hash = scope.function_mysql_deepmerge([{ 'key1' => { 'a' => 1, 'b' => 2 }, 'key2' => { 'c' => 3 } }, { 'key1' => { 'b' => 99 } }])
+      hash['key1'].should == { 'a' => 1, 'b' => 99 }
+      hash['key2'].should == { 'c' => 3 }
+    end
+  end
+end

data/modules/mysql/spec/unit/puppet/provider/database/mysql_spec.rb

@@ -0,0 +1,86 @@
+require 'spec_helper'
+
+provider_class = Puppet::Type.type(:database).provider(:mysql)
+
+describe provider_class do
+  subject { provider_class }
+
+  let(:root_home) { '/root' }
+  let(:defaults_file) { '--defaults-extra-file=/root/.my.cnf' }
+
+  let(:raw_databases) do
+    <<-SQL_OUTPUT
+information_schema
+mydb
+mysql
+performance_schema
+test
+    SQL_OUTPUT
+  end
+
+  let(:parsed_databases) { %w(information_schema mydb mysql performance_schema test) }
+
+  before :each do
+    @resource = Puppet::Type::Database.new(
+      { :charset => 'utf8', :name => 'new_database' }
+    )
+    @provider = provider_class.new(@resource)
+    Facter.stubs(:value).with(:root_home).returns(root_home)
+    Puppet::Util.stubs(:which).with('mysql').returns('/usr/bin/mysql')
+    subject.stubs(:which).with('mysql').returns('/usr/bin/mysql')
+    subject.stubs(:defaults_file).returns('--defaults-extra-file=/root/.my.cnf')
+  end
+
+  describe 'self.instances' do
+    it 'returns an array of databases' do
+      subject.stubs(:mysql).with([defaults_file, '-NBe', 'show databases']).returns(raw_databases)
+
+      databases = subject.instances.collect {|x| x.name }
+      parsed_databases.should match_array(databases)
+    end
+  end
+
+  describe 'create' do
+    it 'makes a user' do
+      subject.expects(:mysql).with([defaults_file, '-NBe', "create database `#{@resource[:name]}` character set #{@resource[:charset]}"])
+      @provider.create
+    end
+  end
+
+  describe 'destroy' do
+    it 'removes a user if present' do
+      subject.expects(:mysqladmin).with([defaults_file, '-f', 'drop', "#{@resource[:name]}"])
+      @provider.destroy
+    end
+  end
+
+  describe 'charset' do
+    it 'returns a charset' do
+      subject.expects(:mysql).with([defaults_file, '-NBe', "show create database `#{@resource[:name]}`"]).returns('mydbCREATE DATABASE `mydb` /*!40100 DEFAULT CHARACTER SET utf8 */')
+      @provider.charset.should == 'utf8'
+    end
+  end
+
+  describe 'charset=' do
+    it 'changes the charset' do
+      subject.expects(:mysql).with([defaults_file, '-NBe', "alter database `#{@resource[:name]}` CHARACTER SET blah"]).returns('0')
+
+      @provider.charset=('blah')
+    end
+  end
+
+  describe 'exists?' do
+    it 'checks if user exists' do
+      subject.expects(:mysql).with([defaults_file, '-NBe', 'show databases']).returns('information_schema\nmydb\nmysql\nperformance_schema\ntest')
+      @provider.exists?
+    end
+  end
+
+  describe 'self.defaults_file' do
+    it 'sets --defaults-extra-file' do
+      File.stubs(:file?).with('#{root_home}/.my.cnf').returns(true)
+      @provider.defaults_file.should == '--defaults-extra-file=/root/.my.cnf'
+    end
+  end
+
+end