Add @mockable(superclass: UIViewController) to Flow.stencil for Mockolo 2.6.1

aleksey-krauchuk-gotinder · aleksey-krauchuk-gotinder · commit 4e5ea3fe5b0a · 2026-04-07T16:10:15.000+02:00
Apply the same fix from Interface.stencil to Flow.stencil, where
ViewControllable protocols are generated for view-owning nodes (e.g. Root).

Made-with: Cursor
diff --git a/Sources/NodesGenerator/Resources/Stencils/Flow.stencil b/Sources/NodesGenerator/Resources/Stencils/Flow.stencil
@@ -24,8 +24,12 @@ internal protocol {{ node_name }}ContextInterface: Context {}
 /// typically inherits protocols that include pre-defined behavior for convenience. Add additional methods for custom
 /// presentation or containment as desired.
 {% if not view_controllable_mock_contents %}
+{% if view_controllable_type == "ViewControllable" %}
+/// @mockable(superclass: UIViewController)
+{% else %}
 /// @mockable
 {% endif %}
+{% endif %}
 @MainActor
 internal protocol {{ node_name }}ViewControllable: {{ view_controllable_type }} {}
 {% endif %}
diff --git a/Tests/NodesGeneratorTests/__Snapshots__/PresetGeneratorTests/testGenerateRootPreset.Contents-RootFlow-swift.txt b/Tests/NodesGeneratorTests/__Snapshots__/PresetGeneratorTests/testGenerateRootPreset.Contents-RootFlow-swift.txt
@@ -13,7 +13,7 @@ internal protocol RootContextInterface: Context {}
 /// Defines the interface between the `Flow` and its view in order to present or contain other views. This protocol
 /// typically inherits protocols that include pre-defined behavior for convenience. Add additional methods for custom
 /// presentation or containment as desired.
-/// @mockable
+/// @mockable(superclass: UIViewController)
 @MainActor
 internal protocol RootViewControllable: ViewControllable {}
 
diff --git a/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-AppKit-Flow.txt b/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-AppKit-Flow.txt
@@ -11,7 +11,7 @@ internal protocol ___VARIABLE_productName___ContextInterface: Context {}
 /// Defines the interface between the `Flow` and its view in order to present or contain other views. This protocol
 /// typically inherits protocols that include pre-defined behavior for convenience. Add additional methods for custom
 /// presentation or containment as desired.
-/// @mockable
+/// @mockable(superclass: UIViewController)
 @MainActor
 internal protocol ___VARIABLE_productName___ViewControllable: ViewControllable {}
 
diff --git a/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-AppKitCreatedForPluginList-Flow.txt b/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-AppKitCreatedForPluginList-Flow.txt
@@ -11,7 +11,7 @@ internal protocol ___VARIABLE_productName___ContextInterface: Context {}
 /// Defines the interface between the `Flow` and its view in order to present or contain other views. This protocol
 /// typically inherits protocols that include pre-defined behavior for convenience. Add additional methods for custom
 /// presentation or containment as desired.
-/// @mockable
+/// @mockable(superclass: UIViewController)
 @MainActor
 internal protocol ___VARIABLE_productName___ViewControllable: ViewControllable {}
 
diff --git a/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-AppKitSwiftUI-Flow.txt b/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-AppKitSwiftUI-Flow.txt
@@ -11,7 +11,7 @@ internal protocol ___VARIABLE_productName___ContextInterface: Context {}
 /// Defines the interface between the `Flow` and its view in order to present or contain other views. This protocol
 /// typically inherits protocols that include pre-defined behavior for convenience. Add additional methods for custom
 /// presentation or containment as desired.
-/// @mockable
+/// @mockable(superclass: UIViewController)
 @MainActor
 internal protocol ___VARIABLE_productName___ViewControllable: ViewControllable {}
 
diff --git a/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-AppKitSwiftUICreatedForPluginList-Flow.txt b/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-AppKitSwiftUICreatedForPluginList-Flow.txt
@@ -11,7 +11,7 @@ internal protocol ___VARIABLE_productName___ContextInterface: Context {}
 /// Defines the interface between the `Flow` and its view in order to present or contain other views. This protocol
 /// typically inherits protocols that include pre-defined behavior for convenience. Add additional methods for custom
 /// presentation or containment as desired.
-/// @mockable
+/// @mockable(superclass: UIViewController)
 @MainActor
 internal protocol ___VARIABLE_productName___ViewControllable: ViewControllable {}
 
diff --git a/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-Custom-Flow.txt b/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-Custom-Flow.txt
@@ -11,7 +11,7 @@ internal protocol ___VARIABLE_productName___ContextInterface: Context {}
 /// Defines the interface between the `Flow` and its view in order to present or contain other views. This protocol
 /// typically inherits protocols that include pre-defined behavior for convenience. Add additional methods for custom
 /// presentation or containment as desired.
