update_test.go RawHistoryFolder

1package agent
2
3import (
4	"context"
5	"errors"
6	"net/http"
7	"os"
8	"path/filepath"
9	"testing"
10)
11
12func TestIsDevVersion(t *testing.T) {
13	for _, version := range []string{"", "dev", "0.0.0", "0.0.0-dev", "0.0.0-dev (abc123, today)", "0.0.0-dev+local"} {
14		if !isDevVersion(version) {
15			t.Errorf("isDevVersion(%q) = false, want true", version)
16		}
17	}
18	for _, version := range []string{"0.0.1", "1.0.0", "0.0.0-rc1"} {
19		if isDevVersion(version) {
20			t.Errorf("isDevVersion(%q) = true, want false", version)
21		}
22	}
23}
24
25type countingRoundTripper struct {
26	calls int
27}
28
29func (r *countingRoundTripper) RoundTrip(*http.Request) (*http.Response, error) {
30	r.calls++
31	return nil, errors.New("unexpected network request")
32}
33
34func TestCheckForUpdateSkipsDevVersionWithoutCacheOrNetwork(t *testing.T) {
35	home := t.TempDir()
36	transport := &countingRoundTripper{}
37	previousTransport := http.DefaultTransport
38	http.DefaultTransport = transport
39	t.Cleanup(func() { http.DefaultTransport = previousTransport })
40
41	ctx, cancel := context.WithCancel(context.Background())
42	cancel()
43
44	info := CheckForUpdate(ctx, home, "0.0.0-dev")
45	if info != (UpdateInfo{}) {
46		t.Fatalf("development check = %+v, want an empty result", info)
47	}
48	if _, err := os.Stat(filepath.Join(home, updateCheckFile)); !os.IsNotExist(err) {
49		t.Fatalf("development check wrote a cache file: %v", err)
50	}
51	if transport.calls != 0 {
52		t.Fatalf("development check made %d network requests, want 0", transport.calls)
53	}
54}