| Server IP : 88.223.91.31 / Your IP : 216.73.216.71 Web Server : LiteSpeed System : Linux id-dci-web1986.main-hosting.eu 5.14.0-611.26.1.el9_7.x86_64 #1 SMP PREEMPT_DYNAMIC Thu Jan 29 05:24:47 EST 2026 x86_64 User : u686484674 ( 686484674) PHP Version : 8.0.30 Disable Function : system, exec, shell_exec, passthru, mysql_list_dbs, ini_alter, dl, symlink, link, chgrp, leak, popen, apache_child_terminate, virtual, mb_send_mail MySQL : OFF | cURL : ON | WGET : ON | Perl : OFF | Python : OFF | Sudo : OFF | Pkexec : OFF Directory : /proc/./self/root/opt/go/pkg/mod/github.com/prometheus/alertmanager@v0.28.1/notify/ |
Upload File : |
// Copyright 2019 Prometheus Team
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package notify
import (
"bytes"
"fmt"
"io"
"net/http"
"path"
"reflect"
"runtime"
"testing"
"github.com/stretchr/testify/require"
)
func TestTruncate(t *testing.T) {
type expect struct {
out string
trunc bool
}
testCases := []struct {
in string
n int
runes expect
bytes expect
}{
{
in: "",
n: 5,
runes: expect{out: "", trunc: false},
bytes: expect{out: "", trunc: false},
},
{
in: "abcde",
n: 2,
runes: expect{out: "ab", trunc: true},
bytes: expect{out: "..", trunc: true},
},
{
in: "abcde",
n: 4,
runes: expect{out: "abc…", trunc: true},
bytes: expect{out: "a…", trunc: true},
},
{
in: "abcde",
n: 5,
runes: expect{out: "abcde", trunc: false},
bytes: expect{out: "abcde", trunc: false},
},
{
in: "abcdefgh",
n: 5,
runes: expect{out: "abcd…", trunc: true},
bytes: expect{out: "ab…", trunc: true},
},
{
in: "a⌘cde",
n: 5,
runes: expect{out: "a⌘cde", trunc: false},
bytes: expect{out: "a…", trunc: true},
},
{
in: "a⌘cdef",
n: 5,
runes: expect{out: "a⌘cd…", trunc: true},
bytes: expect{out: "a…", trunc: true},
},
{
in: "世界cdef",
n: 3,
runes: expect{out: "世界c", trunc: true},
bytes: expect{out: "…", trunc: true},
},
{
in: "❤️✅🚀🔥❌❤️✅🚀🔥❌❤️✅🚀🔥❌❤️✅🚀🔥❌",
n: 19,
runes: expect{out: "❤️✅🚀🔥❌❤️✅🚀🔥❌❤️✅🚀🔥❌…", trunc: true},
bytes: expect{out: "❤️✅🚀…", trunc: true},
},
}
type truncateFunc func(string, int) (string, bool)
for _, tc := range testCases {
for _, fn := range []truncateFunc{TruncateInBytes, TruncateInRunes} {
var truncated bool
var out string
fnPath := runtime.FuncForPC(reflect.ValueOf(fn).Pointer()).Name()
fnName := path.Base(fnPath)
switch fnName {
case "notify.TruncateInRunes":
truncated = tc.runes.trunc
out = tc.runes.out
case "notify.TruncateInBytes":
truncated = tc.bytes.trunc
out = tc.bytes.out
default:
t.Fatalf("unknown function")
}
t.Run(fmt.Sprintf("%s(%s,%d)", fnName, tc.in, tc.n), func(t *testing.T) {
s, trunc := fn(tc.in, tc.n)
require.Equal(t, out, s)
require.Equal(t, truncated, trunc)
})
}
}
}
type brokenReader struct{}
func (b brokenReader) Read([]byte) (int, error) {
return 0, fmt.Errorf("some error")
}
func TestRetrierCheck(t *testing.T) {
for _, tc := range []struct {
retrier Retrier
status int
body io.Reader
retry bool
expectedErr string
}{
{
retrier: Retrier{},
status: http.StatusOK,
body: bytes.NewBuffer([]byte("ok")),
retry: false,
},
{
retrier: Retrier{},
status: http.StatusNoContent,
retry: false,
},
{
retrier: Retrier{},
status: http.StatusBadRequest,
retry: false,
expectedErr: "unexpected status code 400",
},
{
retrier: Retrier{RetryCodes: []int{http.StatusTooManyRequests}},
status: http.StatusBadRequest,
body: bytes.NewBuffer([]byte("invalid request")),
retry: false,
expectedErr: "unexpected status code 400: invalid request",
},
{
retrier: Retrier{RetryCodes: []int{http.StatusTooManyRequests}},
status: http.StatusTooManyRequests,
retry: true,
expectedErr: "unexpected status code 429",
},
{
retrier: Retrier{},
status: http.StatusServiceUnavailable,
body: bytes.NewBuffer([]byte("retry later")),
retry: true,
expectedErr: "unexpected status code 503: retry later",
},
{
retrier: Retrier{},
status: http.StatusBadGateway,
body: &brokenReader{},
retry: true,
expectedErr: "unexpected status code 502",
},
{
retrier: Retrier{CustomDetailsFunc: func(status int, b io.Reader) string {
if status != http.StatusServiceUnavailable {
return "invalid"
}
bs, _ := io.ReadAll(b)
return fmt.Sprintf("server response is %q", string(bs))
}},
status: http.StatusServiceUnavailable,
body: bytes.NewBuffer([]byte("retry later")),
retry: true,
expectedErr: "unexpected status code 503: server response is \"retry later\"",
},
} {
t.Run("", func(t *testing.T) {
retry, err := tc.retrier.Check(tc.status, tc.body)
require.Equal(t, tc.retry, retry)
if tc.expectedErr == "" {
require.NoError(t, err)
return
}
require.EqualError(t, err, tc.expectedErr)
})
}
}