zabbirc 0.0.11 → 0.1.0

data/spec/settings_command_spec.rb

@@ -0,0 +1,125 @@
+describe Zabbirc::Irc::SettingsCommand do
+  let(:ops_builder) { Zabbirc::OpsBuilder.new }
+  let(:mock_nick) { "op1" }
+  let(:mock_op) { ops_builder.build_op(mock_nick, mock_user_settings) }
+  let(:mock_message) { double("Cinch::Message", user: mock_op.irc_user) }
+  let(:mock_user_settings) { nil }
+  let(:settings_command) { Zabbirc::Irc::SettingsCommand.new ops_builder.ops, mock_message, cmd }
+
+
+  describe "#show_settings" do
+    let(:mock_user_settings) { {primary_channel: "#channel-1", events_priority: "high", notify: false } }
+    let(:expected_msg) { "#{mock_nick}: Default settings: notify: false, notify_recoveries: true, primary_channel: #channel-1, events_priority: high" }
+    let(:cmd) { "show" }
+    it "should show settings" do
+      expect(mock_message).to receive(:reply).with(expected_msg)
+      settings_command.run
+    end
+  end
+
+  describe "#set_setting" do
+    shared_examples "set_setting" do |key, value, expected_setting_value|
+      let(:expected_msg) { "#{mock_nick}: setting `#{key}` has been set to `#{expected_setting_value}`" }
+      let(:cmd) { "set #{key} #{value}"}
+      it "should set #{key} setting to #{value}" do
+        expect(mock_message).to receive(:reply).with(expected_msg)
+        settings_command.run
+        expect(mock_op.setting.get(key)).to eq expected_setting_value
+      end
+    end
+
+
+    context "notify" do
+      it_should_behave_like "set_setting", "notify", "false", false
+      it_should_behave_like "set_setting", "notify", "true", true
+    end
+
+    context "events_priority" do
+      it_should_behave_like "set_setting", "events_priority", "high", :high
+      it_should_behave_like "set_setting", "events_priority", "5", :disaster
+    end
+
+    context "primary_channel" do
+      before do
+        mock_op.add_channel double("#channel1", name: "#channel1")
+        mock_op.add_channel double("#channel2double", name: "#channel2")
+      end
+      it_should_behave_like "set_setting", "primary_channel", "#channel1", "#channel1"
+      it_should_behave_like "set_setting", "primary_channel", "#channel2", "#channel2"
+    end
+
+    context "group specific" do
+      let(:group1) { double "HostGroup", id: 1, name: "Group1" }
+      let(:group2) { double "HostGroup", id: 2, name: "Group2" }
+      let(:groups) { [group1, group2] }
+      before do
+        allow(Zabbirc::Zabbix::HostGroup).to receive(:get) { groups }
+        allow(Zabbirc::Zabbix::HostGroup).to receive(:get).with(hash_including(search: { name: anything })) do |params|
+          groups.select{|g| g.name =~ /#{params[:search][:name]}/ }
+        end
+      end
+
+      shared_examples "set_group_setting" do |key, value, expected_setting_value|
+        let(:expected_msg) { "#{mock_nick}: setting `#{key}` has been set to `#{expected_setting_value}` for host groups: #{affected_host_group.name}" }
+        let(:cmd) { "set #{key} #{value} hostgroups #{affected_host_group.name}"}
+        it "should set #{key} setting to #{value} for host group" do
+          old_global_value = mock_op.setting.get(key)
+
+          expect(mock_message).to receive(:reply).with(expected_msg)
+          settings_command.run
+          expect(mock_op.setting.get(key, host_group_id: affected_host_group.id)).to eq expected_setting_value
+
+          expect(mock_op.setting.get(key)).to eq old_global_value
+        end
+      end
+
+      shared_examples "set_all_groups_setting" do |key, value, expected_setting_value|
+        let(:settings_command1) { Zabbirc::Irc::SettingsCommand.new ops_builder.ops, mock_message, cmd1 }
+        let(:settings_command2) { Zabbirc::Irc::SettingsCommand.new ops_builder.ops, mock_message, cmd2 }
+
+        let!(:old_global_value) { mock_op.setting.get(key, host_group_id: affected_host_group.id) }
+        let(:expected_msg1) { "#{mock_nick}: setting `#{key}` has been set to `#{expected_setting_value}` for host groups: #{affected_host_group.name}" }
+        let(:expected_msg2) { "#{mock_nick}: setting `#{key}` has been set to `#{old_global_value}` for all host groups" }
+        let(:cmd1) { "set #{key} #{value} hostgroups #{affected_host_group.name}"}
+        let(:cmd2) { "set #{key} #{old_global_value} hostgroups-all"}
+        it "should set #{key} setting to #{value} for all host groups" do
+          # sets host group specific setting
+          expect(mock_message).to receive(:reply).with(expected_msg1)
+          settings_command1.run
+          expect(mock_op.setting.get(key, host_group_id: affected_host_group.id)).to eq expected_setting_value
+          expect(mock_op.setting.get(key)).to eq old_global_value
+
+          expect(mock_message).to receive(:reply).with(expected_msg2)
+          settings_command2.run
+          expect(mock_op.setting.get(key, host_group_id: affected_host_group.id)).to eq old_global_value
+          expect(mock_op.setting.get(key)).to eq old_global_value
+        end
+      end
+
+      it_should_behave_like "set_group_setting", "notify", "false", false do
+        let(:affected_host_group) { group1 }
+      end
+
+      it_should_behave_like "set_group_setting", "events_priority", "high", :high do
+        let(:affected_host_group) { group1 }
+      end
+
+      it_should_behave_like "set_group_setting", "events_priority", "5", :disaster do
+        let(:affected_host_group) { group1 }
+      end
+
+      it_should_behave_like "set_all_groups_setting", "notify", "false", false do
+        let(:affected_host_group) { group1 }
+      end
+
+      it_should_behave_like "set_all_groups_setting", "events_priority", "high", :high do
+        let(:affected_host_group) { group1 }
+      end
+
+      it_should_behave_like "set_all_groups_setting", "events_priority", "5", :disaster do
+        let(:affected_host_group) { group1 }
+      end
+
+    end
+  end
+end

data/spec/support/ops_builder.rb

@@ -0,0 +1,23 @@
+module Zabbirc
+  class OpsBuilder
+    include RSpec::Mocks::ExampleMethods
+
+    attr_reader :ops
+    def initialize
+      @ops = OpList.new
+    end
+
+    def build_op name, settings=nil
+      @op_ids ||= 0
+      zabbix_user = Zabbix::User.new(alias: name, userid: (@op_ids+=1))
+      irc_user = double "Cinch::User", nick: name, user: name
+      op = Op.new(zabbix_user)
+      op.set_irc_user irc_user
+      settings.each do |key, value|
+        op.setting.set key, value
+      end if settings
+      @ops.add op
+      op
+    end
+  end
+end

data/tmp/playground.rb

@@ -0,0 +1,22 @@
+# tulak:
+#   notify: true
+#   notify_recoveries: true
+#   primary_channel: "#zabbirc-test"
+#   events_priority: :high
+#   host_groups:
+#     1:
+#       events_priority: :disaster
+
+!settings set notify false hostgroups Linux servers, Zabbix servers
+
+setting.get(:notify) => default
+setting.get(:notify, "Linux servers") => hostgroup-setting || default
+
+include Zabbirc
+s = Setting.new
+s.set :events_priority, :average, host_group_id: 1
+s.set :events_priority, :warning, host_group_id: 2
+s.set :events_priority, :high, host_group_id: 1
+
+
+c = SettingsCommand.new

data/tmp/stopwatch.rb

@@ -0,0 +1,52 @@
+module Stopwatch
+  def self.catch formatted=true
+    start_time = Time.now
+    yield
+    end_time = Time.now
+    duration = end_time - start_time
+    if formatted
+      duration.to_dot_time(true)
+    else
+      duration
+    end
+
+  end
+end
+
+class Integer
+  def to_dot_time mili=false
+    s = self
+    h = s / 3600
+    s -= h * 3600
+
+    m = s / 60
+    s -= m * 60
+
+    h = "%02d" % h
+    m = "%02d" % m
+    s = "%02#{mili ? "f" : "d"}" % s
+
+    [h, m, s].join(":")
+  end
+
+  def dec2bin
+    "0b%b" % self
+  end
+end
+
+class Float
+  def to_dot_time mili=false
+    s = self
+    h = (s / 3600.0).floor
+    s -= h * 3600.0
+
+    m = (s / 60.0).floor
+    s -= m * 60.0
+
+    h = "%02d" % h
+    m = "%02d" % m
+    s = "%02#{mili ? "f" : "d"}" % s
+
+    [h, m, s].join(":")
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: zabbirc
 version: !ruby/object:Gem::Version
-  version: 0.0.11
+  version: 0.1.0
 platform: ruby
 authors:
 - Filip Zachar
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-01-08 00:00:00.000000000 Z
+date: 2015-11-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -16,20 +16,20 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 4.1.6
+        version: 4.2.4
     - - ">="
       - !ruby/object:Gem::Version
-        version: 4.1.6
+        version: 4.2.4
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 4.1.6
+        version: 4.2.4
     - - ">="
       - !ruby/object:Gem::Version
-        version: 4.1.6
+        version: 4.2.4
 - !ruby/object:Gem::Dependency
   name: dotenv
   requirement: !ruby/object:Gem::Requirement
@@ -118,9 +118,14 @@ files:
 - lib/zabbirc.rb
 - lib/zabbirc/configuration.rb
 - lib/zabbirc/id_shortener.rb
+- lib/zabbirc/irc/base_command.rb
+- lib/zabbirc/irc/event_command.rb
 - lib/zabbirc/irc/help.rb
+- lib/zabbirc/irc/help_command.rb
+- lib/zabbirc/irc/host_command.rb
 - lib/zabbirc/irc/plugin.rb
 - lib/zabbirc/irc/plugin_methods.rb
+- lib/zabbirc/irc/settings_command.rb
 - lib/zabbirc/logger.rb
 - lib/zabbirc/op.rb
 - lib/zabbirc/op_list.rb
@@ -131,21 +136,27 @@ files:
 - lib/zabbirc/services/ops.rb
 - lib/zabbirc/setting.rb
 - lib/zabbirc/stop_error.rb
+- lib/zabbirc/version.rb
 - lib/zabbirc/zabbix/connection.rb
 - lib/zabbirc/zabbix/event.rb
 - lib/zabbirc/zabbix/host.rb
+- lib/zabbirc/zabbix/host_group.rb
 - lib/zabbirc/zabbix/resource/associations.rb
 - lib/zabbirc/zabbix/resource/base.rb
 - lib/zabbirc/zabbix/resource/finders.rb
 - lib/zabbirc/zabbix/trigger.rb
 - lib/zabbirc/zabbix/user.rb
-- spec/bot_spec.rb
+- spec/event_command_spec.rb
+- spec/host_command_spec.rb
 - spec/id_shortener_spec.rb
+- spec/settings_command_spec.rb
 - spec/spec_helper.rb
 - spec/support/mock_bot.rb
+- spec/support/ops_builder.rb
 - templates/zabbirc_config.rb
 - tmp/module_test.rb
 - tmp/playground.rb
+- tmp/stopwatch.rb
 - tmp/thread_test.rb
 homepage: https://github.com/tulak/zabbirc
 licenses:

data/spec/bot_spec.rb

@@ -1,212 +0,0 @@
-describe Zabbirc::Irc::PluginMethods do
-  # let(:service) { Zabbirc::ServiceMock.new }
-  let(:mock_message) { double("Cinch::Message", user: mock_user) }
-  let(:mock_user) { double("Cinch::User", nick: mock_nick, login: mock_login) }
-  let(:bot) { Zabbirc::MockBot.new }
-  let(:mock_nick) { "op1" }
-  let(:mock_login) { "op1" }
-  let(:mock_user_settings) { nil }
-
-  before do
-    bot.setup_op mock_nick, mock_user_settings
-  end
-
-  describe "#acknowledge_event" do
-    let(:event) { double "Event", id: 1, label: "Event 1 label" }
-    let(:message) { "ack message" }
-    before do
-      allow(event).to receive(:acknowledge).and_return(true)
-      allow(Zabbirc::Zabbix::Event).to receive(:find).and_return(event)
-    end
-
-    it "should acknowledge event" do
-      shorten_id = Zabbirc.events_id_shortener.get_shorten_id event.id
-      expect(mock_message).to receive(:reply).with("#{mock_nick}: Event `#{event.label}` acknowledged with message: #{message}")
-      bot.acknowledge_event mock_message, shorten_id, message
-    end
-  end
-
-  describe "#host_status" do
-    before do
-      allow(Zabbirc::Zabbix::Host).to receive(:get).and_return(hosts)
-    end
-
-    context "reporting" do
-      before do
-        allow(Zabbirc::Zabbix::Trigger).to receive(:get).and_return(problem_triggers)
-      end
-      let(:host) { double "Host", id: 1, name: "Host-1" }
-      let(:hosts) { [host] }
-      let(:problem_trigger) { double "Trigger", priority: Zabbirc::Priority.new(1), label: "problem_trigger", value: 1 }
-      context "problem trigger" do
-        let(:problem_triggers) { [problem_trigger] }
-        let(:expected_msg) do
-          msg = ["#{mock_nick}: Host: #{host.name} - status: #{problem_triggers.size} problems"]
-          problem_triggers.each do |trigger|
-            msg << "#{mock_nick}: status: #{trigger.label}"
-          end
-          msg.join("\n")
-        end
-        it "should report problem" do
-          expect(mock_message).to receive(:reply).with(expected_msg)
-          bot.host_status mock_message, host.name
-        end
-      end
-
-      context "ok trigger" do
-        let(:problem_triggers) { [] }
-        let(:expected_msg) { "#{mock_nick}: Host: #{host.name} - status: OK" }
-        it "should report problem" do
-          expect(mock_message).to receive(:reply).with(expected_msg)
-          bot.host_status mock_message, host.name
-        end
-      end
-    end # context reporting
-
-    context "host identification" do
-      context "no hosts" do
-        let(:hosts) { [] }
-        let(:host_name) { "undefined_host_name" }
-        it "should not found host" do
-          expect(mock_message).to receive(:reply).with("#{mock_nick}: Host not found `#{host_name}`")
-          bot.host_status mock_message, host_name
-        end
-      end
-
-      context "2 - 10 hosts" do
-        let(:host1) { double "Host1", id: 1, name: "host-1" }
-        let(:host2) { double "Host2", id: 2, name: "host-2" }
-        let(:hosts) { [host1, host2] }
-        let(:expected_msg) { "#{mock_nick}: Found #{hosts.size} hosts: #{hosts.collect(&:name).join(', ')}. Be more specific" }
-        it "should print host names" do
-          expect(mock_message).to receive(:reply).with(expected_msg)
-          bot.host_status mock_message, "host"
-        end
-      end
-
-      context "more than 10 hosts" do
-        let(:hosts) { double "HostsArray", size: 11 }
-        let(:expected_msg) { "#{mock_nick}: Found #{hosts.size} Be more specific" }
-        it "should print host names" do
-          expect(mock_message).to receive(:reply).with(expected_msg)
-          bot.host_status mock_message, "host"
-        end
-      end
-    end # context host identification
-  end # context #host_status
-
-  context "#host_latest" do
-    let(:host) { double "Host", id: 1, name: "Host1" }
-    let(:hosts) { [host] }
-    let(:event1) { double "Event", label: "Event 1 label" }
-    let(:events) { [event1] }
-    let(:expected_msg) do
-      msg = ["#{mock_nick}: Host: #{host.name} - showing last #{events.size} events"]
-      events.each do |event|
-        msg << "#{mock_nick}: !latest: #{event.label}"
-      end
-      msg.join("\n")
-    end
-    before do
-      allow(Zabbirc::Zabbix::Host).to receive(:get).and_return(hosts)
-      allow(Zabbirc::Zabbix::Event).to receive(:get).and_return(events)
-    end
-
-    it "should print latest events" do
-      expect(mock_message).to receive(:reply).with(expected_msg)
-      bot.host_latest mock_message, "Host1", nil
-    end
-  end
-
-  describe "#list_events" do
-    before do
-      allow(Zabbirc::Zabbix::Event).to receive(:recent).and_return(recent_events)
-    end
-    context "no last events" do
-      let(:recent_events) { [] }
-
-      it "should report no last events" do
-        expect(mock_message).to receive(:reply).with("#{mock_nick}: No last events for priority `#{Zabbirc::Priority.new(0)}`")
-        bot.list_events mock_message
-      end
-    end
-
-    context "some last events" do
-      let(:event1_information) { double "Event1", label: "Event 1 label", priority: Zabbirc::Priority.new(:information) }
-      let(:event2_information) { double "Event2", label: "Event 2 label", priority: Zabbirc::Priority.new(:information) }
-      let(:event3_high)        { double "Event3", label: "Event 3 label", priority: Zabbirc::Priority.new(:high) }
-      let(:recent_events) { [event1_information, event2_information, event3_high] }
-      let(:expected_msg) { recent_events.collect{|e| "#{mock_nick}: #{e.label}"}.join("\n") }
-
-      before do
-        recent_events.each do |e|
-          allow(e).to receive(:any_host_matches?).and_return(false)
-        end
-      end
-
-      it "should report all last events" do
-        expect(mock_message).to receive(:reply).with(expected_msg)
-        bot.list_events mock_message
-      end
-
-      context "with high priority filtered" do
-        let(:expected_msg) { [event3_high].collect{|e| "#{mock_nick}: #{e.label}"}.join("\n") }
-        it "should report high priority last events" do
-          expect(mock_message).to receive(:reply).with(expected_msg)
-          bot.list_events mock_message, "high"
-        end
-      end
-
-      context "with host filtered" do
-        let(:expected_msg) { [event1_information].collect{|e| "#{mock_nick}: #{e.label}"}.join("\n") }
-        it "should report host matched last events" do
-          allow(event1_information).to receive(:any_host_matches?).and_return(true  )
-          expect(mock_message).to receive(:reply).with(expected_msg)
-          bot.list_events mock_message, "information", "host1"
-        end
-      end
-    end
-  end
-
-  describe "#show_settings" do
-    let(:mock_user_settings) { {primary_channel: "#channel-1", events_priority: "high", notify: false } }
-    let(:expected_msg) { "#{mock_nick}: notify: false, notify_recoveries: true, primary_channel: #channel-1, events_priority: high" }
-    it "should show settings" do
-      expect(mock_message).to receive(:reply).with(expected_msg)
-      bot.show_settings mock_message
-    end
-  end
-
-  describe "#set_setting" do
-    shared_examples "set_setting" do |key, value, expected_setting_value|
-      let(:expected_msg) { "#{mock_nick}: setting `#{key}` has been set to `#{expected_setting_value}`" }
-      let(:op) { bot.get_op mock_nick }
-      it "should set #{key} setting to #{value}" do
-        expect(mock_message).to receive(:reply).with(expected_msg)
-        bot.set_setting mock_message, key, value
-        expect(op.setting.get(key)).to eq expected_setting_value
-      end
-    end
-
-
-    context "notify" do
-      it_should_behave_like "set_setting", "notify", "false", false
-      it_should_behave_like "set_setting", "notify", "true", true
-    end
-
-    context "events_priority" do
-      it_should_behave_like "set_setting", "events_priority", "high", :high
-      it_should_behave_like "set_setting", "events_priority", "5", :disaster
-    end
-
-    context "primary_channel" do
-      before do
-        op.add_channel double("#channel1", name: "#channel1")
-        op.add_channel double("#channel2double", name: "#channel2")
-      end
-      it_should_behave_like "set_setting", "primary_channel", "#channel1", "#channel1"
-      it_should_behave_like "set_setting", "primary_channel", "#channel2", "#channel2"
-    end
-
-  end
-end