#lang scheme/base

(require "debug.ss"
         "test-base.ss")

; Helpers ----------------------------------------

(define-syntax capture-output
  (syntax-rules ()
    [(_ expr ...)
     (let ([out (open-output-string)])
       (parameterize ([current-output-port out])
         expr ... (get-output-string out)))]))

(define-syntax discard-output
  (syntax-rules ()
    [(_ expr ...)
     (let ([out (open-output-string)])
       (parameterize ([current-output-port out])
         expr ...))]))

; Test suite -----------------------------------

(define debug-tests
  (test-suite "debug.ss"
    
    (test-equal? "debug : passes value transparently"
      (discard-output (debug "Message" (+ 1 2 3)))
      6)
    
    (test-equal? "debug : prints value"
      (capture-output (debug "Message" (+ 1 2 3)))
      "Message:\n  6\n")
    
    (test-equal? "debug* : passes value transparently"
      (discard-output (debug* "Message" + 1 2 3))
      6)
    
    (test-equal? "debug* : prints value"
      (capture-output (debug* "Message" + 1 2 3))
      "Message:\n  6\n")
    
    (test-equal? "debug-location : prints value"
      (capture-output (debug-location))
      "Reached debug-test.ss:44:22\n")
    
    (test-equal? "debug-location : prints value and expression"
      (capture-output (debug-location (+ 1 2 3)))
      "Reached debug-test.ss:48:22:\n  6\n")
    
    (test-equal? "debug-enabled?"
      (parameterize ([debug-enabled? #f])
       (capture-output (debug "Message" (+ 1 2 3))))
      "")
    
    (test-equal? "define-debug"
      (capture-output (define-debug a 2) (void))
      "a:\n  2\n")
    
    (test-equal? "define-values-debug"
      (capture-output (define-values-debug (a b) (values 1 2)) (void))
      "(a b):\n  (1 2)\n")
    
    (test-equal? "let-debug"
      (capture-output (let-debug ([a 1] [b 2])
                                 (+ a b)))
      "a:\n  1\nb:\n  2\n")
    
    (test-equal? "let*-debug"
      (capture-output (let*-debug ([a 1] [b (+ a 2)])
                                  (+ a b)))
      "a:\n  1\nb:\n  3\n")
    
    (test-equal? "letrec-debug"
      (capture-output (letrec-debug ([a 1] [b 2] [c 3])
                                    (+ a b c)))
      "a:\n  1\nb:\n  2\nc:\n  3\n")
    
    (test-equal? "let-values-debug"
      (capture-output (let-values-debug ([(a b c) (values 1 2 3)]
                                         [(d e f) (values (+ 4 5) (+ 6 7) (+ 8 9))])
                                        (+ a b c d e f)))
      "(a b c):\n  (1 2 3)\n(d e f):\n  (9 13 17)\n")
    
    (test-equal? "let*-values-debug"
      (capture-output (let*-values-debug ([(a b) (values 1 2)]
                                          [(c d) (values (+ a b) (- a b))])
                                        (+ a b c d)))
      "(a b):\n  (1 2)\n(c d):\n  (3 -1)\n")
    
    (test-equal? "letrec-values-debug"
      (capture-output (letrec-values-debug ([(a b) (values 1 2)]
                                          [(c d) (values (+ a b) (- a b))])
                                        (+ a b c d)))
      "(a b):\n  (1 2)\n(c d):\n  (3 -1)\n")
    
    (test-case "with-pretty-indent"
      (parameterize ([pretty-print-columns 6])
        (check-equal? (pretty-format (list 1 2 3 4 5))
                      "(1\n 2\n 3\n 4\n 5)")
        (check-equal? (with-pretty-indent "=="
                        (pretty-format (list 1 2 3 4 5)))
                      "==(1\n== 2\n== 3\n== 4\n== 5)")
        (check-equal? (with-pretty-indent "==="
                        (pretty-format (list 1 2 3 4 5)))
                      "===(1\n=== 2\n=== 3\n=== 4\n=== 5)")))
    
    (test-case "exn-context"
      (let ([ctxt (exn-context (make-exn "Test" (current-continuation-marks)))])
        (check-pred list? ctxt)
        (check-true (andmap symbol? ctxt))))))

; Provide statements -----------------------------

(provide debug-tests)