diff options
Diffstat (limited to 'weed/worker/client_tls_test.go')
| -rw-r--r-- | weed/worker/client_tls_test.go | 146 |
1 files changed, 0 insertions, 146 deletions
diff --git a/weed/worker/client_tls_test.go b/weed/worker/client_tls_test.go deleted file mode 100644 index d95d5f4f5..000000000 --- a/weed/worker/client_tls_test.go +++ /dev/null @@ -1,146 +0,0 @@ -package worker - -import ( - "strings" - "testing" - "time" - - "google.golang.org/grpc" - "google.golang.org/grpc/credentials/insecure" -) - -func TestGrpcClientTLSDetection(t *testing.T) { - // Test that the client can be created with a dial option - dialOption := grpc.WithTransportCredentials(insecure.NewCredentials()) - client := NewGrpcAdminClient("localhost:33646", "test-worker", dialOption) - - // Test that the client has the correct dial option - if client.dialOption == nil { - t.Error("Client should have a dial option") - } - - t.Logf("Client created successfully with dial option") -} - -func TestCreateAdminClientGrpc(t *testing.T) { - // Test client creation - admin server port gets transformed to gRPC port - dialOption := grpc.WithTransportCredentials(insecure.NewCredentials()) - client, err := CreateAdminClient("localhost:23646", "test-worker", dialOption) - if err != nil { - t.Fatalf("Failed to create admin client: %v", err) - } - - if client == nil { - t.Fatal("Client should not be nil") - } - - // Verify it's the correct type - grpcClient, ok := client.(*GrpcAdminClient) - if !ok { - t.Fatal("Client should be GrpcAdminClient type") - } - - // The admin address should be transformed to the gRPC port (HTTP + 10000) - expectedAddress := "localhost:33646" // 23646 + 10000 - if grpcClient.adminAddress != expectedAddress { - t.Errorf("Expected admin address %s, got %s", expectedAddress, grpcClient.adminAddress) - } - - if grpcClient.workerID != "test-worker" { - t.Errorf("Expected worker ID test-worker, got %s", grpcClient.workerID) - } -} - -func TestConnectionTimeouts(t *testing.T) { - // Test that connections have proper timeouts - // Use localhost with a port that's definitely closed - dialOption := grpc.WithTransportCredentials(insecure.NewCredentials()) - client := NewGrpcAdminClient("localhost:1", "test-worker", dialOption) // Port 1 is reserved and won't be open - - // Test that the connection creation fails when actually trying to use it - start := time.Now() - err := client.Connect() // This should fail when trying to establish the stream - duration := time.Since(start) - - if err == nil { - t.Error("Expected connection to closed port to fail") - } else { - t.Logf("Connection failed as expected: %v", err) - } - - // Should fail quickly but not too quickly - if duration > 10*time.Second { - t.Errorf("Connection attempt took too long: %v", duration) - } -} - -func TestConnectionWithDialOption(t *testing.T) { - // Test that the connection uses the provided dial option - dialOption := grpc.WithTransportCredentials(insecure.NewCredentials()) - client := NewGrpcAdminClient("localhost:1", "test-worker", dialOption) // Port 1 is reserved and won't be open - - // Test the actual connection - err := client.Connect() - if err == nil { - t.Error("Expected connection to closed port to fail") - client.Disconnect() // Clean up if it somehow succeeded - } else { - t.Logf("Connection failed as expected: %v", err) - } - - // The error should indicate a connection failure - if err != nil && err.Error() != "" { - t.Logf("Connection error message: %s", err.Error()) - // The error should contain connection-related terms - if !strings.Contains(err.Error(), "connection") && !strings.Contains(err.Error(), "dial") { - t.Logf("Error message doesn't indicate connection issues: %s", err.Error()) - } - } -} - -func TestClientWithSecureDialOption(t *testing.T) { - // Test that the client correctly uses a secure dial option - // This would normally use LoadClientTLS, but for testing we'll use insecure - dialOption := grpc.WithTransportCredentials(insecure.NewCredentials()) - client := NewGrpcAdminClient("localhost:33646", "test-worker", dialOption) - - if client.dialOption == nil { - t.Error("Client should have a dial option") - } - - t.Logf("Client created successfully with dial option") -} - -func TestConnectionWithRealAddress(t *testing.T) { - // Test connection behavior with a real address that doesn't support gRPC - dialOption := grpc.WithTransportCredentials(insecure.NewCredentials()) - client := NewGrpcAdminClient("www.google.com:80", "test-worker", dialOption) // HTTP port, not gRPC - - err := client.Connect() - if err == nil { - t.Log("Connection succeeded unexpectedly") - client.Disconnect() - } else { - t.Logf("Connection failed as expected: %v", err) - } -} - -func TestDialOptionUsage(t *testing.T) { - // Test that the provided dial option is used for connections - dialOption := grpc.WithTransportCredentials(insecure.NewCredentials()) - client := NewGrpcAdminClient("localhost:1", "test-worker", dialOption) // Port 1 won't support gRPC at all - - // Verify the dial option is stored - if client.dialOption == nil { - t.Error("Dial option should be stored in client") - } - - // Test connection fails appropriately - err := client.Connect() - if err == nil { - t.Error("Connection should fail to non-gRPC port") - client.Disconnect() - } else { - t.Logf("Connection failed as expected: %v", err) - } -} |