-/// @mockable
+/// @mockable(superclass: UIViewController)
 @MainActor
 internal protocol ___VARIABLE_productName___ViewControllable: ViewControllable {}
 
diff --git a/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-CustomCreatedForPluginList-Flow.txt b/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-CustomCreatedForPluginList-Flow.txt
@@ -11,7 +11,7 @@ internal protocol ___VARIABLE_productName___ContextInterface: Context {}
 /// Defines the interface between the `Flow` and its view in order to present or contain other views. This protocol
 /// typically inherits protocols that include pre-defined behavior for convenience. Add additional methods for custom
 /// presentation or containment as desired.
-/// @mockable
+/// @mockable(superclass: UIViewController)
 @MainActor
 internal protocol ___VARIABLE_productName___ViewControllable: ViewControllable {}
 
diff --git a/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-UIKit-Flow.txt b/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-UIKit-Flow.txt
@@ -11,7 +11,7 @@ internal protocol ___VARIABLE_productName___ContextInterface: Context {}
 /// Defines the interface between the `Flow` and its view in order to present or contain other views. This protocol
 /// typically inherits protocols that include pre-defined behavior for convenience. Add additional methods for custom
 /// presentation or containment as desired.
-/// @mockable
+/// @mockable(superclass: UIViewController)
 @MainActor
 internal protocol ___VARIABLE_productName___ViewControllable: ViewControllable {}
 
diff --git a/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-UIKitCreatedForPluginList-Flow.txt b/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-UIKitCreatedForPluginList-Flow.txt
@@ -11,7 +11,7 @@ internal protocol ___VARIABLE_productName___ContextInterface: Context {}
 /// Defines the interface between the `Flow` and its view in order to present or contain other views. This protocol
 /// typically inherits protocols that include pre-defined behavior for convenience. Add additional methods for custom
 /// presentation or containment as desired.
-/// @mockable
+/// @mockable(superclass: UIViewController)
 @MainActor
 internal protocol ___VARIABLE_productName___ViewControllable: ViewControllable {}
 
diff --git a/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-UIKitSwiftUI-Flow.txt b/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-UIKitSwiftUI-Flow.txt
@@ -11,7 +11,7 @@ internal protocol ___VARIABLE_productName___ContextInterface: Context {}
 /// Defines the interface between the `Flow` and its view in order to present or contain other views. This protocol
 /// typically inherits protocols that include pre-defined behavior for convenience. Add additional methods for custom
 /// presentation or containment as desired.
-/// @mockable
+/// @mockable(superclass: UIViewController)
 @MainActor
 internal protocol ___VARIABLE_productName___ViewControllable: ViewControllable {}
 
diff --git a/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-UIKitSwiftUICreatedForPluginList-Flow.txt b/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-UIKitSwiftUICreatedForPluginList-Flow.txt
@@ -11,7 +11,7 @@ internal protocol ___VARIABLE_productName___ContextInterface: Context {}
 /// Defines the interface between the `Flow` and its view in order to present or contain other views. This protocol
 /// typically inherits protocols that include pre-defined behavior for convenience. Add additional methods for custom
 /// presentation or containment as desired.
-/// @mockable
+/// @mockable(superclass: UIViewController)
 @MainActor
 internal protocol ___VARIABLE_productName___ViewControllable: ViewControllable {}
 
diff --git a/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-AppKit-Flow.txt b/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-AppKit-Flow.txt
@@ -11,7 +11,7 @@ internal protocol ___VARIABLE_productName___ContextInterface: Context {}
 /// Defines the interface between the `Flow` and its view in order to present or contain other views. This protocol
 /// typically inherits protocols that include pre-defined behavior for convenience. Add additional methods for custom
 /// presentation or containment as desired.
-/// @mockable
+/// @mockable(superclass: UIViewController)
 @MainActor
 internal protocol ___VARIABLE_productName___ViewControllable: ViewControllable {}
 
diff --git a/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-AppKitCreatedForPluginList-Flow.txt b/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-AppKitCreatedForPluginList-Flow.txt
@@ -11,7 +11,7 @@ internal protocol ___VARIABLE_productName___ContextInterface: Context {}
 /// Defines the interface between the `Flow` and its view in order to present or contain other views. This protocol
 /// typically inherits protocols that include pre-defined behavior for convenience. Add additional methods for custom
 /// presentation or containment as desired.
-/// @mockable
+/// @mockable(superclass: UIViewController)
 @MainActor
 internal protocol ___VARIABLE_productName___ViewControllable: ViewControllable {}
 
