#lang scheme/base

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

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

(define-enum vehicle
  (train boat plane))

(define-enum prefixed
  ([a "item a"]
   [b 'B "item b"])
  #:prefix prefix:)

; Tests ------------------------------------------

(define enum-tests
  (test-suite "enum.ss"
    
    (test-case "vehicle"
      (check-pred enum? vehicle)
      (check-equal? (enum-values vehicle) '(train boat plane))
      (check-equal? (enum-pretty-values vehicle) '("train" "boat" "plane"))
      (check-equal? (enum->string vehicle) "train, boat, plane")
      (check-equal? (enum->pretty-string vehicle) "train, boat, plane")
      (check-equal? train 'train)
      (check-equal? boat 'boat)
      (check-equal? plane 'plane)
      (check-true (vehicle? 'train))
      (check-true (vehicle? 'boat))
      (check-true (vehicle? 'plane))
      (check-false (vehicle? 'lemon)))
    
    (test-case "prefixed"
      (check-pred enum? prefixed)
      (check-equal? (enum-values prefixed) '(a B))
      (check-equal? (enum-pretty-values prefixed) '("item a" "item b"))
      (check-equal? (enum->string prefixed) "a, B")
      (check-equal? (enum->pretty-string prefixed) "item a, item b")
      (check-equal? prefix:a 'a)
      (check-equal? prefix:b 'B)
      (check-true (prefixed? 'a))
      (check-true (prefixed? 'B))
      (check-false (prefixed? 'b)))
    
    (test-case "enum->[pretty-]string with separator specified"
      (check-equal? (enum->string vehicle ":") "train:boat:plane")
      (check-equal? (enum->pretty-string vehicle ":") "train:boat:plane"))))

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

(provide enum-tests)