cronicle 0.1.0

data/spec/cronicle_update_spec.rb

@@ -0,0 +1,365 @@
+describe 'Cronicle::Client#apply (update)' do
+  before do
+    on TARGET_HOSTS do |ssh_options|
+      user = ssh_options[:user]
+
+      set_crontab user, <<-CRON.undent
+        FOO=bar
+        ZOO=baz
+        1 1 1 1 1 echo #{user} > /dev/null
+      CRON
+
+      set_crontab :root, <<-CRON.undent
+        FOO=bar
+        ZOO=baz
+        1 1 1 1 1 echo root > /dev/null
+      CRON
+    end
+
+    cronicle(:apply) { <<-RUBY.undent }
+      on servers: /.*/ do
+        job :foo, user: :root, schedule: '1 2 * * *' do
+          puts `uname`
+          puts `whoami`
+        end
+      end
+
+      on servers: /.*/ do
+        job :bar, user: :root, schedule: :@hourly, content: <<-SH.undent
+          #!/bin/sh
+          echo hello
+        SH
+      end
+
+      on servers: /amazon_linux/ do
+        job :foo, user: 'ec2-user', schedule: '1 * * * *' do
+          puts 100
+        end
+      end
+
+      on servers: /ubuntu/ do
+        job :foo, user: :ubuntu, schedule: :@daily do
+          puts 200
+        end
+      end
+    RUBY
+  end
+
+  let(:amzn_crontab_orig) do
+    {
+      "/var/spool/cron/ec2-user" =>
+"FOO=bar
+ZOO=baz
+1 1 1 1 1 echo ec2-user > /dev/null
+1 * * * *\t/var/lib/cronicle/libexec/ec2-user/foo 2>&1 | logger -t cronicle/ec2-user/foo
+",
+      "/var/spool/cron/root" =>
+"FOO=bar
+ZOO=baz
+1 1 1 1 1 echo root > /dev/null
+1 2 * * *\t/var/lib/cronicle/libexec/root/foo 2>&1 | logger -t cronicle/root/foo
+@hourly\t/var/lib/cronicle/libexec/root/bar 2>&1 | logger -t cronicle/root/bar
+"
+    }
+  end
+
+  let(:ubuntu_crontab_orig) do
+    {
+      "/var/spool/cron/crontabs/root" =>
+"FOO=bar
+ZOO=baz
+1 1 1 1 1 echo root > /dev/null
+1 2 * * *\t/var/lib/cronicle/libexec/root/foo 2>&1 | logger -t cronicle/root/foo
+@hourly\t/var/lib/cronicle/libexec/root/bar 2>&1 | logger -t cronicle/root/bar
+",
+      "/var/spool/cron/crontabs/ubuntu" =>
+"FOO=bar
+ZOO=baz
+1 1 1 1 1 echo ubuntu > /dev/null
+@daily\t/var/lib/cronicle/libexec/ubuntu/foo 2>&1 | logger -t cronicle/ubuntu/foo
+"
+    }
+  end
+
+  let(:jobfile) do
+    <<-RUBY.undent
+      on servers: /.*/ do
+        job :foo, user: :root, schedule: '1 2 * * *' do
+          puts `uname`
+          #puts `whoami`
+        end
+      end
+
+      on servers: /.*/ do
+        job :bar, user: :root, schedule: :@hourly, content: <<-SH.undent
+          #!/bin/sh
+          echo hello
+        SH
+      end
+
+      on servers: /amazon_linux/ do
+        job :foo, user: 'ec2-user', schedule: '2 * * * *' do
+          puts 100
+        end
+      end
+
+      on servers: /ubuntu/ do
+        job :foo2, user: :ubuntu, schedule: :@daily do
+          puts 200
+        end
+      end
+    RUBY
+  end
+
+  context 'when cron is updated' do
+    let(:amzn_crontab) do
+      {
+        "/var/spool/cron/ec2-user" =>
+"FOO=bar
+ZOO=baz
+1 1 1 1 1 echo ec2-user > /dev/null
+2 * * * *\t/var/lib/cronicle/libexec/ec2-user/foo 2>&1 | logger -t cronicle/ec2-user/foo
+",
+        "/var/spool/cron/root" =>
+"FOO=bar
+ZOO=baz
+1 1 1 1 1 echo root > /dev/null
+@hourly\t/var/lib/cronicle/libexec/root/bar 2>&1 | logger -t cronicle/root/bar
+1 2 * * *\t/var/lib/cronicle/libexec/root/foo 2>&1 | logger -t cronicle/root/foo
+"
+      }
+    end
+
+    let(:ubuntu_crontab) do
+      {
+        "/var/spool/cron/crontabs/root" =>
+"FOO=bar
+ZOO=baz
+1 1 1 1 1 echo root > /dev/null
+@hourly\t/var/lib/cronicle/libexec/root/bar 2>&1 | logger -t cronicle/root/bar
+1 2 * * *\t/var/lib/cronicle/libexec/root/foo 2>&1 | logger -t cronicle/root/foo
+",
+        "/var/spool/cron/crontabs/ubuntu" =>
+"FOO=bar
+ZOO=baz
+1 1 1 1 1 echo ubuntu > /dev/null
+@daily\t/var/lib/cronicle/libexec/ubuntu/foo2 2>&1 | logger -t cronicle/ubuntu/foo2
+"
+      }
+    end
+
+    it do
+      on :amazon_linux do
+        expect(get_uname).to match /amzn/
+        expect(get_crontabs).to eq amzn_crontab_orig
+
+        expect(get_file('/var/lib/cronicle/libexec/root/foo')).to eq <<-EOS.undent
+          #!/usr/bin/env ruby
+          puts `uname`
+          puts `whoami`
+        EOS
+
+        expect(get_file('/var/lib/cronicle/libexec/root/bar')).to eq <<-EOS.undent
+          #!/bin/sh
+          echo hello
+        EOS
+
+        expect(get_file('/var/lib/cronicle/libexec/ec2-user/foo')).to eq <<-EOS.undent
+          #!/usr/bin/env ruby
+          puts 100
+        EOS
+      end
+
+      on :ubuntu do
+        expect(get_uname).to match /Ubuntu/
+        expect(get_crontabs).to eq ubuntu_crontab_orig
+
+        expect(get_file('/var/lib/cronicle/libexec/root/foo')).to eq <<-EOS.undent
+          #!/usr/bin/env ruby
+          puts `uname`
+          puts `whoami`
+        EOS
+
+        expect(get_file('/var/lib/cronicle/libexec/root/bar')).to eq <<-EOS.undent
+          #!/bin/sh
+          echo hello
+        EOS
+
+        expect(get_file('/var/lib/cronicle/libexec/ubuntu/foo')).to eq <<-EOS.undent
+          #!/usr/bin/env ruby
+          puts 200
+        EOS
+      end
+
+      cronicle(:apply) { jobfile }
+
+      on :amazon_linux do
+        expect(get_uname).to match /amzn/
+        expect(get_crontabs).to eq amzn_crontab
+
+        expect(get_file('/var/lib/cronicle/libexec/root/foo')).to eq <<-EOS.undent
+          #!/usr/bin/env ruby
+          puts `uname`
+          #puts `whoami`
+        EOS
+
+        expect(get_file('/var/lib/cronicle/libexec/root/bar')).to eq <<-EOS.undent
+          #!/bin/sh
+          echo hello
+        EOS
+
+        expect(get_file('/var/lib/cronicle/libexec/ec2-user/foo')).to eq <<-EOS.undent
+          #!/usr/bin/env ruby
+          puts 100
+        EOS
+      end
+
+      on :ubuntu do
+        expect(get_uname).to match /Ubuntu/
+        expect(get_crontabs).to eq ubuntu_crontab
+
+        expect(get_file('/var/lib/cronicle/libexec/root/foo')).to eq <<-EOS.undent
+          #!/usr/bin/env ruby
+          puts `uname`
+          #puts `whoami`
+        EOS
+
+        expect(get_file('/var/lib/cronicle/libexec/root/bar')).to eq <<-EOS.undent
+          #!/bin/sh
+          echo hello
+        EOS
+
+        expect(get_file('/var/lib/cronicle/libexec/ubuntu/foo2')).to eq <<-EOS.undent
+          #!/usr/bin/env ruby
+          puts 200
+        EOS
+      end
+    end
+  end
+
+  context 'when cron is updated (dry-run)' do
+    let(:amzn_crontab) do
+      {
+        "/var/spool/cron/ec2-user" =>
+"FOO=bar
+ZOO=baz
+1 1 1 1 1 echo ec2-user > /dev/null
+2 * * * *\t/var/lib/cronicle/libexec/ec2-user/foo 2>&1 | logger -t cronicle/ec2-user/foo
+",
+        "/var/spool/cron/root" =>
+"FOO=bar
+ZOO=baz
+1 1 1 1 1 echo root > /dev/null
+@hourly\t/var/lib/cronicle/libexec/root/bar 2>&1 | logger -t cronicle/root/bar
+1 2 * * *\t/var/lib/cronicle/libexec/root/foo 2>&1 | logger -t cronicle/root/foo
+"
+      }
+    end
+
+    let(:ubuntu_crontab) do
+      {
+        "/var/spool/cron/crontabs/root" =>
+"FOO=bar
+ZOO=baz
+1 1 1 1 1 echo root > /dev/null
+@hourly\t/var/lib/cronicle/libexec/root/bar 2>&1 | logger -t cronicle/root/bar
+1 2 * * *\t/var/lib/cronicle/libexec/root/foo 2>&1 | logger -t cronicle/root/foo
+",
+        "/var/spool/cron/crontabs/ubuntu" =>
+"FOO=bar
+ZOO=baz
+1 1 1 1 1 echo ubuntu > /dev/null
+@daily\t/var/lib/cronicle/libexec/ubuntu/foo2 2>&1 | logger -t cronicle/ubuntu/foo2
+"
+      }
+    end
+
+    it do
+      on :amazon_linux do
+        expect(get_uname).to match /amzn/
+        expect(get_crontabs).to eq amzn_crontab_orig
+
+        expect(get_file('/var/lib/cronicle/libexec/root/foo')).to eq <<-EOS.undent
+          #!/usr/bin/env ruby
+          puts `uname`
+          puts `whoami`
+        EOS
+
+        expect(get_file('/var/lib/cronicle/libexec/root/bar')).to eq <<-EOS.undent
+          #!/bin/sh
+          echo hello
+        EOS
+
+        expect(get_file('/var/lib/cronicle/libexec/ec2-user/foo')).to eq <<-EOS.undent
+          #!/usr/bin/env ruby
+          puts 100
+        EOS
+      end
+
+      on :ubuntu do
+        expect(get_uname).to match /Ubuntu/
+        expect(get_crontabs).to eq ubuntu_crontab_orig
+
+        expect(get_file('/var/lib/cronicle/libexec/root/foo')).to eq <<-EOS.undent
+          #!/usr/bin/env ruby
+          puts `uname`
+          puts `whoami`
+        EOS
+
+        expect(get_file('/var/lib/cronicle/libexec/root/bar')).to eq <<-EOS.undent
+          #!/bin/sh
+          echo hello
+        EOS
+
+        expect(get_file('/var/lib/cronicle/libexec/ubuntu/foo')).to eq <<-EOS.undent
+          #!/usr/bin/env ruby
+          puts 200
+        EOS
+      end
+
+      cronicle(:apply, dry_run: true) { jobfile }
+
+      on :amazon_linux do
+        expect(get_uname).to match /amzn/
+        expect(get_crontabs).to eq amzn_crontab_orig
+
+        expect(get_file('/var/lib/cronicle/libexec/root/foo')).to eq <<-EOS.undent
+          #!/usr/bin/env ruby
+          puts `uname`
+          puts `whoami`
+        EOS
+
+        expect(get_file('/var/lib/cronicle/libexec/root/bar')).to eq <<-EOS.undent
+          #!/bin/sh
+          echo hello
+        EOS
+
+        expect(get_file('/var/lib/cronicle/libexec/ec2-user/foo')).to eq <<-EOS.undent
+          #!/usr/bin/env ruby
+          puts 100
+        EOS
+      end
+
+      on :ubuntu do
+        expect(get_uname).to match /Ubuntu/
+        expect(get_crontabs).to eq ubuntu_crontab_orig
+
+        expect(get_file('/var/lib/cronicle/libexec/root/foo')).to eq <<-EOS.undent
+          #!/usr/bin/env ruby
+          puts `uname`
+          puts `whoami`
+        EOS
+
+        expect(get_file('/var/lib/cronicle/libexec/root/bar')).to eq <<-EOS.undent
+          #!/bin/sh
+          echo hello
+        EOS
+
+        expect(get_file('/var/lib/cronicle/libexec/ubuntu/foo')).to eq <<-EOS.undent
+          #!/usr/bin/env ruby
+          puts 200
+        EOS
+      end
+    end
+  end
+end

