From cadb15f7afb5e7c88667eb4006209efca17744af Mon Sep 17 00:00:00 2001
From: Julien Dessaux
Date: Mon, 24 Feb 2020 23:05:45 +0100
Subject: Added tests to the main package and completely reworked the code
 around that

---
 main.go | 110 +++-------------------------------------------------------------
 1 file changed, 4 insertions(+), 106 deletions(-)

(limited to 'main.go')

diff --git a/main.go b/main.go
index 62347ab..f55943d 100644
--- a/main.go
+++ b/main.go
@@ -1,114 +1,12 @@
+//+build !test
+
 package main
 
 import (
-	"bareos-zabbix-check/config"
-	"bareos-zabbix-check/job"
-	"bareos-zabbix-check/spool"
-	"bareos-zabbix-check/state"
+	"bareos-zabbix-check/zabbix"
 	"fmt"
-	"log"
-	"os"
-	"path/filepath"
-	"time"
-)
-
-const (
-	spoolFileName = "bareos-zabbix-check.spool"
 )
 
 func main() {
-	var (
-		config        config.Config
-		errorString   string
-		missingString string
-	)
-	config.Init()
-	// Open the state file
-	stateFile, err := os.Open(config.StateFile())
-	if err != nil {
-		fmt.Printf("INFO Couldn't open state file : %s", err)
-		os.Exit(0)
-	}
-	defer stateFile.Close()
-	// parse the state file
-	header, err := state.ParseHeader(stateFile)
-	if err != nil {
-		fmt.Printf("INFO Could not parse state file header : %s", err)
-		os.Exit(0)
-	}
-	if config.Verbose() {
-		log.Printf("Parsed header: %+s\n", header)
-	}
-	// seek to the job entries in the state file
-	offset, err := stateFile.Seek(int64(header.LastJobsAddr), 0)
-	if err != nil {
-		fmt.Printf("INFO Couldn't seek to jobs position in state file : %s", err)
-	}
-	if uint64(offset) != header.LastJobsAddr {
-		fmt.Print("INFO Truncated state file")
-	}
-	// Then parse the jobs in the state file
-	jobs, err := state.ParseJobs(stateFile)
-	if err != nil {
-		fmt.Printf("INFO Could not parse jobs in state file : %s", err)
-	}
-	if config.Verbose() {
-		log.Printf("%d jobs found in state file\n", len(jobs))
-		for i := 0; i < len(jobs); i++ {
-			log.Print(jobs[i])
-		}
-	}
-
-	// We will check for errors in loading the spool file only at the end. If all jobs ran successfully without errors
-	// in the state file and we manage to write a new spool file without errors, then we will ignore any error here to
-	// avoid false positives during backup bootstrap
-	// Open the spool file
-	spoolFile, spoolErr := os.Open(filepath.Join(config.WorkDir(), spoolFileName))
-	var spoolJobs []job.Job
-	if err == nil {
-		defer spoolFile.Close()
-		spoolJobs, spoolErr = spool.Parse(spoolFile)
-	}
-
-	jobs = job.KeepOldestOnly(append(jobs, spoolJobs...))
-
-	// we write this new spool
-	spoolFile, err = os.Create(filepath.Join(config.WorkDir(), spoolFileName))
-	if err == nil {
-		defer spoolFile.Close()
-		err = spool.Serialize(spoolFile, jobs)
-	}
-	if err != nil {
-		fmt.Printf("AVERAGE: Error saving the spool file : %s\n", err)
-		os.Exit(0)
-	}
-
-	now := uint64(time.Now().Unix())
-	// We build the error strings
-	for _, job := range jobs {
-		if job.Success {
-			if job.Timestamp < now-24*3600 {
-				if missingString == "" {
-					missingString = fmt.Sprintf("missing: %s", job.Name)
-				} else {
-					missingString = fmt.Sprintf("%s, %s", missingString, job.Name)
-				}
-			}
-		} else {
-			if errorString == "" {
-				errorString = fmt.Sprintf("errors: %s", job.Name)
-			} else {
-				errorString = fmt.Sprintf("%s, %s", errorString, job.Name)
-			}
-		}
-	}
-	// Finally we output
-	if errorString != "" || missingString != "" {
-		fmt.Printf("AVERAGE: %s %s", errorString, missingString)
-		if spoolErr != nil {
-			fmt.Printf(" additionnal errors: %s", spoolErr)
-		}
-	} else {
-		fmt.Printf("OK")
-	}
+	fmt.Print(zabbix.Main())
 }
-- 
cgit v1.2.3