github.com/pingcap/tiflow@v0.0.0-20240520035814-5bf52d54e205/cdc/redo/common/util_test.go (about)

     1  //  Copyright 2021 PingCAP, Inc.
     2  //
     3  //  Licensed under the Apache License, Version 2.0 (the "License");
     4  //  you may not use this file except in compliance with the License.
     5  //  You may obtain a copy of the License at
     6  //
     7  //      http://www.apache.org/licenses/LICENSE-2.0
     8  //
     9  //  Unless required by applicable law or agreed to in writing, software
    10  //  distributed under the License is distributed on an "AS IS" BASIS,
    11  //  See the License for the specific language governing permissions and
    12  //  limitations under the License.
    13  
    14  package common
    15  
    16  import (
    17  	"testing"
    18  
    19  	"github.com/pingcap/tiflow/cdc/model"
    20  	"github.com/stretchr/testify/require"
    21  )
    22  
    23  func TestGetChangefeedFiles(t *testing.T) {
    24  	t.Parallel()
    25  
    26  	cases := []struct {
    27  		fileNames  []string
    28  		changefeed model.ChangeFeedID
    29  		want       []string
    30  	}{
    31  		{
    32  			fileNames: []string{
    33  				"captureID_test-2_uuid1.log",
    34  				"captureID_test-3_uuid2.log",
    35  				"captureID_test-1_uuid3.log",
    36  			},
    37  			changefeed: model.DefaultChangeFeedID("test-1"),
    38  			want: []string{
    39  				"captureID_test-1_uuid3.log",
    40  			},
    41  		},
    42  		{
    43  			fileNames: []string{
    44  				"captureID_n1_test-2_uuid4.log",
    45  				"captureID_n2_test-2_uuid5.log",
    46  				"captureID_n1_test-1_uuid6.log",
    47  			},
    48  			changefeed: model.ChangeFeedID{Namespace: "n1", ID: "test-2"},
    49  			want: []string{
    50  				"captureID_n1_test-2_uuid4.log",
    51  			},
    52  		},
    53  	}
    54  
    55  	for _, c := range cases {
    56  		got := FilterChangefeedFiles(c.fileNames, c.changefeed)
    57  		require.Equal(t, c.want, got)
    58  	}
    59  }