(require 'ein-connect)
(require 'ein-testing-notebook)

(defmacro eintest:with-connected-buffer (&rest body)
  (declare (indent 0))
  `(let* ((notebook-buffer (ein:testing-notebook-make-empty))
          (notebook (buffer-local-value 'ein:notebook notebook-buffer)))
     (with-temp-buffer
       (erase-buffer)
       (ein:connect-buffer-to-notebook notebook)
       ,@body)))

(ert-deftest ein:get-url-or-port--connect ()
  (eintest:with-connected-buffer
    (should (equal (ein:get-url-or-port)
                   (ein:$notebook-url-or-port notebook)))))

(ert-deftest ein:get-notebook--connect ()
  (eintest:with-connected-buffer
    (should (eq (ein:get-notebook) notebook))))

(ert-deftest ein:get-kernel--connect ()
  (eintest:with-connected-buffer
    (should (eq (ein:get-kernel)
                (ein:$notebook-kernel notebook)))))

(ert-deftest ein:get-cell-at-point--connect ()
  "`ein:get-cell-at-point' is in empty context in connected buffer."
  (eintest:with-connected-buffer
    (should-not (ein:get-cell-at-point))))

(ert-deftest ein:get-traceback-data--connect ()
  (eintest:with-connected-buffer
    ;; FIXME: write test with non-empty TB
    (should-not (ein:get-traceback-data))))

(ert-deftest ein:connect-mode-revert-buffer-resistance ()
  (let ((temp-file (make-temp-file "ein")))
    (unwind-protect
        (with-temp-buffer
          (setq buffer-file-name temp-file)
          (ein:connect-mode 1)
          (setq ein:%connect% 'very-important-value)
          (revert-buffer t t nil)
          (should ein:connect-mode)
          (should (eq ein:%connect% 'very-important-value)))
      (delete-file temp-file))))