// TestCreateUniqueSite tests that the same URL and Site Name can't be entered twice.
func TestCreateUniqueSite(t *testing.T) {
var err error
db, err := database.InitializeTestDB("")
if err != nil {
t.Fatal("Failed to create database:", err)
}
defer db.Close()
s := database.Site{Name: "Test", IsActive: true, URL: "http://www.test.com",
PingIntervalSeconds: 60, TimeoutSeconds: 30}
err = s.CreateSite(db)
if err != nil {
t.Fatal("Failed to create new site:", err)
}
//Test where the URL is the same and the Name is different should fail uniqueness constraint.
s2 := database.Site{Name: "Test2", IsActive: true, URL: "http://www.test.com",
PingIntervalSeconds: 60, TimeoutSeconds: 30}
err = s2.CreateSite(db)
if err == nil {
t.Fatal("Should throw uniqueness constraint error for URL.")
}
//Test where the Name is the same and the URL is different should fail with uniqueness constraint.
s3 := database.Site{Name: "Test", IsActive: true, URL: "http://www.test.edu",
PingIntervalSeconds: 60, TimeoutSeconds: 30}
err = s3.CreateSite(db)
if err == nil {
t.Fatal("Should throw uniqueness constraint error for Name.")
}
}
func addContactToSite(controller *contactsController, contactID int64, siteID int64) error {
var site database.Site
err := site.GetSite(controller.DB, siteID)
if err != nil {
return err
}
err = site.AddContactToSite(controller.DB, contactID)
if err != nil {
return err
}
return nil
}
func removeContactFromSite(controller *contactsController, contactID int64, siteID int64) error {
var site database.Site
err := site.GetSite(controller.DB, siteID)
if err != nil {
return err
}
err = site.RemoveContactFromSite(controller.DB, contactID)
if err != nil {
return err
}
return nil
}
func (controller *sitesController) newPost(rw http.ResponseWriter, req *http.Request) (int, error) {
err := req.ParseForm()
if err != nil {
return http.StatusInternalServerError, err
}
decoder := schema.NewDecoder()
// Ignore unknown keys to prevent errors from the CSRF token.
decoder.IgnoreUnknownKeys(true)
formSite := new(viewmodels.SitesEditViewModel)
err = decoder.Decode(formSite, req.PostForm)
if err != nil {
return http.StatusInternalServerError, err
}
valErrors := validateSiteForm(formSite)
if len(valErrors) > 0 {
isAuthenticated, user := getCurrentUser(rw, req, controller.authorizer)
var contacts database.Contacts
err = contacts.GetContacts(controller.DB)
if err != nil {
return http.StatusInternalServerError, err
}
vm := viewmodels.NewSiteViewModel(formSite, contacts, isAuthenticated, user, valErrors)
vm.CsrfField = csrf.TemplateField(req)
return http.StatusOK, controller.newTemplate.Execute(rw, vm)
}
site := database.Site{}
viewmodels.MapSiteVMtoDB(formSite, &site)
err = site.CreateSite(controller.DB)
if err != nil {
return http.StatusInternalServerError, err
}
//Add any selected contacts
for _, contactSelID := range formSite.SelectedContacts {
err = site.AddContactToSite(controller.DB, contactSelID)
if err != nil {
return http.StatusInternalServerError, err
}
}
// Refresh the pinger with the changes.
err = controller.pinger.UpdateSiteSettings()
if err != nil {
return http.StatusInternalServerError, err
}
http.Redirect(rw, req, "/settings", http.StatusSeeOther)
return http.StatusSeeOther, nil
}
// Create the struct for the Site and its contacts used for testing.
func getTestSite() database.Site {
s1 := database.Site{Name: "Test", IsActive: true, URL: "http://www.google.com",
PingIntervalSeconds: 2, TimeoutSeconds: 1}
// Create first contact
c1 := database.Contact{Name: "Joe Contact", EmailAddress: "*****@*****.**", SmsNumber: "5125551212",
SmsActive: false, EmailActive: true}
// Create second contact
c2 := database.Contact{Name: "Jack Contact", EmailAddress: "*****@*****.**", SmsNumber: "5125551213",
SmsActive: true, EmailActive: false}
// Add the contacts to the sites
s1.Contacts = append(s1.Contacts, c1, c2)
return s1
}
// TestGetSites tests the saving of the ping information to the DB.
func TestSavePings(t *testing.T) {
db, err := database.InitializeTestDB("")
if err != nil {
t.Fatal("Failed to create database:", err)
}
defer db.Close()
s1 := database.Site{Name: "Test", IsActive: true, URL: "http://www.github.com",
PingIntervalSeconds: 1, TimeoutSeconds: 30}
err = s1.CreateSite(db)
if err != nil {
t.Fatal("Failed to create new site:", err)
}
// For this test will pass the normal GetSites to use the DB...
pinger.ResetHitCount()
p := pinger.NewPinger(db, pinger.GetSites, pinger.RequestURLMock,
notifier.SendEmailMock, notifier.SendSmsMock)
p.Start()
// Sleep to allow running the tests before stopping.
time.Sleep(7 * time.Second)
p.Stop()
// Get the site pings since the test began and validate.
var saved database.Site
err = saved.GetSitePings(db, s1.SiteID, time.Now().Add(-10*time.Second), time.Now())
if err != nil {
t.Fatal("Failed to retrieve site pings:", err)
}
if !saved.Pings[0].SiteDown {
t.Error("First ping should show site down.")
}
if saved.Pings[3].SiteDown {
t.Error("Fourth ping should show site up.")
}
// Get the Site updates to make sure the status changes are being set.
err = saved.GetSite(db, s1.SiteID)
if err != nil {
t.Fatal("Failed to retrieve site updates:", err)
}
if saved.LastStatusChange.IsZero() {
t.Error("Last Status Change time not saved.")
}
if saved.LastPing.IsZero() {
t.Error("Last Ping time not saved.")
}
if !saved.IsSiteUp {
t.Error("Site should be saved as up.")
}
}
func (controller *sitesController) editGet(rw http.ResponseWriter, req *http.Request) (int, error) {
vars := mux.Vars(req)
siteID, err := strconv.ParseInt(vars["siteID"], 10, 64)
if err != nil {
return http.StatusInternalServerError, err
}
// Get the site to edit
site := new(database.Site)
err = site.GetSite(controller.DB, siteID)
if err != nil {
return http.StatusInternalServerError, err
}
// Get all of the contacts to display in the table.
var contacts database.Contacts
err = contacts.GetContacts(controller.DB)
if err != nil {
return http.StatusInternalServerError, err
}
// Also get the site contacts to display in a table.
err = site.GetSiteContacts(controller.DB, siteID)
if err != nil {
return http.StatusInternalServerError, err
}
selectedContacts := []int64{}
for _, contact := range site.Contacts {
selectedContacts = append(selectedContacts, contact.ContactID)
}
isAuthenticated, user := getCurrentUser(rw, req, controller.authorizer)
siteEdit := new(viewmodels.SitesEditViewModel)
viewmodels.MapSiteDBtoVM(site, siteEdit)
siteEdit.SelectedContacts = selectedContacts
vm := viewmodels.EditSiteViewModel(siteEdit, contacts, isAuthenticated, user, make(map[string]string))
vm.CsrfField = csrf.TemplateField(req)
return http.StatusOK, controller.editTemplate.Execute(rw, vm)
}
func (controller *sitesController) getDetails(rw http.ResponseWriter, req *http.Request) (int, error) {
vars := mux.Vars(req)
siteID, err := strconv.ParseInt(vars["siteID"], 10, 64)
if err != nil {
return http.StatusInternalServerError, err
}
site := new(database.Site)
err = site.GetSite(controller.DB, siteID)
if err != nil {
return http.StatusInternalServerError, err
}
// Also get the contacts to display in a table.
err = site.GetSiteContacts(controller.DB, siteID)
if err != nil {
return http.StatusInternalServerError, err
}
isAuthenticated, user := getCurrentUser(rw, req, controller.authorizer)
vm := viewmodels.GetSiteDetailsViewModel(site, isAuthenticated, user)
return http.StatusOK, controller.detailsTemplate.Execute(rw, vm)
}
// MapSiteVMtoDB maps the site view model properties to the site database properties.
func MapSiteVMtoDB(siteVM *SitesEditViewModel, site *database.Site) error {
site.SiteID = siteVM.SiteID
site.Name = siteVM.Name
site.IsActive = siteVM.IsActive
site.URL = strings.TrimSpace(siteVM.URL)
site.ContentExpected = strings.TrimSpace(siteVM.ContentExpected)
site.ContentUnexpected = strings.TrimSpace(siteVM.ContentUnexpected)
// Conversion on these two is necessary because they are a string in the
// view model to allow the validation to work
pingInterval, err := strconv.Atoi(siteVM.PingIntervalSeconds)
if err != nil {
return err
}
site.PingIntervalSeconds = pingInterval
timeout, err := strconv.Atoi(siteVM.TimeoutSeconds)
if err != nil {
return err
}
site.TimeoutSeconds = timeout
return nil
}
// TestGetSites tests the database retrieval of the list of sites.
func TestGetSites(t *testing.T) {
var sites database.Sites
db, err := database.InitializeTestDB("")
if err != nil {
t.Fatal("Failed to create database:", err)
}
defer db.Close()
s1 := database.Site{Name: "Test", IsActive: true, URL: "http://www.test.com",
PingIntervalSeconds: 1, TimeoutSeconds: 30}
err = s1.CreateSite(db)
if err != nil {
t.Fatal("Failed to create new site:", err)
}
// Create the second site.
s2 := database.Site{Name: "Test 2", IsActive: true, URL: "http://www.example.com",
PingIntervalSeconds: 1, TimeoutSeconds: 30}
err = s2.CreateSite(db)
if err != nil {
t.Fatal("Failed to create second site:", err)
}
sites, err = pinger.GetSites(db)
if err != nil {
t.Fatal("Failed to get sites:", err)
}
// Verify the first site was Loaded with proper attributes.
if !database.CompareSites(s1, sites[0]) {
t.Error("First saved site not equal to input:\n", sites[0], s1)
}
// Verify the second site was Loaded with proper attributes.
if !database.CompareSites(s2, sites[1]) {
t.Error("Second saved site not equal to input:\n", sites[1], s2)
}
}
// GetSitesMock is a mock of the SQL query to get the sites for pinging
func GetSitesMock(db *sql.DB) (database.Sites, error) {
var sites database.Sites
// Create the first site.
s1 := database.Site{Name: "Test", IsActive: true, URL: "http://www.google.com",
PingIntervalSeconds: 1, TimeoutSeconds: 1, IsSiteUp: true}
// Create the second site.
s2 := database.Site{Name: "Test 2", IsActive: true, URL: "http://www.github.com",
PingIntervalSeconds: 2, TimeoutSeconds: 2, IsSiteUp: true}
// Create the third site as not active.
s3 := database.Site{Name: "Test 3", IsActive: false, URL: "http://www.test.com",
PingIntervalSeconds: 2, TimeoutSeconds: 2}
// Contacts are deliberately set as false for SmsActive and EmailActive so as not to trigger Notifier
c1 := database.Contact{Name: "Joe Contact", EmailAddress: "*****@*****.**", SmsNumber: "5125551212",
SmsActive: false, EmailActive: false}
c2 := database.Contact{Name: "Jack Contact", EmailAddress: "*****@*****.**", SmsNumber: "5125551213",
SmsActive: false, EmailActive: false}
// Add the contacts to the sites
s1.Contacts = append(s1.Contacts, c1, c2)
s2.Contacts = append(s2.Contacts, c1)
s3.Contacts = append(s3.Contacts, c1)
sites = append(sites, s1, s2, s3)
return sites, nil
}
// TestCreatePings tests creating the ping records for a given site.
func TestCreatePings(t *testing.T) {
var err error
db, err := database.InitializeTestDB("")
defer db.Close()
if err != nil {
t.Fatal("Failed to create database:", err)
}
// First create a site to associate with the pings.
s := database.Site{Name: "Test", IsActive: true, URL: "http://www.google.com", PingIntervalSeconds: 60, TimeoutSeconds: 30}
err = s.CreateSite(db)
if err != nil {
t.Fatal("Failed to create new site:", err)
}
// Create a ping result
p1 := database.Ping{SiteID: s.SiteID, TimeRequest: time.Date(2015, time.November, 10, 23, 22, 22, 00, time.UTC),
Duration: 280, HTTPStatusCode: 200, SiteDown: false}
err = p1.CreatePing(db)
if err != nil {
t.Fatal("Failed to create new ping:", err)
}
// Create a second ping result
p2 := database.Ping{SiteID: s.SiteID, TimeRequest: time.Date(2015, time.November, 10, 23, 22, 20, 00, time.UTC),
Duration: 290, HTTPStatusCode: 200, SiteDown: true}
err = p2.CreatePing(db)
if err != nil {
t.Fatal("Failed to create new ping:", err)
}
//Get the saved Ping
var saved database.Site
err = saved.GetSitePings(db, s.SiteID, time.Date(2015, time.November, 10, 23, 00, 00, 00, time.UTC),
time.Date(2015, time.November, 10, 23, 59, 00, 00, time.UTC))
if err != nil {
t.Fatal("Failed to retrieve saved pings:", err)
}
// Verify the first ping was Loaded with proper attibutes and sorted last.
if !reflect.DeepEqual(p1, saved.Pings[1]) {
t.Error("First saved ping not equal to input:\n", saved.Pings[1], p1)
}
// Verify the second ping was Loaded with proper attributes and sorted first.
if !reflect.DeepEqual(p2, saved.Pings[0]) {
t.Error("Second saved ping not equal to input:\n", saved.Pings[0], p2)
}
// Verify that the site reflects the last ping time.
err = saved.GetSite(db, s.SiteID)
if err != nil {
t.Fatal("Failed to retrieve site:", err)
}
if saved.LastPing != p2.TimeRequest {
t.Error("Last Ping on site does not match input:\n", saved.LastPing, p1.TimeRequest)
}
//Get the first ping for the site.
firstping, err := s.GetFirstPing(db)
if err != nil {
t.Fatal("Failed to retrieve first ping for the site:", err)
}
if firstping != p2.TimeRequest {
t.Error("First Ping on site does not match input:\n", firstping, p2.TimeRequest)
}
// Create a third ping with conflicting times should error.
p3 := database.Ping{SiteID: s.SiteID, TimeRequest: time.Date(2015, time.November, 10, 23, 22, 20, 00, time.UTC),
Duration: 300, HTTPStatusCode: 200, SiteDown: false}
err = p3.CreatePing(db)
if err == nil {
t.Fatal("Conflicting pings should throw error.")
}
}
func (controller *sitesController) editPost(rw http.ResponseWriter, req *http.Request) (int, error) {
err := req.ParseForm()
if err != nil {
return http.StatusInternalServerError, err
}
decoder := schema.NewDecoder()
// Ignore unknown keys to prevent errors from the CSRF token.
decoder.IgnoreUnknownKeys(true)
formSite := new(viewmodels.SitesEditViewModel)
err = decoder.Decode(formSite, req.PostForm)
if err != nil {
return http.StatusInternalServerError, err
}
valErrors := validateSiteForm(formSite)
if len(valErrors) > 0 {
isAuthenticated, user := getCurrentUser(rw, req, controller.authorizer)
var contacts database.Contacts
err = contacts.GetContacts(controller.DB)
if err != nil {
return http.StatusInternalServerError, err
}
vm := viewmodels.EditSiteViewModel(formSite, contacts, isAuthenticated, user, valErrors)
vm.CsrfField = csrf.TemplateField(req)
return http.StatusOK, controller.editTemplate.Execute(rw, vm)
}
// Get the site to edit
site := new(database.Site)
err = site.GetSite(controller.DB, formSite.SiteID)
if err != nil {
return http.StatusInternalServerError, err
}
err = viewmodels.MapSiteVMtoDB(formSite, site)
if err != nil {
return http.StatusInternalServerError, err
}
err = site.UpdateSite(controller.DB)
if err != nil {
return http.StatusInternalServerError, err
}
//Loop selected ones first and if it's not already in the site then add it.
for _, contactSelID := range formSite.SelectedContacts {
if !int64InSlice(int64(contactSelID), formSite.SiteContacts) {
err = site.AddContactToSite(controller.DB, contactSelID)
if err != nil {
return http.StatusInternalServerError, err
}
}
}
// Loop existing site contacts and if it's not in the selected items then remove it.
for _, contactSiteID := range formSite.SiteContacts {
if !int64InSlice(int64(contactSiteID), formSite.SelectedContacts) {
err = site.RemoveContactFromSite(controller.DB, contactSiteID)
if err != nil {
return http.StatusInternalServerError, err
}
}
}
// Refresh the pinger with the changes.
err = controller.pinger.UpdateSiteSettings()
if err != nil {
return http.StatusInternalServerError, err
}
http.Redirect(rw, req, "/settings", http.StatusSeeOther)
return http.StatusSeeOther, nil
}
// 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()
}
}
}
// TestCreateAndGetMultipleSites tests creating more than one active sites
// with contacts in the database and then retrieving them.
func TestCreateAndGetMultipleSites(t *testing.T) {
var err error
db, err := database.InitializeTestDB("")
if err != nil {
t.Fatal("Failed to create database:", err)
}
defer db.Close()
// Create the first site.
s1 := database.Site{Name: "Test", IsActive: true, URL: "http://www.google.com",
PingIntervalSeconds: 60, TimeoutSeconds: 30, ContentExpected: "Expected 1",
ContentUnexpected: "Unexpected 1"}
err = s1.CreateSite(db)
if err != nil {
t.Fatal("Failed to create first site:", err)
}
// Create the second site.
s2 := database.Site{Name: "Test 2", IsActive: true, URL: "http://www.test.com",
PingIntervalSeconds: 60, TimeoutSeconds: 30,
LastStatusChange: time.Date(2009, time.November, 10, 23, 0, 0, 0, time.UTC),
LastPing: time.Date(2009, time.November, 10, 23, 0, 0, 0, time.UTC),
ContentExpected: "Expected 2", ContentUnexpected: "Unexpected 2"}
err = s2.CreateSite(db)
if err != nil {
t.Fatal("Failed to create second site:", err)
}
// Create a third site that is marked inactive.
s3 := database.Site{Name: "Test 3", IsActive: false, URL: "http://www.test3.com",
PingIntervalSeconds: 60, TimeoutSeconds: 30, ContentExpected: "Expected 3",
ContentUnexpected: "Unexpected 3"}
err = s3.CreateSite(db)
if err != nil {
t.Fatal("Failed to create third site:", err)
}
// Create first contact
c1 := database.Contact{Name: "Joe Contact", EmailAddress: "*****@*****.**", SmsNumber: "5125551212",
SmsActive: false, EmailActive: false}
err = c1.CreateContact(db)
if err != nil {
t.Fatal("Failed to create new contact:", err)
}
// Associate to the first and second site ID
err = s1.AddContactToSite(db, c1.ContactID)
if err != nil {
t.Fatal("Failed to associate contact 1 with first site:", err)
}
err = s2.AddContactToSite(db, c1.ContactID)
if err != nil {
t.Fatal("Failed to associate contact 1 with second site:", err)
}
// Create second contact
c2 := database.Contact{Name: "Jack Contact", EmailAddress: "*****@*****.**", SmsNumber: "5125551213",
SmsActive: false, EmailActive: false}
err = c2.CreateContact(db)
if err != nil {
t.Fatal("Failed to create new contact:", err)
}
// Associate only to the first site
err = s1.AddContactToSite(db, c2.ContactID)
if err != nil {
t.Fatal("Failed to associate contact 1 with first site:", err)
}
var sites database.Sites
// Get active sites with contacts
err = sites.GetSites(db, true, true)
if err != nil {
t.Fatal("Failed to get all the sites.", err)
}
// Verify that there are only two active sites.
if len(sites) != 2 {
t.Fatal("There should only be two active sites loaded.")
}
// Verify the first site was Loaded with proper attributes.
if !database.CompareSites(s1, sites[0]) {
t.Fatal("First saved site not equal to input:\n", sites[0], s1)
}
// Verify the second site was Loaded with proper attributes.
if !database.CompareSites(s2, sites[1]) {
t.Fatal("Second saved site not equal to input:\n", sites[1], s2)
}
// Verify the first contact was Loaded with proper attributes and sorted last.
if !reflect.DeepEqual(c1, sites[0].Contacts[1]) {
t.Error("Second saved contact not equal to input:\n", sites[0].Contacts[1], c1)
}
// Verify the second contact was loaded with the proper attributes and sorted first.
if !reflect.DeepEqual(c2, sites[0].Contacts[0]) {
t.Error("First saved contact not equal to input:\n", sites[0].Contacts[0], c2)
}
// Verify the first contact was loaded to the second site.
if !reflect.DeepEqual(c1, sites[1].Contacts[0]) {
t.Error("Second saved contact not equal to input:\n", sites[1].Contacts[0], c1)
}
// Verify that the first contact can get both related sites
err = c1.GetContactSites(db)
if err != nil {
t.Error("Error getting the sites for the first contact.")
}
if s1.URL != c1.Sites[0].URL {
t.Error("First contact's first site not as expected:\n", c1.Sites[0].URL, s1.URL)
}
if s2.URL != c1.Sites[1].URL {
t.Error("First contact's second site not as expected:\n", c1.Sites[1].URL, s2.URL)
}
if len(c1.Sites) != 2 {
t.Error("First contact should have two associated site.")
}
// Verify that the second contact can get the only related sites
err = c2.GetContactSites(db)
if err != nil {
t.Error("Error getting the site for the second contact.")
}
if s1.URL != c2.Sites[0].URL {
t.Error("Second contact's first site not as expected:\n", c2.Sites[0].URL, s1.URL)
}
if len(c2.Sites) != 1 {
t.Error("Second contact should only have one associated site.")
}
// Test for just the active sites without the contacts
var sitesNoContacts database.Sites
err = sitesNoContacts.GetSites(db, true, false)
if err != nil {
t.Fatal("Failed to get all the sites.", err)
}
// Verify the first site was Loaded with proper attributes and no contacts.
if !database.CompareSites(s1, sitesNoContacts[0]) {
t.Error("First saved site not equal to GetActiveSites results:\n", sitesNoContacts[0], s1)
}
// Verify the second site was Loaded with proper attributes and no contacts.
if !database.CompareSites(s2, sitesNoContacts[1]) {
t.Error("Second saved site not equal to GetActiveSites results:\n", sitesNoContacts[1], s2)
}
// Test for all of the sites without the contacts
var allSitesNoContacts database.Sites
err = allSitesNoContacts.GetSites(db, false, false)
if err != nil {
t.Fatal("Failed to get all of the sites.", err)
}
// Verify that there are 3 total sites.
if len(allSitesNoContacts) != 3 {
t.Error("There should be three total sites loaded.")
}
}
// TestCreateSiteAndContacts tests creating a site and adding new contacts
// in the database and then retrieving it.
func TestCreateSiteAndContacts(t *testing.T) {
db, err := database.InitializeTestDB("")
if err != nil {
t.Fatal("Failed to create database:", err)
}
defer db.Close()
// First create a site to associate with the contacts.
// Note: SiteID is ignored for create but is used in the test comparison
s := database.Site{SiteID: 1, Name: "Test", IsActive: true, URL: "http://www.google.com",
PingIntervalSeconds: 60, TimeoutSeconds: 30, IsSiteUp: true, ContentExpected: "Expected Content",
ContentUnexpected: "Unexpected Content"}
err = s.CreateSite(db)
if err != nil {
t.Fatal("Failed to create new site:", err)
}
// siteID should be 1 on the first create.
if s.SiteID != 1 {
t.Fatal("Expected 1, got ", s.SiteID)
}
//Get the saved site
var site database.Site
err = site.GetSite(db, s.SiteID)
if err != nil {
t.Fatal("Failed to retrieve new site:", err)
}
//Verify the saved site is same as the input.
if !database.CompareSites(site, s) {
t.Error("New site saved not equal to input:\n", site, s)
}
//Update the saved site
sUpdate := database.Site{SiteID: 1, Name: "Test Update", IsActive: false,
URL: "http://www.example.com", PingIntervalSeconds: 30, TimeoutSeconds: 15,
ContentExpected: "Updated Content", ContentUnexpected: "Updated Unexpected",
IsSiteUp: true,
}
site.Name = sUpdate.Name
site.URL = sUpdate.URL
site.IsActive = sUpdate.IsActive
site.PingIntervalSeconds = sUpdate.PingIntervalSeconds
site.TimeoutSeconds = sUpdate.TimeoutSeconds
site.ContentExpected = sUpdate.ContentExpected
site.ContentUnexpected = sUpdate.ContentUnexpected
site.IsSiteUp = sUpdate.IsSiteUp
err = site.UpdateSite(db)
if err != nil {
t.Fatal("Failed to update site:", err)
}
//Get the updated site
var siteUpdated database.Site
err = siteUpdated.GetSite(db, s.SiteID)
if err != nil {
t.Fatal("Failed to retrieve updated site:", err)
}
//Verify the saved site is same as the input.
if !database.CompareSites(siteUpdated, sUpdate) {
t.Error("Updated site saved not equal to input:\n", siteUpdated, sUpdate)
}
// Create first contact - ContactID is for referencing the contact get test
c := database.Contact{Name: "Joe Contact", EmailAddress: "*****@*****.**", SmsNumber: "5125551212",
SmsActive: false, EmailActive: false, ContactID: 1}
err = c.CreateContact(db)
if err != nil {
t.Fatal("Failed to create new contact:", err)
}
// Associate to the site ID
err = site.AddContactToSite(db, c.ContactID)
if err != nil {
t.Fatal("Failed to associate contact with site:", err)
}
// Create second contact
c2 := database.Contact{Name: "Jill Contact", EmailAddress: "*****@*****.**", SmsNumber: "5125551213",
SmsActive: false, EmailActive: false}
err = c2.CreateContact(db)
if err != nil {
t.Fatal("Failed to create new site:", err)
}
// Associate the contact to the site
err = site.AddContactToSite(db, c2.ContactID)
if err != nil {
t.Error("Failed to associate contact2 with site:", err)
}
//Get the saved site contacts
err = site.GetSiteContacts(db, site.SiteID)
if err != nil {
t.Error("Failed to retrieve site contacts:", err)
}
// Verify the first contact was Loaded as the last in list by sort order
if !reflect.DeepEqual(c, site.Contacts[1]) {
t.Error("New contact saved not equal to input:\n", site.Contacts[1], c)
}
// Verify the second contact was Loaded as the first in list by sort order
if !reflect.DeepEqual(c2, site.Contacts[0]) {
t.Error("New contact saved not equal to input:\n", site.Contacts[0], c2)
}
// Remove second contact from site.
err = site.RemoveContactFromSite(db, c2.ContactID)
if err != nil {
t.Fatal("Failed to remove contact2 from site:", err)
}
//Get the saved site contacts again
err = site.GetSiteContacts(db, site.SiteID)
if err != nil {
t.Fatal("Failed to retrieve site contacts:", err)
}
if len(site.Contacts) != 1 {
t.Fatal("Site should have only one contact after removal")
}
// Get the first contact via the GetContact method
c1Get := database.Contact{}
err = c1Get.GetContact(db, c.ContactID)
if err != nil {
t.Error("Failed to retrieve the first contact.")
}
// Verify the first contact was retrieved OK
if !reflect.DeepEqual(c, c1Get) {
t.Error("Retrieved contact saved not equal to input:\n", c1Get, c)
}
// Update the first contact.
c1Update := database.Contact{Name: "Jane Contact", EmailAddress: "*****@*****.**", SmsNumber: "5125551313",
SmsActive: true, EmailActive: true, ContactID: 1}
c1Get.Name = c1Update.Name
c1Get.EmailAddress = c1Update.EmailAddress
c1Get.SmsNumber = c1Update.SmsNumber
c1Get.EmailActive = c1Update.EmailActive
c1Get.SmsActive = c1Update.SmsActive
err = c1Get.UpdateContact(db)
if err != nil {
t.Error("Failed to update the first contact.")
}
// Get the first contact again after update
c1Get2 := database.Contact{}
err = c1Get2.GetContact(db, c1Update.ContactID)
if err != nil {
t.Error("Failed to retrieve the first contact.")
}
// Verify the first contact was retrieved OK
if !reflect.DeepEqual(c1Update, c1Get2) {
t.Error("Retrieved updated contact saved not equal to input:\n", c1Get2, c1Update)
}
}
// TestUpdateSiteStatus tests updating the up/down status of the site.
func TestUpdateSiteStatus(t *testing.T) {
db, err := database.InitializeTestDB("")
if err != nil {
t.Fatal("Failed to create database:", err)
}
defer db.Close()
// First create a site to update status.
s := database.Site{SiteID: 1, Name: "Test", IsActive: true, URL: "http://www.google.com", PingIntervalSeconds: 60, TimeoutSeconds: 30}
err = s.CreateSite(db)
if err != nil {
t.Fatal("Failed to create new site:", err)
}
// Update the status of the site to down
err = s.UpdateSiteStatus(db, false)
if err != nil {
t.Fatal("Failed to update site status:", err)
}
//Get the saved site
var updatedSite database.Site
err = updatedSite.GetSite(db, s.SiteID)
if err != nil {
t.Fatal("Failed to retrieve updated site:", err)
}
if updatedSite.IsSiteUp != false {
t.Errorf("Site status should be down.")
}
// Update the status of the site to up
err = s.UpdateSiteStatus(db, true)
if err != nil {
t.Fatal("Failed to update site status:", err)
}
// Update the first ping time of the site.
firstPingTime := time.Date(2015, time.November, 10, 23, 22, 22, 00, time.UTC)
err = s.UpdateSiteFirstPing(db, firstPingTime)
if err != nil {
t.Fatal("Failed to update first ping time:", err)
}
err = updatedSite.GetSite(db, s.SiteID)
if err != nil {
t.Fatal("Failed to retrieve updated site:", err)
}
if updatedSite.IsSiteUp != true {
t.Errorf("Site status should be up.")
}
if updatedSite.FirstPing != firstPingTime {
t.Errorf("Site first ping time %s does not match input %s.", updatedSite.FirstPing, firstPingTime)
}
}
// TestDeleteContacts tests creating a site and two contacts
// in the database and then deleting one of the contacts.
func TestDeleteContact(t *testing.T) {
db, err := database.InitializeTestDB("")
if err != nil {
t.Fatal("Failed to create database:", err)
}
defer db.Close()
// First create a site to associate with the contacts.
// Note: SiteID is ignored for create but is used in the test comparison
s := database.Site{SiteID: 1, Name: "Test", IsActive: true, URL: "http://www.google.com", PingIntervalSeconds: 60, TimeoutSeconds: 30, IsSiteUp: true}
err = s.CreateSite(db)
if err != nil {
t.Fatal("Failed to create new site:", err)
}
// Create first contact - ContactID is for referencing the contact get test
c := database.Contact{Name: "Joe Contact", EmailAddress: "*****@*****.**", SmsNumber: "5125551212",
SmsActive: false, EmailActive: false, ContactID: 1}
err = c.CreateContact(db)
if err != nil {
t.Fatal("Failed to create new contact:", err)
}
// Associate to the site ID
err = s.AddContactToSite(db, c.ContactID)
if err != nil {
t.Fatal("Failed to associate contact with site:", err)
}
// Create second contact
c2 := database.Contact{Name: "Jill Contact", EmailAddress: "*****@*****.**", SmsNumber: "5125551213",
SmsActive: false, EmailActive: false}
err = c2.CreateContact(db)
if err != nil {
t.Fatal("Failed to create new site:", err)
}
// Associate the contact to the site
err = s.AddContactToSite(db, c2.ContactID)
if err != nil {
t.Error("Failed to associate contact2 with site:", err)
}
err = s.GetSiteContacts(db, s.SiteID)
if err != nil {
t.Error("Failed to retrieve site contacts:", err)
}
if len(s.Contacts) != 2 {
t.Error("There should two contacts before deletion.")
}
// Delete the second contact
err = c2.DeleteContact(db)
if err != nil {
t.Fatal("Failed to delete contact 2:", err)
}
// Verify that it was deleted OK and not associated with the site, and
// that contact1 is still there.
err = s.GetSiteContacts(db, s.SiteID)
if err != nil {
t.Error("Failed to retrieve site contacts:", err)
}
if len(s.Contacts) != 1 {
t.Error("There should only be one contact for the site after deletion.")
}
if !reflect.DeepEqual(c, s.Contacts[0]) {
t.Error("Remaining contact not equal to input:\n", s.Contacts[0], c)
}
// Also verify that the contacts are correct.
var contacts database.Contacts
err = contacts.GetContacts(db)
if err != nil {
t.Fatal("Failed to get all contacts.", err)
}
if len(contacts) != 1 {
t.Error("There should only be one contact in the DB after deletion.")
}
if contacts[0].SiteCount != 1 {
t.Error("Site count should be 1.")
}
}