Add missing tests (#132)

Author: xperimental

internal/client/client.go

@@ -11,7 +11,8 @@ import (
 )
 
 const (
-	nextcloudTokenHeader = "NC-Token"
+	nextcloudTokenHeader  = "NC-Token"
+	maintenanceModeHeader = "X-Nextcloud-Maintenance-Mode"
 )
 
 var (
@@ -62,7 +63,7 @@ func New(infoURL, username, password, authToken string, timeout time.Duration, u
 		case http.StatusTooManyRequests:
 			return nil, ErrRatelimit
 		case http.StatusServiceUnavailable:
-			if res.Header.Get("X-Nextcloud-Maintenance-Mode") != "" {
+			if res.Header.Get(maintenanceModeHeader) != "" {
 				return nil, ErrMaintenanceMode
 			}
 

internal/client/client_test.go

@@ -9,6 +9,7 @@ import (
 	"time"
 
 	"github.com/google/go-cmp/cmp"
+
 	"github.com/xperimental/nextcloud-exporter/internal/testutil"
 	"github.com/xperimental/nextcloud-exporter/serverinfo"
 )
@@ -134,6 +135,25 @@ func TestClient(t *testing.T) {
 			},
 			wantErr: ErrRatelimit,
 		},
+		{
+			desc: "unavailable",
+			handler: func(t *testing.T) http.Handler {
+				return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
+					w.WriteHeader(http.StatusServiceUnavailable)
+				})
+			},
+			wantErr: ErrUnavailable,
+		},
+		{
+			desc: "maintenance",
+			handler: func(t *testing.T) http.Handler {
+				return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
+					w.Header().Set(maintenanceModeHeader, "1")
+					w.WriteHeader(http.StatusServiceUnavailable)
+				})
+			},
+			wantErr: ErrMaintenanceMode,
+		},
 	}
 
 	for _, tc := range tt {