74 lines
2 KiB
Go
74 lines
2 KiB
Go
// Copyright 2020 Frédéric Guillot. All rights reserved.
|
|
// Use of this source code is governed by the Apache 2.0
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package client // import "miniflux.app/http/client"
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"miniflux.app/config"
|
|
)
|
|
|
|
func TestClientWithDelay(t *testing.T) {
|
|
clt := New("http://httpbin.org/delay/5")
|
|
clt.ClientTimeout = 1
|
|
_, err := clt.Get()
|
|
if err == nil {
|
|
t.Fatal(`The client should stops after 1 second`)
|
|
}
|
|
}
|
|
|
|
func TestClientWithError(t *testing.T) {
|
|
clt := New("http://httpbin.org/status/502")
|
|
clt.ClientTimeout = 1
|
|
response, err := clt.Get()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if response.StatusCode != 502 {
|
|
t.Fatalf(`Unexpected response status code: %d`, response.StatusCode)
|
|
}
|
|
|
|
if !response.HasServerFailure() {
|
|
t.Fatal(`A 500 error is considered as server failure`)
|
|
}
|
|
}
|
|
|
|
func TestClientWithResponseTooLarge(t *testing.T) {
|
|
clt := New("http://httpbin.org/bytes/100")
|
|
clt.ClientMaxBodySize = 10
|
|
_, err := clt.Get()
|
|
if err == nil {
|
|
t.Fatal(`The client should fails when reading a response too large`)
|
|
}
|
|
}
|
|
|
|
func TestClientWithBasicAuth(t *testing.T) {
|
|
clt := New("http://httpbin.org/basic-auth/testuser/testpassword")
|
|
clt.WithCredentials("testuser", "testpassword")
|
|
_, err := clt.Get()
|
|
if err != nil {
|
|
t.Fatalf(`The client should be authenticated successfully: %v`, err)
|
|
}
|
|
}
|
|
|
|
func TestClientRequestUserAgent(t *testing.T) {
|
|
clt := New("http://httpbin.org")
|
|
if clt.requestUserAgent != DefaultUserAgent {
|
|
t.Errorf(`The client had default User-Agent %q, wanted %q`, clt.requestUserAgent, DefaultUserAgent)
|
|
}
|
|
|
|
userAgent := "Custom User Agent"
|
|
os.Setenv("HTTP_CLIENT_USER_AGENT", userAgent)
|
|
opts, err := config.NewParser().ParseEnvironmentVariables()
|
|
if err != nil {
|
|
t.Fatalf(`Parsing config failed: %v`, err)
|
|
}
|
|
clt = NewClientWithConfig("http://httpbin.org", opts)
|
|
if clt.requestUserAgent != userAgent {
|
|
t.Errorf(`The client had User-Agent %q, wanted %q`, clt.requestUserAgent, userAgent)
|
|
}
|
|
}
|