warcraft_test.go

// Copyright 2021 Wayback Archiver. All rights reserved.
// Use of this source code is governed by the MIT
// license that can be found in the LICENSE file.

package warcraft // import "github.com/wabarc/warcraft"

import (
	"context"
	"fmt"
	"io/ioutil"
	"net/http"
	"net/url"
	"os"
	"testing"

	"github.com/wabarc/helper"
)

func TestDownload(t *testing.T) {
	if _, err := findWgetExecPath(); err != nil {
		t.Skip(err.Error(), ", skipped")
	}

	_, mux, server := helper.MockServer()
	defer server.Close()

	mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintf(w, "Hello, Golang.")
	})

	uri := server.URL
	in, err := url.Parse(uri)
	if err != nil {
		t.Fatal(err)
	}

	dir, err := ioutil.TempDir("", "warcraft")
	if err != nil {
		t.Fatal(err)
	}
	defer os.RemoveAll(dir)

	warc := New()
	path, err := warc.Download(context.TODO(), in)
	if err != nil {
		t.Fatal(err)
	}
	defer os.Remove(path)

	if !helper.Exists(path) {
		t.Log(path)
		t.Errorf(`download warc file failed`)
	}
}

func TestUserAgent(t *testing.T) {
	if _, err := findWgetExecPath(); err != nil {
		t.Skip(err.Error(), ", skipped")
	}

	_, mux, server := helper.MockServer()
	defer server.Close()

	userAgent := "Mozilla/5.0 (Windows NT 6.1; Win64; x64; rv:47.0) Gecko/20100101 Firefox/47.0"
	mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		ua := r.Header.Get("User-Agent")
		if ua != userAgent {
			t.Fatalf("Unexpected create warc with User-Agent, got %s instead of %s", ua, userAgent)
		}
		fmt.Fprintf(w, "Hello, Golang.")
	})

	uri := server.URL
	in, err := url.Parse(uri)
	if err != nil {
		t.Fatal(err)
	}

	dir, err := ioutil.TempDir("", "warcraft")
	if err != nil {
		t.Fatal(err)
	}
	defer os.RemoveAll(dir)

	warc := &Warcraft{UserAgent: userAgent}
	path, err := warc.Download(context.TODO(), in)
	if err != nil {
		t.Fatal(err)
	}
	defer os.Remove(path)

	if !helper.Exists(path) {
		t.Log(path)
		t.Errorf(`download warc file failed`)
	}
}