Update test to cover mptcp

uber · Aug 21, 2023 · 88106eb · 88106eb
1 parent 797c669
commit 88106eb
Showing 1 changed file with 18 additions and 16 deletions.
diff --git a/channel_test.go b/channel_test.go
@@ -44,24 +44,26 @@ func toMap(fields LogFields) map[string]interface{} {
 }
 
 func TestNewChannel(t *testing.T) {
-	ch, err := NewChannel("svc", &ChannelOptions{
-		ProcessName: "pname",
-		EnableMPTCP: true,
-	})
-	require.NoError(t, err, "NewChannel failed")
-	assert.Equal(t, LocalPeerInfo{
-		ServiceName: "svc",
-		PeerInfo: PeerInfo{
+	for _, mptcp := range []bool{true, false} {
+		ch, err := NewChannel("svc", &ChannelOptions{
 			ProcessName: "pname",
-			HostPort:    ephemeralHostPort,
-			IsEphemeral: true,
-			Version: PeerVersion{
-				Language:        "go",
-				LanguageVersion: strings.TrimPrefix(runtime.Version(), "go"),
-				TChannelVersion: VersionInfo,
+			EnableMPTCP: mptcp,
+		})
+		require.NoError(t, err, "NewChannel failed")
+		assert.Equal(t, LocalPeerInfo{
+			ServiceName: "svc",
+			PeerInfo: PeerInfo{
+				ProcessName: "pname",
+				HostPort:    ephemeralHostPort,
+				IsEphemeral: true,
+				Version: PeerVersion{
+					Language:        "go",
+					LanguageVersion: strings.TrimPrefix(runtime.Version(), "go"),
+					TChannelVersion: VersionInfo,
+				},
 			},
-		},
-	}, ch.PeerInfo(), "Wrong local peer info")
+		}, ch.PeerInfo(), "Wrong local peer info")
+	}
 }
 
 func TestLoggers(t *testing.T) {