Re-instate register functions

hummingbird-project · Jan 30, 2024 · c2c77d2 · c2c77d2
1 parent c9d5021
commit c2c77d2
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 29 deletions.
diff --git a/Sources/HummingbirdMustache/Library.swift b/Sources/HummingbirdMustache/Library.swift
@@ -44,6 +44,14 @@ public struct HBMustacheLibrary: Sendable {
         self.templates = try Self.loadTemplates(from: directory, withExtension: `extension`)
     }
 
+    /// Register template under name
+    /// - Parameters:
+    ///   - template: Template
+    ///   - name: Name of template
+    public mutating func register(_ template: HBMustacheTemplate, named name: String) {
+        self.templates[name] = template
+    }
+
     /// Register template under name
     /// - Parameters:
     ///   - mustache: Mustache text

diff --git a/Tests/HummingbirdMustacheTests/PartialTests.swift b/Tests/HummingbirdMustacheTests/PartialTests.swift
@@ -60,7 +60,9 @@ final class PartialTests: XCTestCase {
     {{/empty(.)}}
 
     """)
-    let library = HBMustacheLibrary(templates: ["base": template, "user": template2])
+    var library = HBMustacheLibrary()
+    library.register(template, named: "base")
+    library.register(template2, named: "user") // , withTemplate: String)// = HBMustacheLibrary(templates: ["base": template, "user": template2])
 
     let object: [String: Any] = ["names": ["john", "adam", "claire"]]
     XCTAssertEqual(library.render(object, withTemplate: "base"), """
@@ -98,33 +100,8 @@ final class PartialTests: XCTestCase {
 
   /// test inheritance
   func testInheritance() throws {
-    let library = try HBMustacheLibrary(templates: [
-      "header": .init(string: """
-        <head>
-        <title>{{$title}}Default title{{/title}}</title>
-        </head>
-
-        """),
-      "base": .init(string: """
-        <html>
-        {{$header}}{{/header}}
-        {{$content}}{{/content}}
-        </html>
-
-        """),
-      "mypage": .init(string: """
-        {{<base}}
-        {{$header}}
-        {{<header}}
-        {{$title}}My page title{{/title}}
-        {{/header}}
-        {{/header}}
-        {{$content}}<h1>Hello world</h1>{{/content}}
-        {{/base}}
-
-        """)
-      ])
-/*    try library.register(
+    var library = HBMustacheLibrary()
+    try library.register(
       """
       <head>
       <title>{{$title}}Default title{{/title}}</title>
@@ -156,7 +133,7 @@ final class PartialTests: XCTestCase {
 
       """,
       named: "mypage"
-    )*/
+    )
     XCTAssertEqual(library.render({}, withTemplate: "mypage")!, """
     <html>
     <head>