Add login test

Author: kaiwk

.gitignore

@@ -11,7 +11,7 @@
 *.~undo-tree
 
 # ignore these dirty local tests.
+test-account
 test.el
-test-leetcode-account.el
-problem.graphql
+*.graphql
 *.http

leetcode-tests.el

@@ -169,8 +169,9 @@ When ACCOUNT or PASSWORD is empty string it will show a prompt."
             ("login"               . ("" :data ,account))
             ("password"            . ("" :data ,password)))
    :success
-   (cl-function (lambda (&key data &allow-other-keys)
-                  (leetcode-global-loading-mode -1)))
+   (cl-function
+    (lambda (&key data &allow-other-keys)
+      (leetcode-global-loading-mode -1)))
    :error
    (cl-function
     (lambda (&rest args &key error-thrown &allow-other-keys)
@@ -187,45 +188,44 @@ When ACCOUNT or PASSWORD is empty string it will show a prompt."
           (request-cookie-alist
            (concat "." leetcode--domain) "/" t)))))
 
-(defun leetcode--set-user-and-problems (response)
+(defun leetcode--set-user-and-problems (res)
   "Set `leetcode--user' and `leetcode--problems'.
-If user isn't login, only `leetcode--problems' will be set.
-RESPONSE is a request.el response."
-  (let ((data (request-response-data response)))
-    ;; user
-    (setq leetcode--user (plist-put leetcode--user :username (alist-get 'user_name data)))
-    (setq leetcode--user (plist-put leetcode--user :solved (alist-get 'num_solved data)))
-    (setq leetcode--user (plist-put leetcode--user :easy (alist-get 'ac_easy data)))
-    (setq leetcode--user (plist-put leetcode--user :medium (alist-get 'ac_medium data)))
-    (setq leetcode--user (plist-put leetcode--user :hard (alist-get 'ac_hard data)))
-    ;; problem list
-    (setq leetcode--problems (plist-put leetcode--problems :num (alist-get 'num_total data)))
-    (setq leetcode--problems (plist-put leetcode--problems :tag "all"))
-    (setq leetcode--problems
-          (plist-put leetcode--problems :problems
-                     (let ((raw-vec (alist-get 'stat_status_pairs data))
-                           (len (plist-get leetcode--problems :num))
-                           problems)
-                       (dolist (i (number-sequence 0 (1- len)))
-                         (let* ((cur (aref raw-vec i))
-                                (stat (alist-get 'stat cur))
-                                (status (alist-get 'status cur))
-                                (difficulty (alist-get 'level (alist-get 'difficulty cur)))
-                                (paid-only (eq (alist-get 'paid_only cur) t))
-                                (question-id (alist-get 'question_id stat))
-                                (total-submitted (alist-get 'total_submitted stat))
-                                (total-acs (alist-get 'total_acs stat)))
-                           (push
-                            (list
-                             :status status
-                             :id question-id
-                             :pos (- len i)
-                             :title (alist-get 'question__title stat)
-                             :acceptance (format "%.1f%%" (* 100 (/ (float total-acs) total-submitted)))
-                             :difficulty difficulty
-                             :paid-only paid-only)
-                            problems)))
-                       problems)))))
+If user isn't login, only `leetcode--problems' will be set. RES
+is an alist comes from `leetcode--api-all-problems'."
+  ;; user
+  (setq leetcode--user (plist-put leetcode--user :username (alist-get 'user_name  res)))
+  (setq leetcode--user (plist-put leetcode--user :solved   (alist-get 'num_solved res)))
+  (setq leetcode--user (plist-put leetcode--user :easy     (alist-get 'ac_easy    res)))
+  (setq leetcode--user (plist-put leetcode--user :medium   (alist-get 'ac_medium  res)))
+  (setq leetcode--user (plist-put leetcode--user :hard     (alist-get 'ac_hard    res)))
+  ;; problem list
+  (setq leetcode--problems (plist-put leetcode--problems :num (alist-get 'num_total res)))
+  (setq leetcode--problems (plist-put leetcode--problems :tag "all"))
+  (setq leetcode--problems
+        (plist-put leetcode--problems :problems
+                   (let ((raw-vec (alist-get 'stat_status_pairs res))
+                         (len (plist-get leetcode--problems :num))
+                         problems)
+                     (dolist (i (number-sequence 0 (1- len)))
+                       (let* ((cur (aref raw-vec i))
+                              (status          (alist-get 'status cur))
+                              (stat            (alist-get 'stat cur))
+                              (question-id     (alist-get 'question_id stat))
+                              (total-submitted (alist-get 'total_submitted stat))
+                              (total-acs       (alist-get 'total_acs stat))
+                              (difficulty      (alist-get 'level (alist-get 'difficulty cur)))
+                              (paid-only (eq (alist-get 'paid_only cur) t)))
+                         (push
+                          (list
+                           :status status
+                           :id question-id
+                           :pos (- len i)
+                           :title (alist-get 'question__title stat)
+                           :acceptance (format "%.1f%%" (* 100 (/ (float total-acs) total-submitted)))
+                           :difficulty difficulty
+                           :paid-only paid-only)
+                          problems)))
+                     problems))))
 
 (defun leetcode--slugify-title (title)
   "Make TITLE a slug title.
@@ -305,7 +305,7 @@ row."
 (defun leetcode--problems-rows ()
   "Generate tabulated list rows from `leetcode--problems'.
 Return a list of rows, each row is a vector:
- ([<checkmark> <position> <acceptance> <difficulty>] ...)"
+\([<checkmark> <position> <title> <acceptance> <difficulty>] ...)"
   (let ((problems (reverse (plist-get leetcode--problems :problems)))
         (easy-tag "easy")
         (medium-tag "medium")
@@ -355,7 +355,8 @@ Return a list of rows, each row is a vector:
                 ,(leetcode--referer leetcode--url-login))
      :parser 'json-read)
     (deferred:nextc it
-      (lambda (response) (leetcode--set-user-and-problems response)))
+      (lambda (response)
+        (leetcode--set-user-and-problems (request-response-data response))))
     (deferred:nextc it
       (lambda ()
         (let* ((header-names '(" " "#" "Problem" "Acceptance" "Difficulty"))
@@ -379,18 +380,17 @@ Return a list of rows, each row is a vector:
   (interactive)
   (if (get-buffer leetcode--buffer-name)
       (switch-to-buffer leetcode--buffer-name)
-    (if (leetcode--login-p)
-        (deferred:$
+    (deferred:$
+      (if (leetcode--login-p)
+          (deferred:next
+            (lambda ()
+              (message "User have been login in.")))
+        (leetcode--login leetcode-account leetcode-password))
+      (deferred:nextc it
+        (lambda ()
           (deferred:nextc (leetcode-problems-refresh)
             (lambda ()
-              (switch-to-buffer leetcode--buffer-name))))
-      (deferred:$
-        (leetcode--login leetcode-account leetcode-password)
-        (deferred:nextc it
-          (lambda ()
-            (deferred:nextc (leetcode-problems-refresh)
-              (lambda ()
-                (switch-to-buffer leetcode--buffer-name)))))))))
+              (switch-to-buffer leetcode--buffer-name))))))))
 
 (defun leetcode--buffer-content (buf)
   "Get content without text properties of BUF."

leetcode.el

@@ -0,0 +1,79 @@
+;;; leetcode-tests.el --- Test for leetcode.el       -*- lexical-binding: t; -*-
+
+;; Copyright (C) 2019  Wang Kai
+
+;; Author: Wang Kai <[email protected]>
+;; Keywords: extensions
+
+;; This program is free software; you can redistribute it and/or modify
+;; it under the terms of the GNU General Public License as published by
+;; the Free Software Foundation, either version 3 of the License, or
+;; (at your option) any later version.
+
+;; This program is distributed in the hope that it will be useful,
+;; but WITHOUT ANY WARRANTY; without even the implied warranty of
+;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+;; GNU General Public License for more details.
+
+;; You should have received a copy of the GNU General Public License
+;; along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+;;; Commentary:
+;; Test for leetcode.el
+;;; Code:
+
+(require 'leetcode)
+
+(ert-deftest leetcode-test-slugify-title ()
+  (should (equal (leetcode--slugify-title "Two Sum") "two-sum"))
+  (should (equal (leetcode--slugify-title "Reverse Nodes in k-Group") "reverse-nodes-in-k-group"))
+  (should (equal (leetcode--slugify-title "Pow(x, n)") "powx-n"))
+  (should (equal (leetcode--slugify-title "Implement strStr()") "implement-strstr"))
+  (should (equal (leetcode--slugify-title "String to Integer (atoi)") "string-to-integer-atoi")))
+
+(ert-deftest leetcode-test-fetch-csrftoken ()
+  (should (equal (leetcode--csrf-token) (leetcode--csrf-token))))
+
+(ert-deftest leetcode-test-login ()
+  (let ((leetcode-account
+         (with-temp-buffer
+           (insert-file-contents "test-account")
+           (car (split-string (string-trim (buffer-string)) "\n"))))
+        (leetcode-password
+         (with-temp-buffer
+           (insert-file-contents "test-account")
+           (cadr (split-string (string-trim (buffer-string)) "\n")))))
+    (setq leetcode--user nil)
+    (setq leetcode--problems nil)
+    (should (equal (leetcode--login-p) nil))
+    (deferred:sync! (leetcode--login leetcode-account leetcode-password))
+    (deferred:sync! (leetcode-problems-refresh))
+    (should (equal (not (leetcode--login-p)) nil))
+    (setq leetcode--user nil)
+    (setq leetcode--problems nil)
+    (kill-buffer (get-buffer leetcode--buffer-name))))
+
+(ert-deftest leetcode-test-make-tabulated-headers ()
+  (should (equal (leetcode--make-tabulated-headers
+                  '(" " "#" "Problem" "Acceptance" "Difficulty")
+                  '(["✓" "1" "   Two Sum" "43.3%" "easy"]
+                    ["✓" "2" "   Add Two Numbers" "30.9%" "medium"]
+                    ["✓" "3"    "Longest Substring Without Repeating Characters" "28.2%" "medium"]
+                    ["✓" "4"    "Median of Two Sorted Arrays" "26.0%" "hard"]
+                    ["✓" "5"    "Longest Palindromic Substring" "27.0%" "medium"]
+                    [" " "6"    "ZigZag Conversion" "31.3%" "medium"]))
+                 [(" " 1 nil) ("#" 1 nil) ("Problem" 46 nil) ("Acceptance" 10 nil) ("Difficulty" 10 nil)])))
+
+(ert-deftest leetcode-test-get-code-buffer-name ()
+  (let ((leetcode-prefer-language "c"))
+    (should (equal (leetcode--get-code-buffer-name "String to Integer (atoi)")
+                   "string-to-integer-atoi.c")))
+  (let ((leetcode-prefer-language "java"))
+    (should (equal (leetcode--get-code-buffer-name "Pow(x, n)")
+                   "powx-n.java")))
+  (let ((leetcode-prefer-language "python3"))
+    (should (equal (leetcode--get-code-buffer-name "Reverse Nodes in k-Group")
+                   "reverse-nodes-in-k-group.py"))))
+
+(provide 'leetcode-tests)
+;;; leetcode-tests.el ends here