NimYAML/test/yamlTestSuite.nim

#            NimYAML - YAML implementation in Nim
#        (c) Copyright 2016 Felix Krause
#
#    See the file "copying.txt", included in this
#    distribution, for details about the copyright.

import os, terminal, strutils
import testEventParser, common
import "../yaml"

const gitCmd =
    "git clone https://github.com/ingydotnet/yaml-dev-kit.git -b data"

proc echoError(msg: string) =
  styledWriteLine(stdout, fgRed, "[error] ", fgWhite, msg, resetStyle)

proc echoInfo(msg: string) =
  styledWriteLine(stdout, fgGreen, "[info] ", fgWhite, msg, resetStyle)

removeDir("yaml-dev-kit")
if execShellCmd(gitCmd) != 0:
  echoError("Could not check out yaml-dev-kit (no internet connection?)")
  quit(1)

var gotErrors = false

for kind, dirPath in walkDir("yaml-dev-kit"):
  block curTest:
    if kind == pcDir:
      if dirPath[^4..^1] in [".git", "name", "tags", "meta"]: continue
      var
        tagLib = initExtendedTagLibrary()
        parser = newYamlParser(tagLib)
        actualIn = newFileStream(dirPath / "in.yaml")
        actual = parser.parse(actualIn)
        expectedIn = newFileStream(dirPath / "test.event")
        expected = parseEventStream(expectedIn, tagLib)
      styledWriteLine(stdout, fgBlue, "[test] ", fgWhite, dirPath[^4..^1],
                      ": ", strip(readFile(dirPath / "===")), resetStyle)
      var i = 1
      try:
        while not actual.finished():
          let actualEvent = actual.next()
          if expected.finished():
            echoError("At token #" & $i & ": Expected stream end, got " &
                      $actualEvent.kind)
            gotErrors = true
            actualIn.close()
            expectedIn.close()
            break curTest
          let expectedEvent = expected.next()
          if expectedEvent != actualEvent:
            printDifference(expectedEvent, actualEvent)
            echoError("At token #" & $i &
                      ": Actual tokens do not match expected tokens")
            gotErrors = true
            actualIn.close()
            expectedIn.close()
            break curTest
          i.inc()
        if not expected.finished():
          echoError("Got fewer tokens than expected, first missing " &
                    "token: " & $expected.next().kind)
          gotErrors = true
      except:
        gotErrors = true
        let e = getCurrentException()
        if e.parent of YamlParserError:
          let pe = (ref YamlParserError)(e.parent)
          echo "line ", pe.line, ", column ", pe.column, ": ", pe.msg
          echo pe.lineContent
        else: echo e.msg
        echoError("Catched an exception at token #" & $i &
                  " test was not successful")
      actualIn.close()
      expectedIn.close()

if gotErrors:
  echoError("There were errors while running the tests")
  quit(1)
else:
  echoInfo("All tests were successful")
  quit(0)