check-cert-file: add test

Author: wafuwafu13

check-cert-file/lib/check-cert-file.go

@@ -19,15 +19,19 @@ type certOpts struct {
 
 // Do the plugin
 func Do() {
-	ckr := checkCertExpiration()
+	ckr := Run(os.Args[1:])
 	ckr.Name = "CERT Expiry"
 	ckr.Exit()
 }
 
-func checkCertExpiration() *checkers.Checker {
-	opts := certOpts{}
-	psr := flags.NewParser(&opts, flags.Default)
-	_, err := psr.Parse()
+func parseArgs(args []string) (*certOpts, error) {
+	opts := &certOpts{}
+	_, err := flags.ParseArgs(opts, args)
+	return opts, err
+}
+
+func Run(args []string) *checkers.Checker {
+	opts, err := parseArgs(args)
 	if err != nil {
 		os.Exit(1)
 	}

check-cert-file/lib/check-cert-file_test.go

@@ -0,0 +1,36 @@
+package checkcertfile
+
+import (
+	"os"
+	"os/exec"
+	"strings"
+	"testing"
+
+	"github.com/mackerelio/checkers"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestCertFile(t *testing.T) {
+	goroot, err := exec.Command("go", "env", "GOROOT").Output()
+	if err != nil {
+		t.Fatalf("Faild to get GOROOT: %s", err)
+	}
+	err = exec.Command(
+		"go",
+		"run",
+		strings.TrimSuffix(string(goroot), "\n")+"/src/crypto/tls/generate_cert.go",
+		"-host",
+		"localhost",
+		"-duration",
+		"720h0m0s", // 30*24*time.Hour
+	).Run()
+	if err != nil {
+		t.Fatalf("Faild to generate cert.pem: %s", err)
+	}
+
+	ckr := Run([]string{"-f", "./cert.pem", "-w", "35", "-c", "25"})
+	assert.Equal(t, checkers.WARNING, ckr.Status, "should be WARNING")
+
+	os.Remove("./cert.pem")
+	os.Remove("./key.pem")
+}