1 files changed, 46 insertions, 0 deletions
diff --git a/tests/test-org-drill-resume-nil-session.el b/tests/test-org-drill-resume-nil-session.el
index fd64faf..a2b2e7c 100644
--- a/tests/test-org-drill-resume-nil-session.el
+++ b/tests/test-org-drill-resume-nil-session.el
@@ -34,6 +34,52 @@ not an obscure eieio type error."
   (let ((org-drill-last-session nil))
     (should-error (org-drill-resume) :type 'user-error)))
 
+;;;; org-drill-resume — happy paths
+
+(ert-deftest test-org-drill-resume-with-pending-entries-resumes ()
+  "When the prior session has pending entries, resume calls org-drill with resume-p=t."
+  (let* ((session (org-drill-session))
+         (org-drill-last-session session)
+         (resume-args nil))
+    (cl-letf (((symbol-function 'org-drill-entries-pending-p) (lambda (_) t))
+              ((symbol-function 'org-drill)
+               (lambda (&rest args) (setq resume-args args))))
+      (org-drill-resume))
+    (should resume-args)
+    (should (nth 2 resume-args))))
+
+(ert-deftest test-org-drill-resume-finished-with-no-pending-says-finished ()
+  "When the session is done with no remaining pending count, prints the
+'finished' message."
+  (let* ((session (org-drill-session))
+         (org-drill-last-session session)
+         (messages nil))
+    (cl-letf (((symbol-function 'message)
+               (lambda (fmt &rest args)
+                 (when fmt (push (apply #'format fmt args) messages)))))
+      (org-drill-resume))
+    (should (cl-some (lambda (m) (string-match-p "finished" m)) messages))))
+
+(ert-deftest test-org-drill-resume-finished-with-y-starts-new ()
+  "When session is done but pending count is positive and user answers y,
+org-drill-again runs."
+  (let* ((session (org-drill-session))
+         (org-drill-last-session session)
+         (m (let ((mk (make-marker))) (set-marker mk 1) mk))
+         (again-called nil))
+    ;; Done-entries non-empty + new-entries non-empty + done-entries reaches limit
+    ;; tricky to set up cleanly.  Easier: stub the predicates.
+    (cl-letf (((symbol-function 'org-drill-entries-pending-p)
+               (lambda (_) nil))
+              ((symbol-function 'org-drill-pending-entry-count)
+               (lambda (_) 5))
+              ((symbol-function 'y-or-n-p) (lambda (&rest _) t))
+              ((symbol-function 'org-drill-again)
+               (lambda (&rest _) (setq again-called t)))
+              ((symbol-function 'message) #'ignore))
+      (org-drill-resume))
+    (should again-called)))
+
 (provide 'test-org-drill-resume-nil-session)
 
 ;;; test-org-drill-resume-nil-session.el ends here