diff --git a/StencilTests/TemplateTests.swift b/StencilTests/TemplateTests.swift index 93aeaba..dce68e3 100644 --- a/StencilTests/TemplateTests.swift +++ b/StencilTests/TemplateTests.swift @@ -9,4 +9,27 @@ class TemplateTests: XCTestCase { let result = try? template.render(context) XCTAssertEqual(result, "Hello World") } + + func testTemplateNamedInBundle() { + let testBundle = NSBundle(forClass: self.dynamicType) + let template = try? Template(named: "test.html", inBundle: testBundle) + let context = Context(dictionary: [ "target": "Kyle" ]) + + XCTAssertNotNil(template) + let result = try? template!.render(context) + XCTAssertEqual(result, "Hello Kyle!") + } + + + func testTemplateWithNSURL() { + let testBundle = NSBundle(forClass: self.dynamicType) + let URL = testBundle.URLForResource("test", withExtension: "html") + XCTAssertNotNil(URL) + let template = try? Template(URL: URL!) + let context = Context(dictionary: [ "target": "Kyle" ]) + + XCTAssertNotNil(template) + let result = try? template!.render(context) + XCTAssertEqual(result, "Hello Kyle!") + } }