Function TestManagerRefactoring

maintnotifications/manager_test.go:152–337

(t *testing.T)

}

func TestManagerRefactoring(t *testing.T) {
	t.Run("AtomicStateTracking", func(t *testing.T) {
		config := DefaultConfig()
		client := &MockClient{options: &MockOptions{}}

		manager, err := NewManager(client, nil, config)
		if err != nil {
			t.Fatalf("Failed to create maintnotifications manager: %v", err)
		}
		defer manager.Close()

		// Test initial state
		if manager.IsHandoffInProgress() {
			t.Error("Expected no handoff in progress initially")
		}

		if manager.GetActiveOperationCount() != 0 {
			t.Errorf("Expected 0 active operations, got %d", manager.GetActiveOperationCount())
		}

		if manager.GetState() != StateIdle {
			t.Errorf("Expected StateIdle, got %v", manager.GetState())
		}

		// Add an operation
		ctx := context.Background()
		deadline := time.Now().Add(30 * time.Second)
		err = manager.TrackMovingOperationWithConnID(ctx, "new-endpoint:6379", deadline, 12345, 1)
		if err != nil {
			t.Fatalf("Failed to track operation: %v", err)
		}

		// Test state after adding operation
		if !manager.IsHandoffInProgress() {
			t.Error("Expected handoff in progress after adding operation")
		}

		if manager.GetActiveOperationCount() != 1 {
			t.Errorf("Expected 1 active operation, got %d", manager.GetActiveOperationCount())
		}

		if manager.GetState() != StateMoving {
			t.Errorf("Expected StateMoving, got %v", manager.GetState())
		}

		// Remove the operation
		manager.UntrackOperationWithConnID(12345, 1)

		// Test state after removing operation
		if manager.IsHandoffInProgress() {
			t.Error("Expected no handoff in progress after removing operation")
		}

		if manager.GetActiveOperationCount() != 0 {
			t.Errorf("Expected 0 active operations, got %d", manager.GetActiveOperationCount())
		}

		if manager.GetState() != StateIdle {