miniflux/http/client/client_test.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)
	}
}