portforward: tunnel spdy through websockets

Kubernetes-commit: 8b447d8c97e8823b4308eb91cf7d75693e867c61

tools/portforward/fallback_dialer_test.go

@@ -0,0 +1,53 @@
+/*
+Copyright 2024 The Kubernetes Authors.
+
+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,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package portforward
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestFallbackDialer(t *testing.T) {
+	protocol := "v6.fake.k8s.io"
+	// If "shouldFallback" is false, then only primary should be dialed.
+	primary := &fakeDialer{dialed: false}
+	primary.negotiatedProtocol = protocol
+	secondary := &fakeDialer{dialed: false}
+	fallbackDialer := NewFallbackDialer(primary, secondary, alwaysFalse)
+	_, negotiated, err := fallbackDialer.Dial(protocol)
+	assert.True(t, primary.dialed, "no fallback; primary should have dialed")
+	assert.Equal(t, protocol, negotiated, "")
+	assert.False(t, secondary.dialed, "no fallback; secondary should *not* have dialed")
+	assert.Nil(t, err, "error should be nil")
+	// If "shouldFallback" is true, then primary AND secondary should be dialed.
+	primary.dialed = false // reset dialed field
+	primary.err = fmt.Errorf("bad handshake")
+	secondary.dialed = false // reset dialed field
+	secondary.negotiatedProtocol = protocol
+	fallbackDialer = NewFallbackDialer(primary, secondary, alwaysTrue)
+	_, negotiated, err = fallbackDialer.Dial(protocol)
+	assert.True(t, primary.dialed, "fallback; primary should have dialed (first)")
+	assert.True(t, secondary.dialed, "fallback; secondary should have dialed")
+	assert.Equal(t, protocol, negotiated)
+	assert.Nil(t, err)
+}
+
+func alwaysTrue(err error) bool { return true }
+
+func alwaysFalse(err error) bool { return false }