Function TestPget

pget_test.go:26–101

(t *testing.T)

}

func TestPget(t *testing.T) {
	// listening file server
	mux := http.NewServeMux()
	mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		http.Redirect(w, r, "/moo", http.StatusFound)
	})

	mux.HandleFunc("/moo", func(w http.ResponseWriter, r *http.Request) {
		http.Redirect(w, r, "/mooo", http.StatusFound)
	})

	mux.HandleFunc("/mooo", func(w http.ResponseWriter, r *http.Request) {
		http.Redirect(w, r, "/test.tar.gz", http.StatusFound)
	})

	mux.HandleFunc("/test.tar.gz", func(w http.ResponseWriter, r *http.Request) {
		fp := "_testdata/test.tar.gz"
		data, err := ioutil.ReadFile(fp)
		if err != nil {
			t.Errorf("failed to readfile: %s", err)
		}
		http.ServeContent(w, r, fp, time.Now(), bytes.NewReader(data))
	})

	ts := httptest.NewServer(mux)
	defer ts.Close()

	// begin tests
	url := ts.URL

	tmpdir := t.TempDir()

	cfg := &DownloadConfig{
		Filename:      "test.tar.gz",
		ContentLength: 1719652,
		Dirname:       tmpdir,
		Procs:         4,
		URLs:          []string{ts.URL},
		Client:        newDownloadClient(1),
	}

	t.Run("check", func(t *testing.T) {
		target, err := Check(context.Background(), &CheckConfig{
			URLs:    []string{url},
			Timeout: 10 * time.Second,
		})

		if err != nil {
			t.Fatalf("failed to check header: %s", err)
		}

		if len(target.URLs) == 0 {
			t.Fatalf("invalid URL length %d", len(target.URLs))
		}

		// could redirect?
		assert.NotEqual(t, target.URLs[0], url, "failed to get of the last url in the redirect")
	})