import Spectre
import Stencil


fileprivate class CustomNode : NodeType {
  func render(_ context:Context) throws -> String {
    return "Hello World"
  }
}


fileprivate struct Article {
  let title: String
  let author: String
}


func testStencil() {
  describe("Stencil") {
    $0.it("can render the README example") {

      let templateString = "There are {{ articles.count }} articles.\n" +
        "\n" +
        "{% for article in articles %}" +
        "    - {{ article.title }} by {{ article.author }}.\n" +
        "{% endfor %}\n"

      let context = Context(dictionary: [
        "articles": [
          Article(title: "Migrating from OCUnit to XCTest", author: "Kyle Fuller"),
          Article(title: "Memory Management with ARC", author: "Kyle Fuller"),
        ]
      ])

      let template = Template(templateString: templateString)
      let result = try template.render(context)

      let fixture = "There are 2 articles.\n" +
        "\n" +
        "    - Migrating from OCUnit to XCTest by Kyle Fuller.\n" +
        "    - Memory Management with ARC by Kyle Fuller.\n" +
        "\n"

      try expect(result) == fixture
    }

    $0.it("can render a custom template tag") {
      let templateString = "{% custom %}"
      let template = Template(templateString: templateString)

      let namespace = Namespace()
      namespace.registerTag("custom") { parser, token in
        return CustomNode()
      }

      let result = try template.render(Context(namespace: namespace))
      try expect(result) == "Hello World"
    }

    $0.it("can render a simple custom tag") {
      let templateString = "{% custom %}"
      let template = Template(templateString: templateString)

      let namespace = Namespace()
      namespace.registerSimpleTag("custom") { context in
        return "Hello World"
      }

      try expect(try template.render(Context(namespace: namespace))) == "Hello World"
    }
  }
}