Check isViewInjected in imports(with:including:)

Tinder · Sep 19, 2024 · ffaf2c2 · ffaf2c2
1 parent 9917a79
commit ffaf2c2
Show file tree

Hide file tree

Showing 33 changed files with 169 additions and 26 deletions.
diff --git a/Sources/NodesGenerator/StencilTemplate.swift b/Sources/NodesGenerator/StencilTemplate.swift
@@ -226,7 +226,8 @@ public enum StencilTemplate: CustomStringConvertible, Equatable, Sendable {
 
     // swiftlint:disable:next function_body_length cyclomatic_complexity
     public func imports(with config: Config, including uiFramework: UIFramework? = nil) -> Set<String> {
-        switch self {
+        let isViewInjected: Bool = uiFramework == nil
+        return switch self {
         case .analytics:
             config.baseImports
         case .analyticsTests:
@@ -246,6 +247,7 @@ public enum StencilTemplate: CustomStringConvertible, Equatable, Sendable {
                 .union(config.reactiveImports)
         case .contextTests:
             config.baseTestImports
+                .union(isViewInjected ? [] : ["NodesTesting"])
         case .flow:
             config.baseImports
                 .union(["Nodes"])
@@ -270,7 +272,7 @@ public enum StencilTemplate: CustomStringConvertible, Equatable, Sendable {
         case .state:
             config.baseImports
         case .viewController:
-            uiFramework == nil ? [] : config.baseImports
+            isViewInjected ? [] : config.baseImports
                 .union(["Nodes"])
                 .union(config.reactiveImports)
                 .union(config.viewControllerImports)
@@ -279,10 +281,14 @@ public enum StencilTemplate: CustomStringConvertible, Equatable, Sendable {
             uiFramework == nil ? [] : config.baseTestImports
                 .union(config.reactiveImports)
         case .viewState:
-            uiFramework == nil ? [] : config.baseImports
-                .union(["Nodes"])
+            isViewInjected
+                ? []
+                : config.baseImports
+                    .union(["Nodes"])
         case .viewStateFactoryTests:
-            uiFramework == nil ? [] : config.baseTestImports
+            isViewInjected
+                ? []
+                : config.baseTestImports
         case .worker:
             config.baseImports
                 .union(["Nodes"])

diff --git a/Tests/NodesGeneratorTests/StencilTemplateTests.swift b/Tests/NodesGeneratorTests/StencilTemplateTests.swift
@@ -295,7 +295,8 @@ final class StencilTemplateTests: XCTestCase, TestFactories {
                     ]
                 case .contextTests:
                     expect(imports) == [
-                        "<baseTestImport>"
+                        "<baseTestImport>",
+                        "NodesTesting"
                     ]
                 case .flow:
                     expect(imports) == [
@@ -466,6 +467,102 @@ final class StencilTemplateTests: XCTestCase, TestFactories {
             }
         }
     }
+
+    // swiftlint:disable:next cyclomatic_complexity
+    func testImports_givenUIFrameworkNil() {
+        let config: Config = givenConfig()
+        for stencilTemplate in StencilTemplate.allCases {
+            let imports: Set<String> = stencilTemplate.imports(with: config, including: nil)
+            switch stencilTemplate {
+            case .analytics:
+                expect(imports) == [
+                    "<baseImport>"
+                ]
+            case .analyticsTests:
+                expect(imports) == [
+                    "<baseTestImport>"
+                ]
+            case .builder:
+                expect(imports) == [
+                    "Nodes",
+                    "<baseImport>",
+                    "<reactiveImport>",
+                    "<dependencyInjectionImport>",
+                    "<builderImport>"
+                ]
+            case .builderTests:
+                expect(imports) == [
+                    "NodesTesting",
+                    "<baseTestImport>"
+                ]
+            case .context:
+                expect(imports) == [
+                    "Nodes",
+                    "<baseImport>",
+                    "<reactiveImport>"
+                ]
+            case .contextTests:
+                expect(imports) == [
+                    "<baseTestImport>"
+                ]
+            case .flow:
+                expect(imports) == [
+                    "Nodes",
+                    "<baseImport>",
+                    "<flowImport>"
+                ]
+            case .flowTests:
+                expect(imports) == [
+                    "<baseTestImport>"
+                ]
+            case .plugin:
+                expect(imports) == [
+                    "Nodes",
+                    "<baseImport>",
+                    "<dependencyInjectionImport>"
+                ]
+            case .pluginTests:
+                expect(imports) == [
+                    "NodesTesting",
+                    "<baseTestImport>"
+                ]
+            case .pluginList:
+                expect(imports) == [
+                    "Nodes",
+                    "<baseImport>",
+                    "<dependencyInjectionImport>",
+                    "<pluginListImport>"
+                ]
+            case .pluginListTests:
+                expect(imports) == [
+                    "NodesTesting",
+                    "<baseTestImport>"
+                ]
+            case .state:
+                expect(imports) == [
+                    "<baseImport>"
+                ]
+            case .viewController:
+                expect(imports).to(beEmpty())
+            case .viewControllerTests:
+                expect(imports).to(beEmpty())
+            case .viewState:
+                expect(imports).to(beEmpty())
+            case .viewStateFactoryTests:
+                expect(imports).to(beEmpty())
+            case .worker:
+                expect(imports) == [
+                    "Nodes",
+                    "<baseImport>",
+                    "<reactiveImport>"
+                ]
+            case .workerTests:
+                expect(imports) == [
+                    "<baseTestImport>"
+                ]
+            }
+        }
+    }
 }
 
 extension StencilTemplate {

diff --git a/..._/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-AppKit-ContextTests.txt b/..._/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-AppKit-ContextTests.txt
@@ -3,6 +3,7 @@
 #warning("Manually move this test file to the corresponding test target then delete this warning.")
 
 import Nimble
+import NodesTesting
 import XCTest
 
 final class ___VARIABLE_productName___ContextTests: XCTestCase {

diff --git a/...atorTests/testGenerateWithIdentifier.Contents-AppKitCreatedForPluginList-ContextTests.txt b/...atorTests/testGenerateWithIdentifier.Contents-AppKitCreatedForPluginList-ContextTests.txt
@@ -3,6 +3,7 @@
 #warning("Manually move this test file to the corresponding test target then delete this warning.")
 
 import Nimble
+import NodesTesting
 import XCTest
 
 final class ___VARIABLE_productName___ContextTests: XCTestCase {

diff --git a/...TemplateGeneratorTests/testGenerateWithIdentifier.Contents-AppKitSwiftUI-ContextTests.txt b/...TemplateGeneratorTests/testGenerateWithIdentifier.Contents-AppKitSwiftUI-ContextTests.txt
@@ -3,6 +3,7 @@
 #warning("Manually move this test file to the corresponding test target then delete this warning.")
 
 import Nimble
+import NodesTesting
 import XCTest
 
 final class ___VARIABLE_productName___ContextTests: XCTestCase {

diff --git a/...ts/testGenerateWithIdentifier.Contents-AppKitSwiftUICreatedForPluginList-ContextTests.txt b/...ts/testGenerateWithIdentifier.Contents-AppKitSwiftUICreatedForPluginList-ContextTests.txt
@@ -3,6 +3,7 @@
 #warning("Manually move this test file to the corresponding test target then delete this warning.")
 
 import Nimble
+import NodesTesting
 import XCTest
 
 final class ___VARIABLE_productName___ContextTests: XCTestCase {

diff --git a/..._/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-Custom-ContextTests.txt b/..._/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-Custom-ContextTests.txt
@@ -3,6 +3,7 @@
 #warning("Manually move this test file to the corresponding test target then delete this warning.")
 
 import Nimble
+import NodesTesting
 import XCTest
 
 final class ___VARIABLE_productName___ContextTests: XCTestCase {

diff --git a/...atorTests/testGenerateWithIdentifier.Contents-CustomCreatedForPluginList-ContextTests.txt b/...atorTests/testGenerateWithIdentifier.Contents-CustomCreatedForPluginList-ContextTests.txt
@@ -3,6 +3,7 @@
 #warning("Manually move this test file to the corresponding test target then delete this warning.")
 
 import Nimble
+import NodesTesting
 import XCTest
 
 final class ___VARIABLE_productName___ContextTests: XCTestCase {

diff --git a/...__/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-UIKit-ContextTests.txt b/...__/XcodeTemplateGeneratorTests/testGenerateWithIdentifier.Contents-UIKit-ContextTests.txt
@@ -3,6 +3,7 @@
 #warning("Manually move this test file to the corresponding test target then delete this warning.")
 
 import Nimble
+import NodesTesting
 import XCTest
 
 final class ___VARIABLE_productName___ContextTests: XCTestCase {

diff --git a/...ratorTests/testGenerateWithIdentifier.Contents-UIKitCreatedForPluginList-ContextTests.txt b/...ratorTests/testGenerateWithIdentifier.Contents-UIKitCreatedForPluginList-ContextTests.txt
@@ -3,6 +3,7 @@
 #warning("Manually move this test file to the corresponding test target then delete this warning.")
 
 import Nimble
+import NodesTesting
 import XCTest
 
 final class ___VARIABLE_productName___ContextTests: XCTestCase {

diff --git a/...eTemplateGeneratorTests/testGenerateWithIdentifier.Contents-UIKitSwiftUI-ContextTests.txt b/...eTemplateGeneratorTests/testGenerateWithIdentifier.Contents-UIKitSwiftUI-ContextTests.txt
@@ -3,6 +3,7 @@
 #warning("Manually move this test file to the corresponding test target then delete this warning.")
 
 import Nimble
+import NodesTesting
 import XCTest
 
 final class ___VARIABLE_productName___ContextTests: XCTestCase {

diff --git a/...sts/testGenerateWithIdentifier.Contents-UIKitSwiftUICreatedForPluginList-ContextTests.txt b/...sts/testGenerateWithIdentifier.Contents-UIKitSwiftUICreatedForPluginList-ContextTests.txt
@@ -3,6 +3,7 @@
 #warning("Manually move this test file to the corresponding test target then delete this warning.")
 
 import Nimble
+import NodesTesting
 import XCTest
 
 final class ___VARIABLE_productName___ContextTests: XCTestCase {

diff --git a/...pshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-AppKit-ContextTests.txt b/...pshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-AppKit-ContextTests.txt
@@ -3,6 +3,7 @@
 #warning("Manually move this test file to the corresponding test target then delete this warning.")
 
 import Nimble
+import NodesTesting
 import XCTest
 
 final class ___VARIABLE_productName___ContextTests: XCTestCase {

diff --git a/...teGeneratorTests/testGenerateWithURL.Contents-AppKitCreatedForPluginList-ContextTests.txt b/...teGeneratorTests/testGenerateWithURL.Contents-AppKitCreatedForPluginList-ContextTests.txt
@@ -3,6 +3,7 @@
 #warning("Manually move this test file to the corresponding test target then delete this warning.")
 
 import Nimble
+import NodesTesting
 import XCTest
 
 final class ___VARIABLE_productName___ContextTests: XCTestCase {

diff --git a/..._/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-AppKitSwiftUI-ContextTests.txt b/..._/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-AppKitSwiftUI-ContextTests.txt
@@ -3,6 +3,7 @@
 #warning("Manually move this test file to the corresponding test target then delete this warning.")
 
 import Nimble
+import NodesTesting
 import XCTest
 
 final class ___VARIABLE_productName___ContextTests: XCTestCase {

diff --git a/...atorTests/testGenerateWithURL.Contents-AppKitSwiftUICreatedForPluginList-ContextTests.txt b/...atorTests/testGenerateWithURL.Contents-AppKitSwiftUICreatedForPluginList-ContextTests.txt
@@ -3,6 +3,7 @@
 #warning("Manually move this test file to the corresponding test target then delete this warning.")
 
 import Nimble
+import NodesTesting
 import XCTest
 
 final class ___VARIABLE_productName___ContextTests: XCTestCase {

diff --git a/...pshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-Custom-ContextTests.txt b/...pshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-Custom-ContextTests.txt
@@ -3,6 +3,7 @@
 #warning("Manually move this test file to the corresponding test target then delete this warning.")
 
 import Nimble
+import NodesTesting
 import XCTest
 
 final class ___VARIABLE_productName___ContextTests: XCTestCase {

diff --git a/...teGeneratorTests/testGenerateWithURL.Contents-CustomCreatedForPluginList-ContextTests.txt b/...teGeneratorTests/testGenerateWithURL.Contents-CustomCreatedForPluginList-ContextTests.txt
@@ -3,6 +3,7 @@
 #warning("Manually move this test file to the corresponding test target then delete this warning.")
 
 import Nimble
+import NodesTesting
 import XCTest
 
 final class ___VARIABLE_productName___ContextTests: XCTestCase {

diff --git a/...apshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-UIKit-ContextTests.txt b/...apshots__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-UIKit-ContextTests.txt
@@ -3,6 +3,7 @@
 #warning("Manually move this test file to the corresponding test target then delete this warning.")
 
 import Nimble
+import NodesTesting
 import XCTest
 
 final class ___VARIABLE_productName___ContextTests: XCTestCase {

diff --git a/...ateGeneratorTests/testGenerateWithURL.Contents-UIKitCreatedForPluginList-ContextTests.txt b/...ateGeneratorTests/testGenerateWithURL.Contents-UIKitCreatedForPluginList-ContextTests.txt
@@ -3,6 +3,7 @@
 #warning("Manually move this test file to the corresponding test target then delete this warning.")
 
 import Nimble
+import NodesTesting
 import XCTest
 
 final class ___VARIABLE_productName___ContextTests: XCTestCase {

diff --git a/...__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-UIKitSwiftUI-ContextTests.txt b/...__/XcodeTemplateGeneratorTests/testGenerateWithURL.Contents-UIKitSwiftUI-ContextTests.txt
@@ -3,6 +3,7 @@
 #warning("Manually move this test file to the corresponding test target then delete this warning.")
 
 import Nimble
+import NodesTesting
 import XCTest
 
 final class ___VARIABLE_productName___ContextTests: XCTestCase {

diff --git a/...ratorTests/testGenerateWithURL.Contents-UIKitSwiftUICreatedForPluginList-ContextTests.txt b/...ratorTests/testGenerateWithURL.Contents-UIKitSwiftUICreatedForPluginList-ContextTests.txt
@@ -3,6 +3,7 @@
 #warning("Manually move this test file to the corresponding test target then delete this warning.")
 
 import Nimble
+import NodesTesting
 import XCTest
 
 final class ___VARIABLE_productName___ContextTests: XCTestCase {

diff --git a/...PermutationTests/testNodeXcodeTemplatePermutation.AppKit-SwiftUI-CreatedForPluginList.txt b/...PermutationTests/testNodeXcodeTemplatePermutation.AppKit-SwiftUI-CreatedForPluginList.txt
@@ -29,8 +29,9 @@
       - "<baseImport>"
       - "<reactiveImport>"
       - "Nodes"
-    ▿ contextTestsImports: 1 element
+    ▿ contextTestsImports: 2 elements
       - "<baseTestImport>"
+      - "NodesTesting"
     ▿ dependencies: 1 element
       ▿ 2 key/value pairs
         ▿ (2 elements)

diff --git a/...shots__/XcodeTemplatePermutationTests/testNodeXcodeTemplatePermutation.AppKit-SwiftUI.txt b/...shots__/XcodeTemplatePermutationTests/testNodeXcodeTemplatePermutation.AppKit-SwiftUI.txt
@@ -29,8 +29,9 @@
       - "<baseImport>"
       - "<reactiveImport>"
       - "Nodes"
-    ▿ contextTestsImports: 1 element
+    ▿ contextTestsImports: 2 elements
       - "<baseTestImport>"
+      - "NodesTesting"
     ▿ dependencies: 1 element
       ▿ 2 key/value pairs
         ▿ (2 elements)

diff --git a/...s/__Snapshots__/XcodeTemplatePermutationTests/testNodeXcodeTemplatePermutation.AppKit.txt b/...s/__Snapshots__/XcodeTemplatePermutationTests/testNodeXcodeTemplatePermutation.AppKit.txt
@@ -29,8 +29,9 @@
       - "<baseImport>"
       - "<reactiveImport>"
       - "Nodes"
-    ▿ contextTestsImports: 1 element
+    ▿ contextTestsImports: 2 elements
       - "<baseTestImport>"
+      - "NodesTesting"
     ▿ dependencies: 1 element
       ▿ 2 key/value pairs
         ▿ (2 elements)

diff --git a/...eTemplatePermutationTests/testNodeXcodeTemplatePermutation.AppKitCreatedForPluginList.txt b/...eTemplatePermutationTests/testNodeXcodeTemplatePermutation.AppKitCreatedForPluginList.txt
@@ -29,8 +29,9 @@
       - "<baseImport>"
       - "<reactiveImport>"
       - "Nodes"
-    ▿ contextTestsImports: 1 element
+    ▿ contextTestsImports: 2 elements
       - "<baseTestImport>"
+      - "NodesTesting"
     ▿ dependencies: 1 element
       ▿ 2 key/value pairs
         ▿ (2 elements)

diff --git a/...s/__Snapshots__/XcodeTemplatePermutationTests/testNodeXcodeTemplatePermutation.Custom.txt b/...s/__Snapshots__/XcodeTemplatePermutationTests/testNodeXcodeTemplatePermutation.Custom.txt
@@ -29,8 +29,9 @@
       - "<baseImport>"
       - "<reactiveImport>"
       - "Nodes"
-    ▿ contextTestsImports: 1 element
+    ▿ contextTestsImports: 2 elements
       - "<baseTestImport>"
+      - "NodesTesting"
     ▿ dependencies: 1 element
       ▿ 2 key/value pairs
         ▿ (2 elements)

diff --git a/...eTemplatePermutationTests/testNodeXcodeTemplatePermutation.CustomCreatedForPluginList.txt b/...eTemplatePermutationTests/testNodeXcodeTemplatePermutation.CustomCreatedForPluginList.txt
@@ -29,8 +29,9 @@
       - "<baseImport>"
       - "<reactiveImport>"
       - "Nodes"
-    ▿ contextTestsImports: 1 element
+    ▿ contextTestsImports: 2 elements
       - "<baseTestImport>"
+      - "NodesTesting"
     ▿ dependencies: 1 element
       ▿ 2 key/value pairs
         ▿ (2 elements)

diff --git a/...ePermutationTests/testNodeXcodeTemplatePermutation.UIKit-SwiftUI-CreatedForPluginList.txt b/...ePermutationTests/testNodeXcodeTemplatePermutation.UIKit-SwiftUI-CreatedForPluginList.txt
@@ -29,8 +29,9 @@
       - "<baseImport>"
       - "<reactiveImport>"
       - "Nodes"
-    ▿ contextTestsImports: 1 element
+    ▿ contextTestsImports: 2 elements
       - "<baseTestImport>"
+      - "NodesTesting"
     ▿ dependencies: 1 element
       ▿ 2 key/value pairs
         ▿ (2 elements)