bfs/object_test.go at main · bsm/bfs · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
package bfs_test

import (
	"errors"
	"io"
	"testing"

	"github.com/bsm/bfs"
)

func TestObject(t *testing.T) {
	ctx := t.Context()
	obj, err := bfs.NewObject(ctx, "mem:///path/to/file.txt")
	if err != nil {
		t.Fatal("Unexpected error", err)
	}

	t.Run("name", func(t *testing.T) {
		if exp, got := "path/to/file.txt", obj.Name(); exp != got {
			t.Errorf("Expected %q, got %q", exp, got)
		}
	})

	t.Run("not found", func(t *testing.T) {
		if _, err := obj.Head(ctx); !errors.Is(err, bfs.ErrNotFound) {
			t.Errorf("Expected %v, got %v", bfs.ErrNotFound, err)
		}

		if _, err := obj.Open(ctx); !errors.Is(err, bfs.ErrNotFound) {
			t.Errorf("Expected %v, got %v", bfs.ErrNotFound, err)
		}
	})

	t.Run("CRUD", func(t *testing.T) {
		w, err := obj.Create(ctx, nil)
		if err != nil {
			t.Fatal("Unexpected error", err)
		}
		defer w.Discard()

		if n, err := w.Write([]byte("TESTDATA")); err != nil {
			t.Fatal("Unexpected error", err)
		} else if n != 8 {
			t.Errorf("Expected %v, got %v", 8, n)
		}

		if err := w.Commit(); err != nil {
			t.Fatal("Unexpected error", err)
		}

		if i, err := obj.Head(ctx); err != nil {
			t.Fatal("Unexpected error", err)
		} else if i.Size != 8 {
			t.Errorf("Expected %v, got %v", 8, i.Size)
		} else if i.Name != obj.Name() {
			t.Errorf("Expected %v, got %v", obj.Name(), i.Name)
		}

		r, err := obj.Open(ctx)
		if err != nil {
			t.Fatal("Unexpected error", err)
		}
		defer r.Close()

		if data, err := io.ReadAll(r); err != nil {
			t.Fatal("Unexpected error", err)
		} else if exp, got := "TESTDATA", string(data); exp != got {
			t.Errorf("Expected %q, got %q", exp, got)
		}

		if err := r.Close(); err != nil {
			t.Fatal("Unexpected error", err)
		}

		if err := obj.Remove(ctx); err != nil {
			t.Fatal("Unexpected error", err)
		}

		if _, err := obj.Head(ctx); !errors.Is(err, bfs.ErrNotFound) {
			t.Errorf("Expected %v, got %v", bfs.ErrNotFound, err)
		}
	})
}