watch_list 0.1.0

data/spec/monitor_spec.rb

@@ -0,0 +1,303 @@
+describe 'monitor' do
+  it do
+    watch_list do
+      <<-RUBY
+        monitor "http monitor" do
+          target "http://example.com"
+          interval 5
+          paused false
+          alert_contact :email, "#{WATCH_LIST_TEST_EMAIL}"
+          type :http
+        end
+
+        monitor "http monitor (basic auth)" do
+          target "http://example.com"
+          interval 5
+          paused false
+          alert_contact :email, "#{WATCH_LIST_TEST_EMAIL}"
+
+          type :http do
+            httpusername "username"
+            httppassword "password"
+          end
+        end
+
+        monitor "keyword monitor" do
+          target "http://example.com"
+          interval 5
+          paused false
+          alert_contact :email, "#{WATCH_LIST_TEST_EMAIL}"
+
+          type :keyword do
+            keywordtype :exists
+            keywordvalue "Example Domain"
+          end
+        end
+
+        monitor "keyword monitor (basic auth)" do
+          target "http://example.com"
+          interval 5
+          paused false
+          alert_contact :email, "#{WATCH_LIST_TEST_EMAIL}"
+
+          type :keyword do
+            keywordtype :exists
+            keywordvalue "(Example Domain)"
+            httpusername "(username)"
+            httppassword "(password)"
+          end
+        end
+
+        monitor "ping monitor" do
+          target "127.0.0.1"
+          interval 5
+          paused false
+          alert_contact :email, "#{WATCH_LIST_TEST_EMAIL}"
+          type :ping
+        end
+
+        monitor "port monitor" do
+          target "example.com"
+          interval 5
+          paused false
+          alert_contact :email, "#{WATCH_LIST_TEST_EMAIL}"
+
+          type :port do
+            subtype :http
+            port 80
+          end
+        end
+
+        monitor "port monitor (custom)" do
+          target "example.com"
+          interval 5
+          paused false
+          alert_contact :email, "#{WATCH_LIST_TEST_EMAIL}"
+
+          type :port do
+            subtype :custom
+            port 8080
+          end
+        end
+
+        alert_contact do
+          type :email
+          value "#{WATCH_LIST_TEST_EMAIL}"
+        end
+      RUBY
+    end
+
+    expect(watch_list_export {|h| h[:monitors].length > 0 }).to eq(
+      {:monitors=>
+        {"http monitor"=>
+          {:FriendlyName=>"http monitor",
+           :URL=>"http://example.com",
+           :Type=>1,
+           :HTTPUsername=>nil,
+           :HTTPPassword=>nil,
+           :AlertContacts=>[{:Type=>2, :Value=>WATCH_LIST_TEST_EMAIL}],
+           :Interval=>5},
+         "http monitor (basic auth)"=>
+          {:FriendlyName=>"http monitor (basic auth)",
+           :URL=>"http://example.com",
+           :Type=>1,
+           :HTTPUsername=>"username",
+           :HTTPPassword=>"password",
+           :AlertContacts=>[{:Type=>2, :Value=>WATCH_LIST_TEST_EMAIL}],
+           :Interval=>5},
+         "keyword monitor"=>
+          {:FriendlyName=>"keyword monitor",
+           :URL=>"http://example.com",
+           :Type=>2,
+           :KeywordType=>1,
+           :KeywordValue=>"Example Domain",
+           :HTTPUsername=>nil,
+           :HTTPPassword=>nil,
+           :AlertContacts=>[{:Type=>2, :Value=>WATCH_LIST_TEST_EMAIL}],
+           :Interval=>5},
+         "keyword monitor (basic auth)"=>
+          {:FriendlyName=>"keyword monitor (basic auth)",
+           :URL=>"http://example.com",
+           :Type=>2,
+           :KeywordType=>1,
+           :KeywordValue=>"(Example Domain)",
+           :HTTPUsername=>"(username)",
+           :HTTPPassword=>"(password)",
+           :AlertContacts=>[{:Type=>2, :Value=>WATCH_LIST_TEST_EMAIL}],
+           :Interval=>5},
+         "ping monitor"=>
+          {:FriendlyName=>"ping monitor",
+           :URL=>"127.0.0.1",
+           :Type=>3,
+           :AlertContacts=>[{:Type=>2, :Value=>WATCH_LIST_TEST_EMAIL}],
+           :Interval=>5},
+         "port monitor"=>
+          {:FriendlyName=>"port monitor",
+           :URL=>"example.com",
+           :Type=>4,
+           :SubType=>1,
+           :Port=>80,
+           :AlertContacts=>[{:Type=>2, :Value=>WATCH_LIST_TEST_EMAIL}],
+           :Interval=>5},
+         "port monitor (custom)"=>
+          {:FriendlyName=>"port monitor (custom)",
+           :URL=>"example.com",
+           :Type=>4,
+           :SubType=>99,
+           :Port=>8080,
+           :AlertContacts=>[{:Type=>2, :Value=>WATCH_LIST_TEST_EMAIL}],
+           :Interval=>5}},
+       :alert_contacts=>[{:Type=>2, :Value=>WATCH_LIST_TEST_EMAIL}]}
+    )
+
+    watch_list do
+      <<-RUBY
+        monitor "http monitor" do
+          target "http://example.com/test"
+          interval 6
+          paused false
+          alert_contact :email, "#{WATCH_LIST_TEST_EMAIL}"
+          type :http
+        end
+
+        monitor "http monitor2 (basic auth)" do
+          target "http://example.com/test"
+          interval 7
+          paused false
+          alert_contact :email, "#{WATCH_LIST_TEST_EMAIL}"
+
+          type :http do
+            httpusername "username2"
+            httppassword "password2"
+          end
+        end
+
+        monitor "keyword monitor" do
+          target "http://example.com/test"
+          interval 8
+          paused false
+          alert_contact :email, "#{WATCH_LIST_TEST_EMAIL}"
+
+          type :keyword do
+            keywordtype :not_exists
+            keywordvalue "Example Domain2"
+          end
+        end
+
+        monitor "keyword monitor (basic auth)" do
+          target "http://example.com/test"
+          interval 9
+          paused false
+          alert_contact :email, "#{WATCH_LIST_TEST_EMAIL}"
+
+          type :keyword do
+            keywordtype :not_exists
+            keywordvalue "(Example Domain2)"
+            httpusername "(username2)"
+            httppassword "(password2)"
+          end
+        end
+
+        monitor "ping monitor" do
+          target "127.0.0.2"
+          interval 10
+          paused false
+          alert_contact :email, "#{WATCH_LIST_TEST_EMAIL}"
+          type :ping
+        end
+
+        monitor "port monitor" do
+          target "google.com"
+          interval 11
+          paused false
+          alert_contact :email, "#{WATCH_LIST_TEST_EMAIL}"
+
+          type :port do
+            subtype :https
+            port 443
+          end
+        end
+
+        monitor "port monitor (custom)" do
+          target "google.com"
+          interval 12
+          paused false
+          alert_contact :email, "#{WATCH_LIST_TEST_EMAIL}"
+
+          type :port do
+            subtype :custom
+            port 18080
+          end
+        end
+
+        alert_contact do
+          type :email
+          value "#{WATCH_LIST_TEST_EMAIL}"
+        end
+      RUBY
+    end
+
+    expect(watch_list_export {|h| h[:monitors]['port monitor (custom)'][:URL] != 'example.com' }).to eq(
+      {:monitors=>
+        {"http monitor"=>
+          {:FriendlyName=>"http monitor",
+           :URL=>"http://example.com/test",
+           :Type=>1,
+           :HTTPUsername=>nil,
+           :HTTPPassword=>nil,
+           :AlertContacts=>[{:Type=>2, :Value=>WATCH_LIST_TEST_EMAIL}],
+           :Interval=>6},
+         "keyword monitor"=>
+          {:FriendlyName=>"keyword monitor",
+           :URL=>"http://example.com/test",
+           :Type=>2,
+           :KeywordType=>2,
+           :KeywordValue=>"Example Domain2",
+           :HTTPUsername=>nil,
+           :HTTPPassword=>nil,
+           :AlertContacts=>[{:Type=>2, :Value=>WATCH_LIST_TEST_EMAIL}],
+           :Interval=>8},
+         "keyword monitor (basic auth)"=>
+          {:FriendlyName=>"keyword monitor (basic auth)",
+           :URL=>"http://example.com/test",
+           :Type=>2,
+           :KeywordType=>2,
+           :KeywordValue=>"(Example Domain2)",
+           :HTTPUsername=>"(username2)",
+           :HTTPPassword=>"(password2)",
+           :AlertContacts=>[{:Type=>2, :Value=>WATCH_LIST_TEST_EMAIL}],
+           :Interval=>9},
+         "ping monitor"=>
+          {:FriendlyName=>"ping monitor",
+           :URL=>"127.0.0.2",
+           :Type=>3,
+           :AlertContacts=>[{:Type=>2, :Value=>WATCH_LIST_TEST_EMAIL}],
+           :Interval=>10},
+         "port monitor"=>
+          {:FriendlyName=>"port monitor",
+           :URL=>"google.com",
+           :Type=>4,
+           :SubType=>2,
+           :Port=>443,
+           :AlertContacts=>[{:Type=>2, :Value=>WATCH_LIST_TEST_EMAIL}],
+           :Interval=>11},
+         "port monitor (custom)"=>
+          {:FriendlyName=>"port monitor (custom)",
+           :URL=>"google.com",
+           :Type=>4,
+           :SubType=>99,
+           :Port=>18080,
+           :AlertContacts=>[{:Type=>2, :Value=>WATCH_LIST_TEST_EMAIL}],
+           :Interval=>12},
+         "http monitor2 (basic auth)"=>
+          {:FriendlyName=>"http monitor2 (basic auth)",
+           :URL=>"http://example.com/test",
+           :Type=>1,
+           :HTTPUsername=>"username2",
+           :HTTPPassword=>"password2",
+           :AlertContacts=>[{:Type=>2, :Value=>WATCH_LIST_TEST_EMAIL}],
+           :Interval=>7}},
+       :alert_contacts=>[{:Type=>2, :Value=>WATCH_LIST_TEST_EMAIL}]}
+    )
+  end
+end

