// TestNewNotifier tests building the pinger object. func TestNewNotifier(t *testing.T) { site := getTestSite() n := notifier.NewNotifier(site, "Site 1 responding OK", "Site 1 Up", notifier.SendEmailMock, notifier.SendSmsMock) // Verify the first contact was Loaded with proper attributes and sorted last. if !reflect.DeepEqual(site.Contacts, n.Site.Contacts) { t.Error("Incoming site contacts are not the same as the notifier contacts:\n", site.Contacts, n.Site.Contacts) } }
// TestNotifyError tests calling the Notifications with errors on each send method. func TestNotifyError(t *testing.T) { site := getTestSite() n := notifier.NewNotifier(site, "Site 1 responding OK", "Site 1 Up", notifier.SendEmailErrorMock, notifier.SendSmsErrorMock) n.Notify() results, err := pinger.GetLogContent() if err != nil { t.Fatal("Failed to get log results.", err) } if !strings.Contains(results, "Error sending SMS: Error - no response from server.") { t.Error("Failed to report error in SMS send to Jack Contact.") } if !strings.Contains(results, "Error sending email: Error - no response from server.") { t.Error("Failed to report error in email send to Joe Contact.") } }
// TestNotify tests calling the Notifications successfully. func TestNotify(t *testing.T) { pinger.CreatePingerLog("", true) site := getTestSite() n := notifier.NewNotifier(site, "Site 1 responding OK", "Site 1 Up", notifier.SendEmailMock, notifier.SendSmsMock) n.Notify() results, err := pinger.GetLogContent() if err != nil { t.Fatal("Failed to get log results.", err) } if !strings.Contains(results, "Sending notifications for Jack Contact Site 1 Up Site 1 responding OK") { t.Error("Failed to report successful send to Jack Contact.") } if !strings.Contains(results, "Sending notifications for Joe Contact Site 1 Up Site 1 responding OK") { t.Error("Failed to report successful send to Joe Contact.") } }
// ping does the actual pinging of the site and calls the notifications func ping(s database.Site, db *sql.DB, requestURL URLRequester, sendEmail notifier.EmailSender, sendSms notifier.SmsSender, wg *sync.WaitGroup, stop chan struct{}) { defer wg.Done() // Initialize the previous state of site to the database value. On site creation will initialize to true. siteWasUp := s.IsSiteUp var statusChange bool var partialDetails string var partialSubject string for { // initialize statusChange to false and only notify on change of siteWasUp status statusChange = false // Check for a quit signal to stop the pinging select { case <-stop: log.Println("Stopping ", s.Name) return case <-time.After(time.Duration(s.PingIntervalSeconds) * time.Second): // Do nothing } if !s.IsActive { log.Println(s.Name, "Paused") continue } bodyContent, statusCode, responseTime, err := requestURL(s.URL, s.TimeoutSeconds) log.Println(s.Name, "Pinged") // Setup ping information for recording. p := database.Ping{SiteID: s.SiteID, TimeRequest: time.Now()} if err != nil { // Check if the error is due to the Internet not being Accessible if _, ok := err.(InternetAccessError); ok { log.Println(s.Name, "Unable to determine site status -", err) continue } log.Println(s.Name, "Error", err) if siteWasUp { statusChange = true partialSubject = "Site is Down" partialDetails = "Site is down, Error is " + err.Error() } siteWasUp = false } else if statusCode < 200 || statusCode > 299 { // Check if the status code is in the 2xx range. log.Println(s.Name, "Error - HTTP Status Code is", statusCode) if siteWasUp { statusChange = true partialSubject = "Site is Down" partialDetails = "Site is down, HTTP Status Code is " + strconv.Itoa(statusCode) + "." } siteWasUp = false } else { siteUp := true // if the site settings require check the content. if siteUp && s.ContentExpected != "" && !strings.Contains(bodyContent, s.ContentExpected) { siteUp = false log.Println(s.Name, "Error - required body content missing: ", s.ContentExpected) if siteWasUp { statusChange = true partialSubject = "Site is Down" partialDetails = "Site is Down, required body content missing: " + s.ContentExpected + "." } } if siteUp && s.ContentUnexpected != "" && strings.Contains(bodyContent, s.ContentUnexpected) { siteUp = false log.Println(s.Name, "Error - body content content has excluded content: ", s.ContentUnexpected) if siteWasUp { statusChange = true partialSubject = "Site is Down" partialDetails = "Site is Down, body content content has excluded content: " + s.ContentUnexpected + "." } } if siteUp && !siteWasUp { statusChange = true partialSubject = "Site is Up" partialDetails = fmt.Sprintf("Site is now up, response time was %v.", responseTime) siteWasUp = true } siteWasUp = siteUp } // Save the ping details p.Duration = int(responseTime.Nanoseconds() / 1e6) p.HTTPStatusCode = statusCode p.SiteDown = !siteWasUp // Save ping to db. err = p.CreatePing(db) if err != nil { log.Println("Error saving to ping to db:", err) } // Do the notifications if applicable if statusChange { // Update the site Status err = s.UpdateSiteStatus(db, siteWasUp) if err != nil { log.Println("Error updating site status:", err) } // Do the notifications if applicable subject := s.Name + ": " + partialSubject details := s.Name + " at " + s.URL + ": " + partialDetails log.Println("Will notify status change for", s.Name+":", details) n := notifier.NewNotifier(s, details, subject, sendEmail, sendSms) n.Notify() } } }