nagios-herald 0.0.2

data/test/env_files/nagios_vars.EXAMPLE

@@ -0,0 +1,197 @@
+NAGIOS_SERVICESTATE=CRITICAL
+NAGIOS_HOSTNOTIFICATIONNUMBER=0
+NAGIOS_HOSTNOTIFICATIONID=0
+NAGIOS_ARG8=
+NAGIOS_SERVICEACKAUTHOR=ops
+NAGIOS_COMMANDFILE=/usr/nagios/var/rw/nagios.cmd
+NAGIOS_ARG9=
+NAGIOS_HOSTADDRESS=10.101.194.198
+NAGIOS_LASTHOSTDOWN=0
+NAGIOS_LASTSERVICEWARNING=0
+NAGIOS_TEMPPATH=/tmp/nagios
+NAGIOS_TOTALHOSTSERVICESCRITICAL=1
+NAGIOS_HOSTACKAUTHOR=
+NAGIOS_TEMPFILE=/usr/nagios/var/nagios.tmp
+NAGIOS_NOTIFICATIONAUTHOR=
+NAGIOS_ISVALIDTIME=
+TERM=screen-256color
+NAGIOS_SERVICEGROUPALIAS=
+NAGIOS_ARG28=
+NAGIOS_HOSTEXECUTIONTIME=4.006
+NAGIOS_SERVICEACKCOMMENT=
+NAGIOS_ARG29=
+NAGIOS_LOGFILE=/usr/nagios/var/nagios.log
+NAGIOS_TOTALHOSTSUNREACHABLEUNHANDLED=
+NAGIOS_SERVICEACKAUTHORNAME=
+NAGIOS_NEXTVALIDTIME=
+NAGIOS_ARG26=
+NAGIOS_TIMET=1368738398
+NAGIOS_HOSTGROUPNAMES=test
+NAGIOS_MAXSERVICEATTEMPTS=3
+NAGIOS_ARG1=
+NAGIOS_ARG27=
+NAGIOS_HOSTSTATEID=0
+NAGIOS_LASTSERVICECRITICAL=1368738397
+NAGIOS_ARG2=
+NAGIOS_ARG24=
+NAGIOS_EVENTSTARTTIME=1367776508
+NAGIOS_ARG3=
+NAGIOS_ARG25=
+NAGIOS_HOSTACKCOMMENT=
+NAGIOS_SERVICENOTES=
+NAGIOS_SERVICEGROUPNOTESURL=
+NAGIOS_HOSTGROUPMEMBERS=web.example.com
+NAGIOS_NOTIFICATIONCOMMENT=
+NAGIOS_ARG4=
+NAGIOS_ARG22=
+NAGIOS_TOTALHOSTPROBLEMSUNHANDLED=
+NAGIOS_HOSTEVENTID=0
+NAGIOS_ARG5=
+NAGIOS_ARG23=
+NAGIOS_TOTALHOSTSERVICESOK=0
+NAGIOS_ARG6=
+NAGIOS_ARG20=
+NAGIOS_TIME=21:06:38
+NAGIOS_HOSTACTIONURL=
+NAGIOS_CONTACTGROUPALIAS=People who care about continuous integration servers
+NAGIOS_ARG7=
+NAGIOS_ARG21=
+NAGIOS_SERVICEPERFDATA=/=31269MB;36287;2015;0;40319 /dev/shm=82MB;2236;124;0;2485 /data=44700860MB;54876558;3048697;0;60973954
+NAGIOS_LASTSERVICESTATE=CRITICAL
+NAGIOS_CONTACTNAME=ops
+NAGIOS_HOSTPERCENTCHANGE=0.00
+NAGIOS_HOSTGROUPALIAS=Test Host Group
+LD_LIBRARY_PATH=:/usr/local/lib
+NAGIOS_RESOURCEFILE=/usr/nagios/etc/resource.cfg
+NAGIOS_CONTACTGROUPMEMBERS=ops
+NAGIOS_ARG19=
+NAGIOS_SERVICECHECKTYPE=ACTIVE
+NAGIOS_ARG18=
+NAGIOS_SERVICEATTEMPT=3
+NAGIOS_TOTALHOSTSDOWNUNHANDLED=
+NAGIOS_HOSTDOWNTIME=0
+NAGIOS_HOSTCHECKCOMMAND=check-host-alive
+NAGIOS_TOTALHOSTSERVICESUNKNOWN=0
+NAGIOS_SERVICEGROUPNAME=
+NAGIOS_SERVICEGROUPACTIONURL=
+NAGIOS_LASTSERVICESTATECHANGE=1368738086
+NAGIOS_RETENTIONDATAFILE=/usr/nagios/var/retention.dat
+NAGIOS_TOTALSERVICEPROBLEMSUNHANDLED=
+NAGIOS_LASTSERVICEEVENTID=8
+NAGIOS_TOTALHOSTSERVICES=1
+NAGIOS_CONTACTGROUPNAME=test
+NAGIOS_ARG13=
+NAGIOS_CONTACTPAGER=
+NAGIOS_NOTIFICATIONTYPE=PROBLEM
+NAGIOS_LASTSERVICEPROBLEMID=3
+NAGIOS_ARG12=
+NAGIOS_LASTHOSTCHECK=1368738363
+NAGIOS_ADMINPAGER=ops@example.com
+NAGIOS_SERVICECHECKCOMMAND=check_nrpe!check_disk
+NAGIOS_TOTALSERVICESWARNING=
+NAGIOS_LASTHOSTSTATEID=0
+NAGIOS_ARG11=
+NAGIOS_HOSTPERFDATA=rta=0.496000ms;3000.000000;5000.000000;0.000000 pl=0%;80;100;0
+NAGIOS_LASTSERVICEOK=1368737846
+NAGIOS_TOTALHOSTSDOWN=
+NAGIOS_SERVICENOTIFICATIONNUMBER=1
+NAGIOS_CONTACTGROUPNAMES=test
+NAGIOS_ARG10=
+NAGIOS_SERVICEPERCENTCHANGE=6.25
+NAGIOS_TOTALHOSTPROBLEMS=
+NAGIOS_TOTALSERVICESOK=
+NAGIOS_TOTALHOSTSERVICESWARNING=0
+NAGIOS_ARG17=
+PATH=/sbin:/usr/sbin:/bin:/usr/bin
+NAGIOS_HOSTOUTPUT=PING OK - Packet loss = 0%, RTA = 0.50 ms
+NAGIOS_NOTIFICATIONRECIPIENTS=ops
+NAGIOS_ARG16=
+NAGIOS_MAXHOSTATTEMPTS=1
+NAGIOS_ARG15=
+NAGIOS_TOTALHOSTSUNREACHABLE=
+NAGIOS_TOTALSERVICESWARNINGUNHANDLED=
+NAGIOS_ARG14=
+NAGIOS_STATUSDATAFILE=/usr/nagios/var/status.dat
+NAGIOS_TOTALSERVICESUNKNOWN=
+NAGIOS_LASTHOSTPROBLEMID=0
+NAGIOS_ARG31=
+PWD=/
+NAGIOS_SHORTDATETIME=05-16-2013 21:06:38
+NAGIOS_HOSTSTATE=UP
+NAGIOS_LASTHOSTUP=1368738368
+NAGIOS_SERVICEGROUPNOTES=
+NAGIOS_ARG30=
+NAGIOS_SERVICEPERFDATAFILE=
+NAGIOS_ARG32=
+LANG=en_US.UTF-8
+NAGIOS_CONTACTALIAS=Ops Engineer
+NAGIOS_HOSTNOTESURL=
+NAGIOS_SERVICELATENCY=0.189
+NAGIOS_TOTALSERVICESCRITICAL=
+NAGIOS_SERVICEDOWNTIME=0
+NAGIOS_SERVICEGROUPNAMES=
+NAGIOS_DATE=05-16-2013
+NAGIOS_SERVICEEVENTID=9
+NAGIOS_HOSTACKAUTHORALIAS=
+NAGIOS_NOTIFICATIONAUTHORALIAS=
+NAGIOS_SERVICEDURATION=0d 0h 5m 12s
+NAGIOS_HOSTSTATETYPE=HARD
+NAGIOS_SERVICEPROBLEMID=4
+NAGIOS_HOSTCHECKTYPE=ACTIVE
+NAGIOS_HOSTGROUPNOTES=
+NAGIOS_LASTSERVICECHECK=1368738397
+NAGIOS_ADMINEMAIL=ops@example.com
+NAGIOS_OBJECTCACHEFILE=/usr/nagios/var/objects.cache
+NAGIOS_HOSTLATENCY=0.163
+NAGIOS_HOSTGROUPACTIONURL=
+SHLVL=4
+NAGIOS_LONGDATETIME=Thu May 16 21:06:38 UTC 2013
+NAGIOS_HOSTDISPLAYNAME=web.example.com
+NAGIOS_SERVICESTATEID=2
+NAGIOS_SERVICENOTESURL=
+NAGIOS_TOTALSERVICEPROBLEMS=
+NAGIOS_HOSTPROBLEMID=0
+NAGIOS_CONTACTADDRESS0=
+NAGIOS_HOSTDURATIONSEC=966190
+NAGIOS_TOTALSERVICESUNKNOWNUNHANDLED=
+NAGIOS_LASTHOSTSTATE=UP
+NAGIOS_CONTACTADDRESS1=
+NAGIOS_SERVICEEXECUTIONTIME=0.028
+NAGIOS_TOTALSERVICESCRITICALUNHANDLED=
+NAGIOS_CONTACTADDRESS2=
+NAGIOS_LONGHOSTOUTPUT=
+NAGIOS_SERVICEGROUPMEMBERS=
+NAGIOS_CONTACTADDRESS3=
+NAGIOS_SERVICEDURATIONSEC=312
+NAGIOS_CONTACTADDRESS4=
+NAGIOS_HOSTGROUPNOTESURL=
+NAGIOS_CONTACTADDRESS5=
+NAGIOS_SERVICEOUTPUT=DISK CRITICAL - free space: / 7002 MB (18% inode 60%): /data 16273093 MB (26% inode 99%):
+NAGIOS_NOTIFICATIONNUMBER=1
+NAGIOS_HOSTDURATION=11d 4h 23m 10s
+NAGIOS_MAINCONFIGFILE=/usr/nagios/etc/nagios.cfg
+NAGIOS_CONTACTEMAIL=ops@example.com
+NAGIOS_SERVICESTATETYPE=HARD
+NAGIOS_LASTHOSTUNREACHABLE=0
+NAGIOS_LONGSERVICEOUTPUT=\nTHRESHOLDS - WARNING:15%;CRITICAL:20%;\n\nFilesystem            Size  Used Avail Use% Mounted on\n/dev/vda               40G   31G  6.9G  82% /\ntmpfs                 2.5G   83M  2.4G   4% /dev/shm\nnfs.example.example.com:/mnt/user/homes\n                       59T   43T   16T  74% /data\n
+NAGIOS_HOSTATTEMPT=1
+NAGIOS_HOSTNOTES=
+NAGIOS_SERVICEISVOLATILE=0
+NAGIOS_SERVICEACTIONURL=http://runbook.example.com/disk_space_alerts.html
+NAGIOS_SERVICEDISPLAYNAME=Disk Space
+NAGIOS_TOTALHOSTSUP=
+NAGIOS_LASTHOSTSTATECHANGE=1367772208
+NAGIOS_LASTSERVICEUNKNOWN=0
+NAGIOS_HOSTPERFDATAFILE=
+NAGIOS_LASTHOSTEVENTID=0
+NAGIOS_HOSTACKAUTHORNAME=
+NAGIOS_NOTIFICATIONAUTHORNAME=
+NAGIOS_HOSTALIAS=web.example.com
+NAGIOS_SERVICEDESC=Disk Space
+NAGIOS_HOSTGROUPNAME=test
+NAGIOS_PROCESSSTARTTIME=1367776508
+NAGIOS_HOSTNAME=web.example.com
+NAGIOS_SERVICENOTIFICATIONID=45
+NAGIOS_SERVICEACKAUTHORALIAS=
+NAGIOS_LASTSERVICESTATEID=2
+_=/bin/env

