zk/tree_walker_test.go at main · Shopify/zk · 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
package zk

import (
	"context"
	"reflect"
	"testing"
)

func TestTreeWalker(t *testing.T) {
	paths := []string{
		"/gozk-test-walker",
		"/gozk-test-walker/a",
		"/gozk-test-walker/a/b",
		"/gozk-test-walker/a/c",
		"/gozk-test-walker/a/c/d",
	}

	expectVisitedExact := func(t *testing.T, expected []string, visited []string) {
		if !reflect.DeepEqual(expected, visited) {
			t.Fatalf("%s saw unexpected paths:\n Expected: %+v\n Got:      %+v",
				t.Name(), expected, visited)
		}
	}

	WithTestCluster(t, 1, nil, logWriter{t: t, p: "[ZKERR] "}, func(t *testing.T, tc *TestCluster) {
		WithConnectAll(t, tc, func(t *testing.T, c *Conn, _ <-chan Event) {
			for _, p := range paths {
				if path, err := c.Create(context.Background(), p, []byte{1, 2, 3, 4}, 0, WorldACL(PermAll)); err != nil {
					t.Fatalf("Create returned error: %+v", err)
				} else if path != p {
					t.Fatalf("Create returned different path '%s' != '%s'", path, p)
				}
			}

			t.Run("All_BreadthFirstOrder", func(t *testing.T) {
				nodes, walkErr := NewTreeWalker(c.Children, "/gozk-test-walker", BreadthFirstOrder).All(context.Background())
				var visited []string
				for path := range nodes {
					visited = append(visited, path)
				}
				if err := walkErr(); err != nil {
					t.Fatalf("All returned an error: %+v", err)
				}

				expected := []string{
					"/gozk-test-walker",
					"/gozk-test-walker/a",
					"/gozk-test-walker/a/b",
					"/gozk-test-walker/a/c",
					"/gozk-test-walker/a/c/d",
				}
				expectVisitedExact(t, expected, visited)
			})

			t.Run("All_DepthFirstOrder", func(t *testing.T) {
				nodes, walkErr := NewTreeWalker(c.Children, "/gozk-test-walker", DepthFirstOrder).All(context.Background())
				var visited []string
				for path := range nodes {
					visited = append(visited, path)
				}
				if err := walkErr(); err != nil {
					t.Fatalf("All returned an error: %+v", err)
				}

				expected := []string{
					"/gozk-test-walker/a/b",
					"/gozk-test-walker/a/c/d",
					"/gozk-test-walker/a/c",
					"/gozk-test-walker/a",
					"/gozk-test-walker",
				}
				expectVisitedExact(t, expected, visited)
			})
		})
	})
}