// TestNewPinger tests building the pinger object.
func TestNewPinger(t *testing.T) {
db, _ := sql.Open("testdb", "")
pinger.CreatePingerLog("", true)
p := pinger.NewPinger(db, pinger.GetSitesMock, pinger.RequestURLMock,
notifier.SendEmailMock, notifier.SendSmsMock)
if len(p.Sites) != 3 {
t.Fatal("Incorrect number of sites returned in new pinger.")
}
results, err := pinger.GetLogContent()
if err != nil {
t.Fatal("Failed to get log results.", err)
}
if !strings.Contains(results, "SITE: Test, http://www.google.com") {
t.Fatal("Failed to load first site.")
}
if !strings.Contains(results, "SITE: Test 2, http://www.github.com") {
t.Fatal("Failed to load second site.")
}
if !strings.Contains(results, "SITE: Test 3, http://www.test.com") {
t.Fatal("Failed to load third site.")
}
}
// TestStartAndRestartPinger starts up the pinger and then stops it after 3 seconds
func TestStartAndRestartPinger(t *testing.T) {
// Fake db for testing.
db, _ := sql.Open("testdb", "")
pinger.CreatePingerLog("", true)
pinger.ResetHitCount()
p := pinger.NewPinger(db, pinger.GetSitesMock, pinger.RequestURLMock,
notifier.SendEmailMock, notifier.SendSmsMock)
p.Start()
// Sleep to allow running the tests before stopping.
time.Sleep(5 * time.Second)
p.Stop() // Test Restart after stop
p.Start()
p.Stop()
results, err := pinger.GetLogContent()
if err != nil {
t.Fatal("Failed to get log results.", err)
}
if !strings.Contains(results, "Client.Timeout") {
t.Errorf("Failed to report timeout error.")
}
if !strings.Contains(results, "Test 3 Paused") {
t.Errorf("Failed to report paused site.")
}
if !strings.Contains(results, "Error - HTTP Status Code") {
t.Fatal("Failed to report bad HTTP Status Code.")
}
if !strings.Contains(results, "Sending Notification of Site Contacts about Test 2: Site is Down...") {
t.Fatal("Failed to report site being down.")
}
if !strings.Contains(results, "Will notify status change for Test 2: Test 2 at http://www.github.com: Site is now up, response time was 300ms.") {
t.Fatal("Failed to report change in notification.")
}
}
// 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.")
}
}
// TestStartPingerErrorWithGetSites verifies that an error is handled when the get sites returns
// an error.
func TestStartPingerErrorWithGetSites(t *testing.T) {
db, _ := sql.Open("testdb", "")
pinger.CreatePingerLog("", true)
pinger.ResetHitCount()
p := pinger.NewPinger(db, pinger.GetSitesErrorMock, pinger.RequestURLMock,
notifier.SendEmailMock, notifier.SendSmsMock)
p.Start()
results, err := pinger.GetLogContent()
if err != nil {
t.Fatal("Failed to get log results.", err)
}
if !strings.Contains(results, "Timeout accessing the SQL database.") {
t.Fatal("Failed to report error getting the sites from the DB.")
}
}
// TestStartEmptySitesPinger verifies that proper reporting is done for the case of no active sites.
func TestStartEmptySitesPinger(t *testing.T) {
db, _ := sql.Open("testdb", "")
pinger.CreatePingerLog("", true)
pinger.ResetHitCount()
p := pinger.NewPinger(db, pinger.GetEmptySitesMock, pinger.RequestURLMock,
notifier.SendEmailMock, notifier.SendSmsMock)
p.Start()
results, err := pinger.GetLogContent()
if err != nil {
t.Fatal("Failed to get log results.", err)
}
if !strings.Contains(results, "No active sites set up for pinging in the database!") {
t.Fatal("Failed to report empty sites.")
}
}
func main() {
var err error
// Setup the main db.
var db *sql.DB
err = pinger.CreatePingerLog("", false)
if err != nil {
fatalError("Failed to initialize logging:", err)
}
startLog("Starting go-ping-sites version " + version + ", website on port " +
config.Settings.Website.HTTPPort + "...")
db, err = database.InitializeDB("go-ping-sites.db", "db-seed.toml")
if err != nil {
fatalError("Failed to initialize database:", err)
}
defer db.Close()
// Validate the required config for the website.
_, err = govalidator.ValidateStruct(config.Settings.Website)
if err != nil {
fatalError("Error in config.toml configuration:", err)
}
// Setup the auth
// create the authorization backend
authBackend, err = createAuthBackendFile()
if err != nil {
fatalError("Failed to create Auth Backend: ", err)
}
cookieKey := []byte(config.Settings.Website.CookieKey)
secureCookie := config.Settings.Website.SecureHTTPS
roles = getRoles()
authorizer, err = httpauth.NewAuthorizer(authBackend, cookieKey, "user", roles)
createDefaultUser()
// Start the Pinger
p := pinger.NewPinger(db, pinger.GetSites, pinger.RequestURL, notifier.SendEmail, notifier.SendSms)
p.Start()
// Start the web server.
templates := controllers.PopulateTemplates("templates")
controllers.Register(db, authorizer, authBackend, roles, templates, p, version, cookieKey, secureCookie)
err = http.ListenAndServe(":"+config.Settings.Website.HTTPPort, nil) // set listen port
if err != nil {
fatalError("ListenAndServe: ", err)
}
}
// TestInternetAccessError starts up the pinger and then stops it after 3 seconds
func TestInternetAccessError(t *testing.T) {
// Fake db for testing.
db, _ := sql.Open("testdb", "")
pinger.CreatePingerLog("", true)
p := pinger.NewPinger(db, pinger.GetSitesMock, pinger.RequestURLBadInternetAccessMock,
notifier.SendEmailMock, notifier.SendSmsMock)
p.Start()
time.Sleep(5 * time.Second)
p.Stop()
results, err := pinger.GetLogContent()
if err != nil {
t.Fatal("Failed to get log results.", err)
}
if !strings.Contains(results, "Unable to determine site status - connect: network is unreachable") {
t.Fatal("Failed to report Internet Access Error: ", results)
}
}
// TestUpdateSiteSettings starts up the pinger and then updates the site settings.
func TestUpdateSiteSettings(t *testing.T) {
// Fake db for testing.
db, _ := sql.Open("testdb", "")
pinger.CreatePingerLog("", true)
pinger.ResetHitCount()
p := pinger.NewPinger(db, pinger.GetSitesMock, pinger.RequestURLMock,
notifier.SendEmailMock, notifier.SendSmsMock)
p.Start()
// Test UpdateSiteSettings
p.UpdateSiteSettings()
p.Stop()
results, err := pinger.GetLogContent()
if err != nil {
t.Fatal("Failed to get log results.", err)
}
if !strings.Contains(results, "Updating the site settings due to change...") {
t.Fatal("Failed to launch update site settings.")
}
}
// TestContentCheck tests the content checks (Must Include and Must Not Include)
func TestContentCheck(t *testing.T) {
// Fake db for testing.
db, _ := sql.Open("testdb", "")
pinger.CreatePingerLog("", true)
p := pinger.NewPinger(db, pinger.GetSitesContentMock, pinger.RequestURLContentMock,
notifier.SendEmailMock, notifier.SendSmsMock)
p.Start()
// Sleep to allow running the tests before stopping.
time.Sleep(2 * time.Second)
p.Stop()
results, err := pinger.GetLogContent()
if err != nil {
t.Fatal("Failed to get log results.", err)
}
if !strings.Contains(results, "Error - body content content has excluded content: Bad response text") {
t.Fatal("Failed to report excluded content.")
}
if !strings.Contains(results, "Error - required body content missing: Good response text") {
t.Fatal("Failed to report missing required content.")
}
}
// TestUpdateSiteSettingsError tests when UpdateSiteSettings returns a DB error.
func TestUpdateSiteSettingsError(t *testing.T) {
// Fake db for testing.
db, _ := sql.Open("testdb", "")
pinger.CreatePingerLog("", true)
pinger.ResetHitCount()
p := pinger.NewPinger(db, pinger.GetSitesErrorMock, pinger.RequestURLMock,
notifier.SendEmailMock, notifier.SendSmsMock)
p.Start()
// Test UpdateSiteSettings with error due to database.
err := p.UpdateSiteSettings()
if err == nil {
t.Error("Failed to report error with getting site settings.")
}
p.Stop()
results, err := pinger.GetLogContent()
if err != nil {
t.Fatal("Failed to get log results.", err)
}
if !strings.Contains(results, "Timeout accessing the SQL database.") {
t.Fatal("Failed to report problem updating the sites.")
}
}