data/test/unit/test_config.rb

@@ -0,0 +1,31 @@
+require 'minitest/autorun'
+
+# I assume cat'ing to the LOAD path goes away when we're a real gem.
+$LOAD_PATH.unshift File.join(File.dirname(__FILE__), '..', '..', 'lib')
+require 'nagios-herald'
+require 'nagios-herald/config'
+
+# Test the Config module.
+class TestConfig < MiniTest::Unit::TestCase
+
+  # Initial setup before we execute tests
+  def setup
+    @options = {}
+    @options['config_file'] = File.expand_path(File.join(File.dirname(__FILE__), '..', '..', 'etc', 'config.yml.example'))
+    @options['message_type'] = "pager"
+  end
+
+  def teardown
+  end
+
+  # Make sure we can load the config and read values from it.
+  # Read values from command-line options and the config file.
+  def test_load_config
+    NagiosHerald::Config.load(@options)
+    assert_equal "pager", NagiosHerald::Config.config['message_type']                       # Command line
+    assert_equal "ganglia.example.com", NagiosHerald::Config.config['servers']['ganglia']   # Config file
+    assert_equal "splunkuser",  NagiosHerald::Config.config['splunk']['username']           # Config file
+    assert_equal "http://logstash.example.com:9200",  NagiosHerald::Config.config['logstash']['url']           # Config file
+  end
+
+end