data/spec/host_list_spec.rb

@@ -0,0 +1,162 @@
+describe Cronicle::HostList do
+  let(:host_list) { described_class.new(source) }
+
+  context 'when csv is passed' do
+    let(:source) { "foo,bar" }
+
+    it do
+      expect(host_list.all).to match_array ["foo", "bar"]
+    end
+
+    it do
+      expect(host_list.select(servers: /foo/)).to match_array ["foo"]
+    end
+
+    it do
+      expect(host_list.select(roles: /web/)).to match_array []
+    end
+  end
+
+  context 'when list is passed' do
+    let(:source) do
+      <<-EOS.undent
+        foo
+        bar
+      EOS
+    end
+
+    it do
+      expect(host_list.all).to match_array ["foo", "bar"]
+    end
+
+    it do
+      expect(host_list.select(servers: /foo/)).to match_array ["foo"]
+    end
+
+    it do
+      expect(host_list.select(roles: /web/)).to match_array []
+    end
+  end
+
+  context 'when JSON(servers) is passed' do
+    context 'when single server' do
+      let(:source) do
+        <<-EOS.undent
+          {
+            "servers": "foo"
+          }
+        EOS
+      end
+
+      it do
+        expect(host_list.all).to match_array ["foo"]
+      end
+
+      it do
+        expect(host_list.select(servers: /foo/)).to match_array ["foo"]
+      end
+
+      it do
+        expect(host_list.select(roles: /web/)).to match_array []
+      end
+    end
+
+    context 'when multiple servers' do
+      let(:source) do
+        <<-EOS.undent
+          {
+            "servers": ["foo", "bar"]
+          }
+        EOS
+      end
+
+      it do
+        expect(host_list.all).to match_array ["foo", "bar"]
+      end
+
+      it do
+        expect(host_list.select(servers: /foo/)).to match_array ["foo"]
+      end
+
+      it do
+        expect(host_list.select(roles: /web/)).to match_array []
+      end
+    end
+
+    context 'when multiple servers with role' do
+      let(:source) do
+        <<-EOS.undent
+          {
+            "servers": {
+              "foo": "db",
+              "bar": "web"
+            }
+          }
+        EOS
+      end
+
+      it do
+        expect(host_list.all).to match_array ["foo", "bar"]
+      end
+
+      it do
+        expect(host_list.select(servers: /foo/)).to match_array ["foo"]
+      end
+
+      it do
+        expect(host_list.select(roles: /web/)).to match_array ["bar"]
+      end
+    end
+  end
+
+  context 'when JSON(roles) is passed' do
+    context 'when single server' do
+      let(:source) do
+        <<-EOS.undent
+          {
+            "roles": {
+              "web": "bar"
+            }
+          }
+        EOS
+      end
+
+      it do
+        expect(host_list.all).to match_array ["bar"]
+      end
+
+      it do
+        expect(host_list.select(servers: /foo/)).to match_array []
+      end
+
+      it do
+        expect(host_list.select(roles: /web/)).to match_array ["bar"]
+      end
+    end
+
+    context 'when multiple servers' do
+      let(:source) do
+        <<-EOS.undent
+          {
+            "roles": {
+              "db": ["foo", "bar"],
+              "web": ["zoo", "baz"]
+            }
+          }
+        EOS
+      end
+
+      it do
+        expect(host_list.all).to match_array ["foo", "bar", "zoo", "baz"]
+      end
+
+      it do
+        expect(host_list.select(servers: /oo/)).to match_array ["foo", "zoo"]
+      end
+
+      it do
+        expect(host_list.select(roles: /web/)).to match_array ["zoo", "baz"]
+      end
+    end
+  end
+end