Function startNodeMock

internal/nodehub/hub_test.go:102–119

(t *testing.T, cc *grpc.ClientConn, nodeID string)

}

func startNodeMock(t *testing.T, cc *grpc.ClientConn, nodeID string) *nodeMock {
	t.Helper()
	client := nodev1.NewNodeAgentClient(cc)
	ctx := metadata.AppendToOutgoingContext(context.Background(), "x-node-id", nodeID)
	stream, err := client.Session(ctx)
	if err != nil {
		t.Fatalf("Session start: %v", err)
	}
	m := &nodeMock{
		stream:    stream,
		cancelIDs: make(chan string, 16),
		acks:      make(chan []byte, 16),
		handlers:  make(map[string]func(string, []byte) (bool, []byte, string)),
		done:      make(chan struct{}),
	}
	go m.recvLoop()
	return m
}

func (m *nodeMock) handle(method string, fn func(reqID string, body []byte) (ok bool, respBody []byte, errStr string)) {
	m.mu.Lock()