data/test/unit/test_executor.rb

@@ -0,0 +1,65 @@
+require 'minitest/autorun'
+
+# I assume cat'ing to the LOAD path goes away when we're a real gem.
+$LOAD_PATH.unshift File.join(File.dirname(__FILE__), '..', '..', 'lib')
+require 'nagios-herald'
+require 'nagios-herald/executor'
+
+# Test the Executor class.
+class TestExecutor < MiniTest::Unit::TestCase
+  include NagiosHerald::Util
+
+  # initial setup before we execute tests
+  def setup
+    @executor = NagiosHerald::Executor.new
+    @options = {}
+    @options[:env] = File.join(File.dirname(__FILE__), '..', 'env_files', 'nagios_vars.EXAMPLE')
+  end
+
+  def teardown
+  end
+
+  # We expect a NagiosHerald::Executor object.
+  def test_new_executor
+    assert_instance_of NagiosHerald::Executor, @executor
+  end
+
+  # Make sure we can load an environment file and read some env variables.
+  # We'll use Util::get_nagios_var just like Executor does.
+  def test_load_env_file
+    @executor.load_env_from_file(@options[:env])
+    assert_equal "ops@example.com", get_nagios_var('NAGIOS_CONTACTEMAIL')
+    assert_equal "PROBLEM",  get_nagios_var('NAGIOS_NOTIFICATIONTYPE')
+  end
+
+  # Get the file names of each of the formatter classes and confirm they load properly.
+  # This helps ensure formatters are named and subclassed properly.
+  def test_load_formatters
+    formatter_dir = File.join(File.dirname(__FILE__), '..', '..', 'lib', 'nagios-herald', 'formatters')
+    formatter_class_files = Dir["#{formatter_dir}/*.rb"]
+    @executor.load_formatters
+    formatter_class_files.each do |formatter_class_file|
+      class_file = File.basename(formatter_class_file, '.rb')   # strip the extension
+      next if class_file.eql?('base')   # base.rb doesn't get loaded
+      assert NagiosHerald::Formatter.formatters.has_key?(class_file), "'#{formatter_dir}/#{class_file}.rb' was not loaded. "\
+      "Check that it's named and subclassed properly.\n"\
+      "See https://github.com/etsy/nagios-herald/blob/master/docs/formatters.md for help."
+    end
+  end
+
+  # Get the file names of each of the message classes and confirm they load properly.
+  # This helps ensure messages are named and subclassed properly.
+  def test_load_messages
+    message_dir = File.join(File.dirname(__FILE__), '..', '..', 'lib', 'nagios-herald', 'messages')
+    message_class_files = Dir["#{message_dir}/*.rb"]
+    @executor.load_messages
+    message_class_files.each do |message_class_file|
+      class_file = File.basename(message_class_file, '.rb')   # strip the extension
+      next if class_file.eql?('base')   # base.rb doesn't get loaded
+      assert NagiosHerald::Message.message_types.has_key?(class_file), "'#{message_dir}/#{class_file}.rb' was not loaded. "\
+      "Check that it's named and subclassed properly.\n"\
+      "See https://github.com/etsy/nagios-herald/blob/master/docs/messages.md for help."
+    end
+  end
+
+end

