Function TestLocalNodeStore

pkg/node/local_node_store_test.go:34–112

(t *testing.T)

}

func TestLocalNodeStore(t *testing.T) {
	var waitObserve sync.WaitGroup
	var observed []uint32
	expected := []uint32{1, 2, 3, 4, 5}

	waitObserve.Add(1)

	ts := testSynchronizer{identity: make(chan uint32, 1)}

	// observe observes changes to the LocalNodeStore and completes
	// waitObserve after the last change has been observed.
	observe := func(store *LocalNodeStore) {
		store.Observe(context.TODO(),
			func(n LocalNode) {
				observed = append(observed, n.NodeIdentity)

				if n.NodeIdentity == expected[len(expected)-1] {
					waitObserve.Done()
				}
			},
			func(err error) {},
		)
	}

	// update adds a start hook to the application that modifies
	// the local node.
	update := func(lc cell.Lifecycle, store *LocalNodeStore) {
		lc.Append(cell.Hook{
			OnStart: func(cell.HookContext) error {
				// emit 2, 3, 4, 5
				for _, i := range expected[1:] {
					if i == 5 {
						ts.identity <- i
						continue
					}

					store.Update(func(n *LocalNode) {
						n.NodeIdentity = i
					})
				}
				return nil
			},
		})
	}

	hive := hive.New(
		LocalNodeStoreCell,

		cell.Provide(func() cmtypes.ClusterInfo {
			return cmtypes.ClusterInfo{
				Name: "test",
				ID:   1,
			}
		}),
		cell.Provide(func() LocalNodeSynchronizer { return ts }),
		cell.Invoke(observe),
		cell.Invoke(update),
	)