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 = [
        "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 namespace = Namespace()
      namespace.registerTag("custom") { parser, token in
        return CustomNode()
      }

      let environment = Environment(namespace: namespace)
      let result = try environment.renderTemplate(string: "{% custom %}")

      try expect(result) == "Hello World"
    }

    $0.it("can render a simple custom tag") {
      let namespace = Namespace()
      namespace.registerSimpleTag("custom") { context in
        return "Hello World"
      }

      let environment = Environment(namespace: namespace)
      let result = try environment.renderTemplate(string: "{% custom %}")

      try expect(result) == "Hello World"
    }
  }
}