github.com/pingcap/ticdc@v0.0.0-20220526033649-485a10ef2652/pkg/pipeline/test.go

// Copyright 2020 PingCAP, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// See the License for the specific language governing permissions and
// limitations under the License.

package pipeline

import "github.com/pingcap/ticdc/pkg/context"

// SendMessageToNode4Test sends messages to specified `Node` through `Receive` in order.
// This function is only for testing.
// Only `Node.Receive` will be called, other functions in `Node` will never be called.
// When the `Receive` function of the `Node` returns an error, this function will return the message that caused the error and the error
func SendMessageToNode4Test(ctx context.Context, node Node, msgs []Message, outputCh chan Message) (Message, error) {
	nodeCtx := newNodeContext(ctx, Message{}, outputCh)
	for _, msg := range msgs {
		err := node.Receive(withMessage(nodeCtx, msg))
		if err != nil {
			return msg, err
		}
	}
	return Message{}, nil
}

// MockNodeContext4Test creates a node context with a message and a output channel for tests.
func MockNodeContext4Test(ctx context.Context, msg Message, outputCh chan Message) NodeContext {
	return newNodeContext(ctx, msg, outputCh)
}