data/test/unit/test_formatter_base.rb

@@ -0,0 +1,131 @@
+require 'minitest/autorun'
+
+# I assume cat'ing to the LOAD path goes away when we're a real gem.
+$LOAD_PATH.unshift File.join(File.dirname(__FILE__), '..', '..', 'lib')
+require 'nagios-herald'
+require 'nagios-herald/config'
+require 'nagios-herald/executor'
+require 'nagios-herald/formatters/base'
+
+# Test Formatter::Base.
+class TestFormatterBase < MiniTest::Unit::TestCase
+
+  # TODO: We need a similar set of tests for RECOVERY emails.
+  # Initial setup before we execute tests
+  def setup
+    @options = {}
+    @options[:message_type] = 'EMAIL'
+    @options[:nagios_url] = "http://nagios.example.com"
+    @formatter = NagiosHerald::Formatter.new(@options)
+    env_file = File.join(File.dirname(__FILE__), '..', 'env_files', 'nagios_vars.EXAMPLE')
+    NagiosHerald::Executor.new.load_env_from_file(env_file) # load an env file for testing
+  end
+
+  def teardown
+    # make certain we don't leave tons of empty temp dirs behind
+    @formatter.clean_sandbox
+  end
+
+  # Test that we have a new NagiosHerald::Formatter object.
+  def test_new_formatter
+    assert_instance_of NagiosHerald::Formatter, @formatter
+  end
+
+  def test_add_content_basic
+    @formatter.add_text('test_add_content', 'This is test text')
+    assert_equal 'This is test text', @formatter.content[:text][:test_add_content]
+    @formatter.add_html('test_add_content', '<b>This is test HTML</b>')
+    assert_equal '<b>This is test HTML</b>', @formatter.content[:html][:test_add_content]
+    @formatter.generate_subject
+    assert_equal "PROBLEM Service web.example.com/Disk Space is CRITICAL", @formatter.content[:subject]
+    attachment_name = "#{@formatter.sandbox}/cat.gif"
+    @formatter.add_attachment(attachment_name)
+    assert @formatter.content[:attachments].include?(attachment_name), "Failed to attach #{attachment_name} to content hash."
+  end
+
+  def test_action_url
+    @formatter.action_url
+    assert_equal "<b>Action URL</b>: http://runbook.example.com/disk_space_alerts.html<br><br>", @formatter.content[:html][:action_url]
+    assert_equal "Action URL: http://runbook.example.com/disk_space_alerts.html\n\n", @formatter.content[:text][:action_url]
+  end
+
+  def test_host_info
+    @formatter.host_info
+    assert_equal "<br><b>Host</b>: web.example.com <b>Service</b>: Disk Space<br/><br>", @formatter.content[:html][:host_info]
+    assert_equal "Host: web.example.com Service: Disk Space\n\n", @formatter.content[:text][:host_info]
+  end
+
+  def test_state_info
+    @formatter.state_info
+    assert_equal "State is now: <b><font style='color:red'>CRITICAL</font></b> for <b>0d 0h 5m 12s</b> (was CRITICAL) after <b>3 / 3</b> checks<br/><br>", @formatter.content[:html][:state_info]
+    assert_equal "State is now: CRITICAL for 0d 0h 5m 12s (was CRITICAL) after 3 / 3 checks\n\n", @formatter.content[:text][:state_info]
+  end
+
+  def test_notification_info
+    @formatter.notification_info
+    assert_equal "Notification sent at: Thu May 16 21:06:38 UTC 2013 (notification number 1)<br><br>", @formatter.content[:html][:notification_info]
+    assert_equal "Notification sent at: Thu May 16 21:06:38 UTC 2013 (notification number 1)\n\n", @formatter.content[:text][:notification_info]
+  end
+
+  def test_additional_info
+    @formatter.additional_info
+    assert_equal "<b>Additional Info</b>: DISK CRITICAL - free space: / 7002 MB (18% inode 60%): /data 16273093 MB (26% inode 99%):<br><br>", @formatter.content[:html][:additional_info]
+    assert_equal "Additional Info: DISK CRITICAL - free space: / 7002 MB (18% inode 60%): /data 16273093 MB (26% inode 99%):\n\n", @formatter.content[:text][:additional_info]
+  end
+
+  def test_additional_details
+    @formatter.additional_details
+    assert_equal "<b>Additional Details</b>: <pre>\nTHRESHOLDS - WARNING:15%;CRITICAL:20%;\n\nFilesystem            Size  Used Avail Use% Mounted on\n/dev/vda               40G   31G  6.9G  82% /\ntmpfs                 2.5G   83M  2.4G   4% /dev/shm\nnfs.example.example.com:/mnt/user/homes\n                       59T   43T   16T  74% /data\n</pre><br><br>", @formatter.content[:html][:additional_details]
+    assert_equal "Additional Details: \nTHRESHOLDS - WARNING:15%;CRITICAL:20%;\n\nFilesystem            Size  Used Avail Use% Mounted on\n/dev/vda               40G   31G  6.9G  82% /\ntmpfs                 2.5G   83M  2.4G   4% /dev/shm\nnfs.example.example.com:/mnt/user/homes\n                       59T   43T   16T  74% /data\n\n", @formatter.content[:text][:additional_details]
+  end
+
+  def test_notes
+    @formatter.notes
+    # There are no notes in the example environment variables.
+    assert_equal "", @formatter.content[:html][:notes]
+    assert_equal "", @formatter.content[:text][:notes]
+  end
+
+  def test_action_url
+    @formatter.action_url
+    assert_equal "<b>Action URL</b>: http://runbook.example.com/disk_space_alerts.html<br><br>", @formatter.content[:html][:action_url]
+    assert_equal "Action URL: http://runbook.example.com/disk_space_alerts.html\n\n", @formatter.content[:text][:action_url]
+  end
+
+  def test_short_state_detail
+    @formatter.short_state_detail
+    assert_equal "DISK CRITICAL - free space: / 7002 MB (18% inode 60%): /data 16273093 MB (26% inode 99%):<br>", @formatter.content[:html][:short_state_detail]
+    assert_equal "DISK CRITICAL - free space: / 7002 MB (18% inode 60%): /data 16273093 MB (26% inode 99%):\n", @formatter.content[:text][:short_state_detail]
+  end
+
+  def test_recipients_email_link
+    @formatter.recipients_email_link
+    assert_equal "Sent to ops<br><br>", @formatter.content[:html][:recipients_email_link]
+    assert_equal "Sent to ops\n\n", @formatter.content[:text][:recipients_email_link]
+  end
+
+  def test_ack_info
+    @formatter.ack_info
+    assert_equal "At Thu May 16 21:06:38 UTC 2013 ops acknowledged web.example.com/Disk Space.<br><br>Comment: ", @formatter.content[:html][:ack_info]
+    assert_equal "At Thu May 16 21:06:38 UTC 2013 ops acknowledged web.example.com/Disk Space.\n\nComment: ", @formatter.content[:text][:ack_info]
+  end
+
+  def test_short_ack_info
+    @formatter.short_ack_info
+    assert_equal "ops  ack'd Disk Space on web.example.com.<br>", @formatter.content[:html][:short_ack_info]
+    assert_equal "ops  ack'd Disk Space on web.example.com.\n", @formatter.content[:text][:short_ack_info]
+  end
+
+  def test_alert_ack_url
+    @formatter.alert_ack_url
+    assert_equal "Acknowledge this alert: http://nagios.example.com/nagios/cgi-bin/cmd.cgi?cmd_typ=34&host=web.example.com&service=Disk%20Space<br>Alternatively, <b>reply</b> to this message with the word '<b><font color='green'>ack</font></b>' in the body to acknowledge the alert.<br>", @formatter.content[:html][:alert_ack_url]
+    assert_equal "Acknowledge this alert: http://nagios.example.com/nagios/cgi-bin/cmd.cgi?cmd_typ=34&host=web.example.com&service=Disk%20Space\nAlternatively, reply to this message with the word 'ack' in the body to acknowledge the alert.\n", @formatter.content[:text][:alert_ack_url]
+  end
+
+  def test_clean_sandbox
+    @formatter.clean_sandbox
+    assert !File.directory?(@formatter.sandbox)
+  end
+
+end
+

