mirror of
https://github.com/go-vikunja/vikunja.git
synced 2026-04-24 22:25:15 +00:00
108 lines
3.3 KiB
Go
108 lines
3.3 KiB
Go
// Vikunja is a to-do list application to facilitate your life.
|
|
// Copyright 2018-present Vikunja and contributors. All rights reserved.
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
package migration
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"strings"
|
|
"sync/atomic"
|
|
"testing"
|
|
)
|
|
|
|
func TestDoPostWithHeaders_RetriesOn500(t *testing.T) {
|
|
var attempts atomic.Int32
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
|
|
count := attempts.Add(1)
|
|
if count < 3 {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
return
|
|
}
|
|
w.WriteHeader(http.StatusOK)
|
|
}))
|
|
defer server.Close()
|
|
|
|
form := url.Values{"key": {"value"}}
|
|
resp, err := DoPostWithHeaders(server.URL, form, map[string]string{})
|
|
|
|
if err != nil {
|
|
t.Fatalf("expected no error, got %v", err)
|
|
}
|
|
defer resp.Body.Close()
|
|
if resp.StatusCode != http.StatusOK {
|
|
t.Errorf("expected status 200, got %d", resp.StatusCode)
|
|
}
|
|
if attempts.Load() != 3 {
|
|
t.Errorf("expected 3 attempts, got %d", attempts.Load())
|
|
}
|
|
}
|
|
|
|
func TestDoPostWithHeaders_GivesUpAfter3Retries(t *testing.T) {
|
|
var attempts atomic.Int32
|
|
expectedBody := "Internal Server Error: database connection failed"
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
|
|
attempts.Add(1)
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
_, _ = w.Write([]byte(expectedBody))
|
|
}))
|
|
defer server.Close()
|
|
|
|
form := url.Values{"key": {"value"}}
|
|
resp, err := DoPostWithHeaders(server.URL, form, map[string]string{})
|
|
|
|
if err == nil {
|
|
t.Fatal("expected error after exhausted retries, got nil")
|
|
}
|
|
if !strings.Contains(err.Error(), expectedBody) {
|
|
t.Errorf("expected error message to contain response body %q, got: %s", expectedBody, err.Error())
|
|
}
|
|
if resp == nil {
|
|
t.Fatal("expected response to be returned with error, got nil")
|
|
}
|
|
defer resp.Body.Close()
|
|
if resp.StatusCode != http.StatusInternalServerError {
|
|
t.Errorf("expected status 500, got %d", resp.StatusCode)
|
|
}
|
|
if attempts.Load() != 3 {
|
|
t.Errorf("expected 3 attempts, got %d", attempts.Load())
|
|
}
|
|
}
|
|
|
|
func TestDoPostWithHeaders_DoesNotRetryOn4xx(t *testing.T) {
|
|
var attempts atomic.Int32
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
|
|
attempts.Add(1)
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
}))
|
|
defer server.Close()
|
|
|
|
form := url.Values{"key": {"value"}}
|
|
resp, err := DoPostWithHeaders(server.URL, form, map[string]string{})
|
|
|
|
if err != nil {
|
|
t.Fatalf("expected no error, got %v", err)
|
|
}
|
|
defer resp.Body.Close()
|
|
if resp.StatusCode != http.StatusBadRequest {
|
|
t.Errorf("expected status 400, got %d", resp.StatusCode)
|
|
}
|
|
if attempts.Load() != 1 {
|
|
t.Errorf("expected 1 attempt (no retries on 4xx), got %d", attempts.Load())
|
|
}
|
|
}
|