[client] fix tests

Author: capcom6

smsgateway/client_test.go

@@ -20,11 +20,12 @@ const (
 )
 
 type mockServerExpectedInput struct {
-	method      string
-	path        string
-	query       string
-	contentType string
-	body        string
+	method        string
+	path          string
+	query         string
+	authorization string
+	contentType   string
+	body          string
 }
 
 type mockServerOutput struct {
@@ -33,6 +34,10 @@ type mockServerOutput struct {
 }
 
 func newMockServer(input mockServerExpectedInput, output mockServerOutput) *httptest.Server {
+	if input.authorization == "" {
+		input.authorization = authorizationHeader
+	}
+
 	return httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		if r.Method != input.method {
 			w.WriteHeader(http.StatusMethodNotAllowed)
@@ -49,7 +54,7 @@ func newMockServer(input mockServerExpectedInput, output mockServerOutput) *http
 			return
 		}
 
-		if r.Header.Get("Authorization") != authorizationHeader {
+		if r.Header.Get("Authorization") != input.authorization {
 			w.WriteHeader(http.StatusUnauthorized)
 			return
 		}
@@ -80,6 +85,43 @@ func newClient(baseURL string) *smsgateway.Client {
 	})
 }
 
+func newJWTClient(baseURL string) *smsgateway.Client {
+	return smsgateway.NewClient(smsgateway.Config{
+		BaseURL: baseURL,
+		Token:   password,
+	})
+}
+
+func TestJWTClient_Send(t *testing.T) {
+	server := newMockServer(mockServerExpectedInput{
+		method:        http.MethodPost,
+		path:          "/messages",
+		authorization: "Bearer " + password,
+		contentType:   "application/json",
+		body:          `{"textMessage":{"text":"Hello World!"},"phoneNumbers":["+1234567890"]}`,
+	}, mockServerOutput{
+		code: http.StatusCreated,
+		body: `{}`,
+	})
+	defer server.Close()
+
+	client := newJWTClient(server.URL)
+
+	t.Run("Success", func(t *testing.T) {
+		message := smsgateway.Message{
+			TextMessage: &smsgateway.TextMessage{
+				Text: "Hello World!",
+			},
+			PhoneNumbers: []string{"+1234567890"},
+		}
+
+		_, err := client.Send(context.Background(), message)
+		if err != nil {
+			t.Errorf("Send() error = %v", err)
+		}
+	})
+}
+
 func TestClient_Send(t *testing.T) {
 	type args struct {
 		ctx     context.Context
@@ -499,6 +541,14 @@ func TestClient_DeleteDevice(t *testing.T) {
 			},
 			wantErr: false,
 		},
+		{
+			name: "Not Found",
+			c:    client,
+			args: args{
+				deviceID: "456",
+			},
+			wantErr: true,
+		},
 	}
 
 	for _, tt := range tests {
@@ -554,28 +604,6 @@ func TestClient_CheckHealth(t *testing.T) {
 			}
 		})
 	}
-
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			server := newMockServer(mockServerExpectedInput{
-				method: http.MethodGet,
-				path:   "/health",
-			}, mockServerOutput{
-				code: tt.code,
-				body: tt.body,
-			})
-			defer server.Close()
-
-			client := newClient(server.URL)
-			resp, err := client.CheckHealth(context.Background())
-			if (err != nil) != tt.wantErr {
-				t.Errorf("CheckHealth error = %v, wantErr %v", err, tt.wantErr)
-			}
-			if !tt.wantErr && !reflect.DeepEqual(resp, tt.want) {
-				t.Errorf("CheckHealth response = %v, want %v", resp, tt.want)
-			}
-		})
-	}
 }
 
 func TestClient_ExportInbox(t *testing.T) {
@@ -876,12 +904,12 @@ func TestClient_GenerateToken(t *testing.T) {
 			name: "Success",
 			req:  smsgateway.TokenRequest{Scopes: []string{"messages:read"}, TTL: 3600},
 			code: http.StatusOK,
-			body: `{"id":"UWSpnfJzsdQ81JSU2Q","token_type":"Bearer","access_token":"vLNTuO17eG7kp0XS0","expires_at":"2021-01-01T00:00:00Z"}`,
+			body: `{"id":"token_id_example","token_type":"Bearer","access_token":"access_token_example","expires_at":"2025-01-01T00:00:00Z"}`,
 			want: smsgateway.TokenResponse{
-				ID:          "UWSpnfJzsdQ81JSU2Q",
+				ID:          "token_id_example",
 				TokenType:   "Bearer",
-				AccessToken: "vLNTuO17eG7kp0XS0",
-				ExpiresAt:   time.Date(2021, 1, 1, 0, 0, 0, 0, time.UTC),
+				AccessToken: "access_token_example",
+				ExpiresAt:   time.Date(2025, 1, 1, 0, 0, 0, 0, time.UTC),
 			},
 			wantErr: false,
 		},