diff --git a/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-AppKitSwiftUI-Flow.txt b/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-AppKitSwiftUI-Flow.txt
@@ -11,7 +11,7 @@ internal protocol ___VARIABLE_productName___ContextInterface: Context {}
 /// Defines the interface between the `Flow` and its view in order to present or contain other views. This protocol
 /// typically inherits protocols that include pre-defined behavior for convenience. Add additional methods for custom
 /// presentation or containment as desired.
-/// @mockable
+/// @mockable(superclass: UIViewController)
 @MainActor
 internal protocol ___VARIABLE_productName___ViewControllable: ViewControllable {}
 
diff --git a/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-AppKitSwiftUICreatedForPluginList-Flow.txt b/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-AppKitSwiftUICreatedForPluginList-Flow.txt
@@ -11,7 +11,7 @@ internal protocol ___VARIABLE_productName___ContextInterface: Context {}
 /// Defines the interface between the `Flow` and its view in order to present or contain other views. This protocol
 /// typically inherits protocols that include pre-defined behavior for convenience. Add additional methods for custom
 /// presentation or containment as desired.
-/// @mockable
+/// @mockable(superclass: UIViewController)
 @MainActor
 internal protocol ___VARIABLE_productName___ViewControllable: ViewControllable {}
 
diff --git a/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-Custom-Flow.txt b/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-Custom-Flow.txt
@@ -11,7 +11,7 @@ internal protocol ___VARIABLE_productName___ContextInterface: Context {}
 /// Defines the interface between the `Flow` and its view in order to present or contain other views. This protocol
 /// typically inherits protocols that include pre-defined behavior for convenience. Add additional methods for custom
 /// presentation or containment as desired.
-/// @mockable
+/// @mockable(superclass: UIViewController)
 @MainActor
 internal protocol ___VARIABLE_productName___ViewControllable: ViewControllable {}
 
diff --git a/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-CustomCreatedForPluginList-Flow.txt b/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-CustomCreatedForPluginList-Flow.txt
@@ -11,7 +11,7 @@ internal protocol ___VARIABLE_productName___ContextInterface: Context {}
 /// Defines the interface between the `Flow` and its view in order to present or contain other views. This protocol
 /// typically inherits protocols that include pre-defined behavior for convenience. Add additional methods for custom
 /// presentation or containment as desired.
-/// @mockable
+/// @mockable(superclass: UIViewController)
 @MainActor
 internal protocol ___VARIABLE_productName___ViewControllable: ViewControllable {}
 
diff --git a/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-UIKit-Flow.txt b/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-UIKit-Flow.txt
@@ -11,7 +11,7 @@ internal protocol ___VARIABLE_productName___ContextInterface: Context {}
 /// Defines the interface between the `Flow` and its view in order to present or contain other views. This protocol
 /// typically inherits protocols that include pre-defined behavior for convenience. Add additional methods for custom
 /// presentation or containment as desired.
-/// @mockable
+/// @mockable(superclass: UIViewController)
 @MainActor
 internal protocol ___VARIABLE_productName___ViewControllable: ViewControllable {}
 
diff --git a/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-UIKitCreatedForPluginList-Flow.txt b/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-UIKitCreatedForPluginList-Flow.txt
@@ -11,7 +11,7 @@ internal protocol ___VARIABLE_productName___ContextInterface: Context {}
 /// Defines the interface between the `Flow` and its view in order to present or contain other views. This protocol
 /// typically inherits protocols that include pre-defined behavior for convenience. Add additional methods for custom
 /// presentation or containment as desired.
-/// @mockable
+/// @mockable(superclass: UIViewController)
 @MainActor
 internal protocol ___VARIABLE_productName___ViewControllable: ViewControllable {}
 
diff --git a/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-UIKitSwiftUI-Flow.txt b/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-UIKitSwiftUI-Flow.txt
@@ -11,7 +11,7 @@ internal protocol ___VARIABLE_productName___ContextInterface: Context {}
 /// Defines the interface between the `Flow` and its view in order to present or contain other views. This protocol
 /// typically inherits protocols that include pre-defined behavior for convenience. Add additional methods for custom
 /// presentation or containment as desired.
-/// @mockable
+/// @mockable(superclass: UIViewController)
 @MainActor
 internal protocol ___VARIABLE_productName___ViewControllable: ViewControllable {}
 
diff --git a/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-UIKitSwiftUICreatedForPluginList-Flow.txt b/Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-UIKitSwiftUICreatedForPluginList-Flow.txt
@@ -11,7 +11,7 @@ internal protocol ___VARIABLE_productName___ContextInterface: Context {}
 /// Defines the interface between the `Flow` and its view in order to present or contain other views. This protocol
 /// typically inherits protocols that include pre-defined behavior for convenience. Add additional methods for custom
 /// presentation or containment as desired.
-/// @mockable
+/// @mockable(superclass: UIViewController)
 @MainActor
 internal protocol ___VARIABLE_productName___ViewControllable: ViewControllable {}
 
