X-Git-Url: http://repo.macrolet.net/gitweb/?a=blobdiff_plain;f=src%2Ftest.lisp;h=aacb11e8236f61520c961a889526cfa46cc51362;hb=ea7021b8bc505e45fff8d226d8ab7fce6cae0f76;hp=99bc8bbab54ff9c0568eb3cbdabbb224ba47a79c;hpb=71b6fc1cd91159a5ff28432701dea1fe85831df1;p=fiveam.git

diff --git a/src/test.lisp b/src/test.lisp
index 99bc8bb..aacb11e 100644
--- a/src/test.lisp
+++ b/src/test.lisp
@@ -24,7 +24,7 @@
   "Create a test named NAME. If NAME is a list it must be of the
 form:
 
-  (name &key depends-on suite)
+  (name &key depends-on suite fixture compile-at profile)
 
 NAME is the symbol which names the test.
 
@@ -45,31 +45,46 @@ If DEPENDS-ON is a symbol it is interpreted as `(AND
 ,depends-on), this is accomadate the common case of one test
 depending on another.
 
-SUITE defaults to the current value of *SUITE*."
-  (destructuring-bind (name &key depends-on (suite nil suite-supplied-p))
-      (ensure-list name)
-    (let (description)
-      (setf description (if (stringp (car body))
-			    (pop body)
-			    ""))
-      `(progn
-	 (setf (get-test ',name) (make-instance 'test-case
-                                                :name ',name
-                                                :runtime-package ,*package*
-                                                :test-lambda
-                                                (lambda ()
-                                                  (funcall (let ((*package* ,*package*))
-                                                             (compile nil '(lambda () ,@body)))))
-                                                :description ,description
-                                                :depends-on ',depends-on))
-	 ,(if suite-supplied-p
-	      `(setf (gethash ',name (tests (get-test ',suite)))
-		     ',name)
-	      `(setf (gethash ',name (tests (or *suite* (get-test 'NIL))))
-		     ',name))
-         (when *run-test-when-defined*
-           (run! ',name))
-	 ',name))))
+FIXTURE specifies a fixtrue to wrap the body in.
+
+If PROFILE is T profiling information will be collected as well."
+  (let* ((tmp (gensym))
+         (suite-arg (getf (cdr (ensure-list name)) :suite tmp))
+         (suite-form (cond
+                       ((eq tmp suite-arg) '*suite*)
+                       (t                  `(get-test ',suite-arg)))))
+    (when (consp name)
+      (remf (cdr name) :suite))
+    (destructuring-bind (name &key depends-on (compile-at :run-time) fixture profile)
+        (ensure-list name)
+      (declare (type (member :run-time :definition-time) compile-at))
+      (let ((description (if (stringp (car body))
+                             (pop body)
+                             ""))
+            (effective-body (if fixture
+                                (destructuring-bind (name &rest args)
+                                    (ensure-list fixture)
+                                  `((with-fixture ,name ,args ,@body)))
+                                body)))
+        `(progn
+           (setf (get-test ',name) (make-instance 'test-case
+                                                  :name ',name
+                                                  :runtime-package ,*package*
+                                                  :test-lambda
+                                                  (lambda ()
+                                                    ,@ (ecase compile-at
+                                                         (:run-time `((funcall
+                                                                       (let ((*package* (find-package ',(package-name *package*))))
+                                                                         (compile nil '(lambda ()
+                                                                                        ,@effective-body))))))
+                                                         (:definition-time effective-body)))
+                                                  :description ,description
+                                                  :depends-on ',depends-on
+                                                  :collect-profiling-info ,profile))
+           (setf (gethash ',name (tests ,suite-form)) ',name)
+           (when *run-test-when-defined*
+             (run! ',name))
+           ',name)))))
 
 (defvar *run-test-when-defined* nil
   "When non-NIL tests are run as soon as they are defined.")