Function TestReAuthOnlyWhenIdle

internal/auth/streaming/pool_hook_state_test.go:14–63

TestReAuthOnlyWhenIdle verifies that re-authentication only happens when a connection is in IDLE state, not when it's IN_USE.

(t *testing.T)

// TestReAuthOnlyWhenIdle verifies that re-authentication only happens when
// a connection is in IDLE state, not when it's IN_USE.
func TestReAuthOnlyWhenIdle(t *testing.T) {
	// Create a connection
	cn := pool.NewConn(nil)

	// Initialize to IDLE state
	cn.GetStateMachine().Transition(pool.StateInitializing)
	cn.GetStateMachine().Transition(pool.StateIdle)

	// Simulate connection being acquired (IDLE → IN_USE)
	if !cn.CompareAndSwapUsed(false, true) {
		t.Fatal("Failed to acquire connection")
	}

	// Verify state is IN_USE
	if state := cn.GetStateMachine().GetState(); state != pool.StateInUse {
		t.Errorf("Expected state IN_USE, got %s", state)
	}

	// Try to transition to UNUSABLE (for reauth) - should fail
	_, err := cn.GetStateMachine().TryTransition([]pool.ConnState{pool.StateIdle}, pool.StateUnusable)
	if err == nil {
		t.Error("Expected error when trying to transition IN_USE → UNUSABLE, but got none")
	}

	// Verify state is still IN_USE
	if state := cn.GetStateMachine().GetState(); state != pool.StateInUse {
		t.Errorf("Expected state to remain IN_USE, got %s", state)
	}

	// Release connection (IN_USE → IDLE)
	if !cn.CompareAndSwapUsed(true, false) {
		t.Fatal("Failed to release connection")
	}

	// Verify state is IDLE
	if state := cn.GetStateMachine().GetState(); state != pool.StateIdle {
		t.Errorf("Expected state IDLE, got %s", state)
	}

	// Now try to transition to UNUSABLE - should succeed
	_, err = cn.GetStateMachine().TryTransition([]pool.ConnState{pool.StateIdle}, pool.StateUnusable)
	if err != nil {
		t.Errorf("Failed to transition IDLE → UNUSABLE: %v", err)
	}

	// Verify state is UNUSABLE
	if state := cn.GetStateMachine().GetState(); state != pool.StateUnusable {
		t.Errorf("Expected state UNUSABLE, got %s", state)
	}
}

// TestReAuthWaitsForConnectionToBeIdle verifies that the re-auth worker
// waits for a connection to become IDLE before performing re-authentication.