data/test/unit/test_formatter_check_cpu_idle_critical.rb

@@ -0,0 +1,135 @@
+require 'minitest/autorun'
+
+$LOAD_PATH.unshift File.join(File.dirname(__FILE__), '..', '..', 'lib')
+require 'nagios-herald'
+require 'nagios-herald/config'
+require 'nagios-herald/executor'
+require 'nagios-herald/formatters/base'
+
+# Test Formatter::Base.
+class TestFormatterCheckCpuIdle < MiniTest::Unit::TestCase
+
+  # TODO: We need a similar set of tests for RECOVERY emails.
+  # Initial setup before we execute tests
+  def setup
+    @options = {}
+    @options[:message_type] = 'EMAIL'
+    @options[:nagios_url] = "http://nagios.example.com"
+    @options[:formatter_name] = 'check_cpu'
+    env_file = File.join(File.dirname(__FILE__), '..', 'env_files', 'check_cpu_idle.CRITICAL')
+    NagiosHerald::Executor.new.load_env_from_file(env_file) # load an env file for testing
+    NagiosHerald::Executor.new.load_formatters
+    NagiosHerald::Executor.new.load_messages
+    formatter_class = NagiosHerald::Formatter.formatters[@options[:formatter_name]]
+    @formatter = formatter_class.new(@options)
+  end
+
+  def teardown
+    # make certain we don't leave tons of empty temp dirs behind
+    @formatter.clean_sandbox
+  end
+
+  # Test that we have a new NagiosHerald::Formatter object.
+  def test_new_formatter
+    assert_instance_of NagiosHerald::Formatter::CheckCpu, @formatter
+  end
+
+  def test_add_content_basic
+    @formatter.add_text('test_add_content', 'This is test text')
+    assert_equal 'This is test text', @formatter.content[:text][:test_add_content]
+    @formatter.add_html('test_add_content', '<b>This is test HTML</b>')
+    assert_equal '<b>This is test HTML</b>', @formatter.content[:html][:test_add_content]
+    @formatter.generate_subject
+    assert_equal "PROBLEM Service web.example.com/CPU is CRITICAL", @formatter.content[:subject]
+    attachment_name = "#{@formatter.sandbox}/cat.gif"
+    @formatter.add_attachment(attachment_name)
+    assert @formatter.content[:attachments].include?(attachment_name), "Failed to attach #{attachment_name} to content hash."
+  end
+
+  def test_action_url
+    @formatter.action_url
+    assert_equal "<b>Action URL</b>: http://runbook.example.com/disk_space_alerts.html<br><br>", @formatter.content[:html][:action_url]
+    assert_equal "Action URL: http://runbook.example.com/disk_space_alerts.html\n\n", @formatter.content[:text][:action_url]
+  end
+
+  def test_host_info
+    @formatter.host_info
+    assert_equal "<br><b>Host</b>: web.example.com <b>Service</b>: CPU<br/><br>", @formatter.content[:html][:host_info]
+    assert_equal "Host: web.example.com Service: CPU\n\n", @formatter.content[:text][:host_info]
+  end
+
+  def test_state_info
+    @formatter.state_info
+    assert_equal "State is now: <b><font style='color:red'>CRITICAL</font></b> for <b>0d 0h 0m 14s</b> (was WARNING) after <b>3 / 3</b> checks<br/><br>", @formatter.content[:html][:state_info]
+    assert_equal "State is now: CRITICAL for 0d 0h 0m 14s (was WARNING) after 3 / 3 checks\n\n", @formatter.content[:text][:state_info]
+  end
+
+  def test_notification_info
+    @formatter.notification_info
+    assert_equal "Notification sent at: Sat May 17 01:34:07 UTC 2014 (notification number 2)<br><br>", @formatter.content[:html][:notification_info]
+    assert_equal "Notification sent at: Sat May 17 01:34:07 UTC 2014 (notification number 2)\n\n", @formatter.content[:text][:notification_info]
+  end
+
+  def test_additional_info
+    @formatter.additional_info
+    assert_equal "<b>Additional Info</b>:<br>CRITICAL CPU <b><font color='red'>idle</font></b> is < 100%: user=3.02% system=3.25% iowait=0.01% <b><font color='red'>idle=93.72%</font></b>", @formatter.content[:html][:additional_info]
+    assert_equal "Additional Info: CRITICAL CPU idle is < 100%: user=3.02% system=3.25% iowait=0.01% idle=93.72%", @formatter.content[:text][:additional_info]
+  end
+
+  def test_additional_details
+    @formatter.additional_details
+    puts
+    assert_equal "<b>Additional Details</b>:<pre><br>TOP 5 PROCESSES BY CPU:<br> %CPU         TIME         USER    PID COMMAND<br><font color='red'> 57.6     00:00:15         root  21751 /usr/bin/ruby /usr/bin/knife search node lastrun_status:success -i</font><br><font color='orange'>  4.0     00:00:00        larry  22001 ps -eo %cpu,cputime,user,pid,args --sort -%cpu</font><br><font color='orange'>  0.7     06:19:12       nobody  12161 /usr/sbin/gmond</font><br><font color='orange'>  0.6   1-02:11:15         root   1424 [kipmi0]</font><br><font color='orange'>  0.5     00:48:01        10231  15079 mosh-server new -s -c 8 -l LANG=en_US.UTF-8</font><br></pre><br>", @formatter.content[:html][:additional_details]
+    assert_equal "Additional Details:\n#TOP 5 PROCESSES BY CPU:\n %CPU         TIME         USER    PID COMMAND\n 57.6     00:00:15         root  21751 /usr/bin/ruby /usr/bin/knife search node lastrun_status:success -i\n  4.0     00:00:00        larry  22001 ps -eo %cpu,cputime,user,pid,args --sort -%cpu\n  0.7     06:19:12       nobody  12161 /usr/sbin/gmond\n  0.6   1-02:11:15         root   1424 [kipmi0]\n  0.5     00:48:01        10231  15079 mosh-server new -s -c 8 -l LANG=en_US.UTF-8\n\n\n", @formatter.content[:text][:additional_details]
+  end
+
+  def test_notes
+    @formatter.notes
+    # There are no notes in the example environment variables.
+    assert_equal "", @formatter.content[:html][:notes]
+    assert_equal "", @formatter.content[:text][:notes]
+  end
+
+  def test_action_url
+    @formatter.action_url
+    assert_equal "", @formatter.content[:html][:action_url]
+    assert_equal "", @formatter.content[:text][:action_url]
+  end
+
+  def test_short_state_detail
+    @formatter.short_state_detail
+    assert_equal "CRITICAL CPU idle is < 100%: user=3.02% system=3.25% iowait=0.01% idle=93.72%<br>", @formatter.content[:html][:short_state_detail]
+    assert_equal "CRITICAL CPU idle is < 100%: user=3.02% system=3.25% iowait=0.01% idle=93.72%\n", @formatter.content[:text][:short_state_detail]
+  end
+
+  def test_recipients_email_link
+    @formatter.recipients_email_link
+    assert_equal "Sent to ops<br><br>", @formatter.content[:html][:recipients_email_link]
+    assert_equal "Sent to ops\n\n", @formatter.content[:text][:recipients_email_link]
+  end
+
+  def test_ack_info
+    @formatter.ack_info
+    assert_equal "At Sat May 17 01:34:07 UTC 2014  acknowledged web.example.com/CPU.<br><br>Comment: ", @formatter.content[:html][:ack_info]
+    assert_equal "At Sat May 17 01:34:07 UTC 2014  acknowledged web.example.com/CPU.\n\nComment: ", @formatter.content[:text][:ack_info]
+  end
+
+  def test_short_ack_info
+    @formatter.short_ack_info
+    assert_equal "  ack'd CPU on web.example.com.<br>", @formatter.content[:html][:short_ack_info]
+    assert_equal "  ack'd CPU on web.example.com.\n", @formatter.content[:text][:short_ack_info]
+  end
+
+  def test_alert_ack_url
+    @formatter.alert_ack_url
+    assert_equal "Acknowledge this alert: http://nagios.example.com/nagios/cgi-bin/cmd.cgi?cmd_typ=34&host=web.example.com&service=CPU<br>Alternatively, <b>reply</b> to this message with the word '<b><font color='green'>ack</font></b>' in the body to acknowledge the alert.<br>", @formatter.content[:html][:alert_ack_url]
+    assert_equal "Acknowledge this alert: http://nagios.example.com/nagios/cgi-bin/cmd.cgi?cmd_typ=34&host=web.example.com&service=CPU\nAlternatively, reply to this message with the word 'ack' in the body to acknowledge the alert.\n", @formatter.content[:text][:alert_ack_url]
+  end
+
+  def test_clean_sandbox
+    @formatter.clean_sandbox
+    assert !File.directory?(@formatter.sandbox)
+  end
+
+end
+