capture_test.go RawHistoryFolder

1package harness
2
3import (
4	"errors"
5	"image"
6	"os"
7	"path/filepath"
8	"strconv"
9	"testing"
10	"time"
11)
12
13func TestCaptureSelectsByCommitTimestamp(t *testing.T) {
14	dir := t.TempDir()
15	token := time.Now().Add(-time.Minute).Truncate(time.Second)
16	for _, tc := range []struct {
17		name     string
18		mtime    time.Time
19		complete bool
20	}{
21		{"frame-000001.png", token.Add(-time.Second), true},
22		{"frame-000002.png", token.Add(time.Second), true},
23		{"frame-000003.png", token.Add(2 * time.Second), false},
24		{"frame-000004.png", token.Add(3 * time.Second), true},
25		{"latest.png", token.Add(4 * time.Second), true},
26	} {
27		path := filepath.Join(dir, tc.name)
28		if tc.complete {
29			if err := SavePNG(path, image.NewNRGBA(image.Rect(0, 0, 2, 2))); err != nil {
30				t.Fatal(err)
31			}
32		} else if err := os.WriteFile(path, []byte("incomplete"), 0600); err != nil {
33			t.Fatal(err)
34		}
35		if err := os.Chtimes(path, tc.mtime, tc.mtime); err != nil {
36			t.Fatal(err)
37		}
38	}
39	request := filepath.Join(dir, "capture-request")
40	if err := os.WriteFile(request, []byte(strconv.FormatInt(token.UnixNano(), 10)), 0600); err != nil {
41		t.Fatal(err)
42	}
43	after, err := captureRequestTime(request)
44	if err != nil || !after.Equal(token) {
45		t.Fatalf("token %v %v", after, err)
46	}
47	// frame-2 finished after the token but may have been rendered before it.
48	name, err := nextFrame(dir, after)
49	if err != nil || name != "frame-000003.png" {
50		t.Fatalf("selected %q %v", name, err)
51	}
52	if name, err := nextFrame(dir, token.Add(2500*time.Millisecond)); err != nil || name != "" {
53		t.Fatalf("one screenshot after the token must not be selected: %q %v", name, err)
54	}
55	capture := filepath.Join(dir, "capture.png")
56	if err := copyNextFrame(dir, after, capture); !errors.Is(err, os.ErrNotExist) {
57		t.Fatalf("incomplete frame accepted: %v", err)
58	}
59	if err := SavePNG(filepath.Join(dir, name), image.NewNRGBA(image.Rect(0, 0, 2, 2))); err != nil {
60		t.Fatal(err)
61	}
62	if err := os.Chtimes(filepath.Join(dir, name), token.Add(2*time.Second), token.Add(2*time.Second)); err != nil {
63		t.Fatal(err)
64	}
65	if err := copyNextFrame(dir, after, capture); err != nil {
66		t.Fatal(err)
67	}
68	if got, err := LoadPNG(capture); err != nil || got.Bounds().Dx() != 2 {
69		t.Fatalf("capture %v %v", got, err)
70	}
71	if err := os.WriteFile(request, []byte("not-a-time"), 0600); err != nil {
72		t.Fatal(err)
73	}
74	if _, err := captureRequestTime(request); err == nil {
75		t.Fatal("invalid token accepted")
76	}
77}