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
}
// 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)
}
}
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
}