ILSpy/ILSpy.Tests/Editor/FoldingContextMenuTests.cs

// Copyright (c) 2026 AlphaSierraPapa for the SharpDevelop Team
//
// Permission is hereby granted, free of charge, to any person obtaining a copy of this
// software and associated documentation files (the "Software"), to deal in the Software
// without restriction, including without limitation the rights to use, copy, modify, merge,
// publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons
// to whom the Software is furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in all copies or
// substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
// INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR
// PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE
// FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
// OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
// DEALINGS IN THE SOFTWARE.

using System.Threading.Tasks;

using Avalonia.Headless.NUnit;
using Avalonia.Threading;

using AwesomeAssertions;

using ICSharpCode.ILSpy.Properties;

using ICSharpCode.ILSpy;
using ICSharpCode.ILSpy.AppEnv;
using ICSharpCode.ILSpy.TextView;
using ICSharpCode.ILSpy.TreeNodes;

using NUnit.Framework;

namespace ICSharpCode.ILSpy.Tests;

[TestFixture]
public class FoldingContextMenuTests
{
	[AvaloniaTest]
	public async Task Toggle_All_Folding_Entry_Collapses_Then_Expands_The_Document()
	{
		// Decompiling a type yields brace foldings; the "Toggle all folding" menu entry must collapse
		// them all, then expand them all on a second invocation -- via DecompilerTextView.ToggleAllFoldings.
		var (window, vm) = await TestHarness.BootAsync(3);

		// A small CoreLib type still yields brace foldings (one per method body) but decompiles fast
		// enough for the headless 15s wait on slow CI runners; a full System.Linq.Enumerable decompile
		// does not.
		var coreLibName = typeof(object).Assembly.GetName().Name!;
		var typeNode = vm.AssemblyTreeModel.FindNode<TypeTreeNode>(coreLibName, "System", "System.Object");
		vm.AssemblyTreeModel.SelectNode(typeNode);
		await vm.DockWorkspace.WaitForDecompiledTextAsync();

		var view = await window.WaitForComponent<DecompilerTextView>();
		for (int i = 0; i < 8; i++)
		{
			Dispatcher.UIThread.RunJobs();
			await Task.Delay(25);
		}

		view.HasFoldings.Should().BeTrue("decompiling a type produces brace foldings");

		var registry = AppComposition.Current.GetExport<ContextMenuEntryRegistry>();
		var toggleAll = registry.GetEntry(nameof(Resources.ToggleFolding));   // "Toggle All Folding"
		var toggleOne = registry.GetEntry(nameof(Resources._ToggleFolding));  // "Toggle Folding"
		var context = new TextViewContext { TextView = view };
		toggleAll.IsVisible(context).Should().BeTrue("the folding entries show when the document has foldings");
		toggleOne.IsVisible(context).Should().BeTrue();
		toggleAll.IsVisible(new TextViewContext { TextView = null }).Should()
			.BeFalse("with no text view there is nothing to fold");

		// Toggle-all forces a uniform state: collapse-all when any fold is open, otherwise expand-all.
		// So consecutive invocations flip between "fully collapsed" (>0) and "fully expanded" (0).
		toggleAll.Execute(context);
		Dispatcher.UIThread.RunJobs();
		int after1 = view.FoldedFoldingCount;

		toggleAll.Execute(context);
		Dispatcher.UIThread.RunJobs();
		int after2 = view.FoldedFoldingCount;

		after1.Should().NotBe(after2, "Toggle all folding must flip the fold state");
		System.Math.Min(after1, after2).Should().Be(0, "one toggle state is fully expanded");
		System.Math.Max(after1, after2).Should().BeGreaterThan(0, "the other toggle state is fully collapsed");
	}
}