data/spec/paused_spec.rb

@@ -0,0 +1,69 @@
+describe 'monitor paused' do
+  it do
+    watch_list do
+      <<-RUBY
+        monitor "http monitor" do
+          target "http://example.com"
+          interval 5
+          paused false
+          alert_contact :email, "#{WATCH_LIST_TEST_EMAIL}"
+          type :http
+        end
+
+        alert_contact do
+          type :email
+          value "#{WATCH_LIST_TEST_EMAIL}"
+        end
+      RUBY
+    end
+
+    exported = watch_list_export(skip_normalize: true) {|h| h[:monitors].length > 0 }
+    expect(exported[:monitors]['http monitor'][:Status]).to_not eq UptimeRobot::Monitor::Status::Paused
+
+    watch_list do
+      <<-RUBY
+        monitor "http monitor" do
+          target "http://example.com"
+          interval 5
+          paused true
+          alert_contact :email, "#{WATCH_LIST_TEST_EMAIL}"
+          type :http
+        end
+
+        alert_contact do
+          type :email
+          value "#{WATCH_LIST_TEST_EMAIL}"
+        end
+      RUBY
+    end
+
+    exported = watch_list_export(skip_normalize: true) {|h|
+      h[:monitors]['http monitor'][:Status] == UptimeRobot::Monitor::Status::Paused
+    }
+
+    expect(exported[:monitors]['http monitor'][:Status]).to eq UptimeRobot::Monitor::Status::Paused
+
+    watch_list do
+      <<-RUBY
+        monitor "http monitor" do
+          target "http://example.com"
+          interval 5
+          paused false
+          alert_contact :email, "#{WATCH_LIST_TEST_EMAIL}"
+          type :http
+        end
+
+        alert_contact do
+          type :email
+          value "#{WATCH_LIST_TEST_EMAIL}"
+        end
+      RUBY
+    end
+
+    exported = watch_list_export(skip_normalize: true) {|h|
+      h[:monitors]['http monitor'][:Status] != UptimeRobot::Monitor::Status::Paused
+    }
+
+    expect(exported[:monitors]['http monitor'][:Status]).to_not eq UptimeRobot::Monitor::Status::Paused
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,149 @@
+require 'watch_list'
+require 'tempfile'
+
+WATCH_LIST_TEST_API_KEY = ENV['WATCH_LIST_TEST_API_KEY']
+WATCH_LIST_TEST_PRIMARY_ALERT_CONTACT_ID = ENV['WATCH_LIST_TEST_PRIMARY_ALERT_CONTACT_ID']
+WATCH_LIST_TEST_EMAIL = ENV['WATCH_LIST_TEST_EMAIL']
+WATCH_LIST_TEST_EMAIL2 = ENV['WATCH_LIST_TEST_EMAIL2']
+LOOP_LIMIT = 60
+
+RSpec.configure do |config|
+  config.before(:each) do
+    cleanup_uptimerobot
+  end
+
+  config.after(:all) do
+    cleanup_uptimerobot(skip_wait: true)
+  end
+end
+
+def watch_list_client(options = {})
+  options = {
+    apiKey: WATCH_LIST_TEST_API_KEY,
+  }.merge(options)
+
+  if ENV['DEBUG'] == '1'
+    options[:debug] = true
+  else
+    options[:logger] = Logger.new('/dev/null')
+  end
+
+  WatchList::Client.new(options)
+end
+
+def watch_list(options = {})
+  client = watch_list_client(options)
+
+  tempfile(yield) do |f|
+    client.apply(f.path)
+  end
+end
+
+def wait_until(options = {}, client = uptimerobot_client)
+  export = proc do
+    WatchList::Exporter.export(client, options)
+  end
+
+  exported = nil
+
+  loop_until do
+    exported = export.call
+    yield(exported)
+  end
+
+  exported
+end
+
+def watch_list_export(options = {}, &block)
+  exported = wait_until(options, &block)
+
+  unless options[:skip_normalize]
+    exported[:monitors].each do |name, attrs|
+      attrs.delete(:ID)
+      attrs.delete(:Status)
+    end
+
+    exported[:alert_contacts].each do |alert_contact|
+      alert_contact.delete(:ID)
+    end
+  end
+
+  exported
+end
+
+def tempfile(content, options = {})
+  basename = "#{File.basename __FILE__}.#{$$}"
+  basename = [basename, options[:ext]] if options[:ext]
+
+  Tempfile.open(basename) do |f|
+    f.puts(content)
+    f.flush
+    f.rewind
+    yield(f)
+  end
+end
+
+def uptimerobot_client(options = {})
+  options = {
+    apiKey: WATCH_LIST_TEST_API_KEY,
+  }.merge(options)
+
+  UptimeRobot::Client.new(apiKey: WATCH_LIST_TEST_API_KEY)
+end
+
+def cleanup_uptimerobot(options = {})
+  client = uptimerobot_client
+  cleanup_uptimerobot_monitor(client, options)
+  cleanup_uptimerobot_alert_contact(client, options)
+end
+
+def cleanup_uptimerobot_monitor(client, options = {})
+  get_monitors = proc do
+    client.getMonitors['monitors']['monitor']
+  end
+
+  monitors = get_monitors.call
+  return if monitors.empty?
+
+  monitors.each do |monitor|
+    client.deleteMonitor(monitorID: monitor['id'])
+  end
+
+  return if options[:skip_wait]
+
+  loop_until do
+    get_monitors.call.length.zero?
+  end
+end
+
+def cleanup_uptimerobot_alert_contact(client, options = {})
+  get_alert_contacts = proc do
+    alert_contacts = client.getAlertContacts['alertcontacts']['alertcontact']
+    alert_contacts.select {|i| i['id'] != WATCH_LIST_TEST_PRIMARY_ALERT_CONTACT_ID }
+  end
+
+  alert_contacts = get_alert_contacts.call
+  return if alert_contacts.empty?
+
+  alert_contacts.each do |alert_contact|
+    client.deleteAlertContact(alertContactID: alert_contact['id'])
+  end
+
+  return if options[:skip_wait]
+
+  loop_until do
+    get_alert_contacts.call.length.zero?
+  end
+end
+
+def loop_until
+  break_loop = false
+
+  LOOP_LIMIT.times do
+    break_loop = yield
+    break if break_loop
+    sleep 1
+  end
+
+  raise 'wait timeout' unless break_loop
+end