Skip to content

Commit ae9a570

Browse files
Fix imports(with:) when UIFramework nil (#830)
* Fix imports(with:) when UIFramework nil * Remove variable
1 parent a2ea948 commit ae9a570

File tree

2 files changed

+8
-21
lines changed

2 files changed

+8
-21
lines changed

Sources/NodesGenerator/StencilTemplate.swift

+4-4
Original file line numberDiff line numberDiff line change
@@ -270,19 +270,19 @@ public enum StencilTemplate: CustomStringConvertible, Equatable, Sendable {
270270
case .state:
271271
config.baseImports
272272
case .viewController:
273-
config.baseImports
273+
uiFramework == nil ? [] : config.baseImports
274274
.union(["Nodes"])
275275
.union(config.reactiveImports)
276276
.union(config.viewControllerImports)
277277
.union(uiFramework.flatMap { [$0.import] } ?? [])
278278
case .viewControllerTests:
279-
config.baseTestImports
279+
uiFramework == nil ? [] : config.baseTestImports
280280
.union(config.reactiveImports)
281281
case .viewState:
282-
config.baseImports
282+
uiFramework == nil ? [] : config.baseImports
283283
.union(["Nodes"])
284284
case .viewStateFactoryTests:
285-
config.baseTestImports
285+
uiFramework == nil ? [] : config.baseTestImports
286286
case .worker:
287287
config.baseImports
288288
.union(["Nodes"])

Tests/NodesGeneratorTests/StencilTemplateTests.swift

+4-17
Original file line numberDiff line numberDiff line change
@@ -445,26 +445,13 @@ final class StencilTemplateTests: XCTestCase, TestFactories {
445445
"<baseImport>"
446446
]
447447
case .viewController:
448-
expect(imports) == [
449-
"Nodes",
450-
"<baseImport>",
451-
"<reactiveImport>",
452-
"<viewControllerImport>"
453-
]
448+
expect(imports).to(beEmpty())
454449
case .viewControllerTests:
455-
expect(imports) == [
456-
"<baseTestImport>",
457-
"<reactiveImport>"
458-
]
450+
expect(imports).to(beEmpty())
459451
case .viewState:
460-
expect(imports) == [
461-
"Nodes",
462-
"<baseImport>"
463-
]
452+
expect(imports).to(beEmpty())
464453
case .viewStateFactoryTests:
465-
expect(imports) == [
466-
"<baseTestImport>"
467-
]
454+
expect(imports).to(beEmpty())
468455
case .worker:
469456
expect(imports) == [
470457
"Nodes",

0 commit comments

Comments
 (0)