Merge pull request mackerelio#687 from wafuwafu13/certfile-test

check-cert-file: add test

Author: yseto

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,32 @@
+package checkcertfile
+
+import (
+	"os/exec"
+	"path/filepath"
+	"runtime"
+	"testing"
+
+	"github.com/mackerelio/checkers"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestCertFile(t *testing.T) {
+	tmpDir := t.TempDir()
+	cmd := exec.Command(
+		"go",
+		"run",
+		filepath.FromSlash(filepath.Join(runtime.GOROOT(), "/src/crypto/tls/generate_cert.go")),
+		"-host",
+		"localhost",
+		"-duration",
+		"720h0m0s", // 30*24*time.Hour
+	)
+	cmd.Dir = tmpDir
+	err := cmd.Run()
+	if err != nil {
+		t.Fatalf("Failed to generate cert.pem: %s", err)
+	}
+
+	ckr := Run([]string{"-f", filepath.Join(tmpDir, "cert.pem"), "-w", "35", "-c", "25"})
+	assert.Equal(t, checkers.WARNING, ckr.Status, "should be WARNING")
+}