From 7224b32c8f93f97ca257d47e3ff1d9ed0b0e6828 Mon Sep 17 00:00:00 2001
From: Pig Fang <g-plane@hotmail.com>
Date: Sat, 27 Apr 2019 23:20:42 +0800
Subject: [PATCH] Add more tests

---
 app/Models/User.php     |  2 +-
 tests/Api/tests/user.rs | 31 +++++++++++++++++++++++++++++++
 2 files changed, 32 insertions(+), 1 deletion(-)

diff --git a/app/Models/User.php b/app/Models/User.php
index e493cc07..1c57568b 100644
--- a/app/Models/User.php
+++ b/app/Models/User.php
@@ -175,7 +175,7 @@ class User extends Authenticatable implements JWTSubject
             $this->storageUsed = $result ?: 0;
         }
 
-        return $this->storageUsed;
+        return (int) $this->storageUsed;
     }
 
     /**
diff --git a/tests/Api/tests/user.rs b/tests/Api/tests/user.rs
index 3e5b66bb..2f3fb856 100644
--- a/tests/Api/tests/user.rs
+++ b/tests/Api/tests/user.rs
@@ -11,6 +11,20 @@ struct User {
     pub score: u32,
 }
 
+#[derive(Deserialize)]
+struct SignResult {
+    pub score: u32,
+    pub storage: Usage,
+    pub remaining_time: u8,
+}
+
+#[derive(Deserialize)]
+struct Usage {
+    pub used: u32,
+    pub total: u32,
+    pub percentage: f32,
+}
+
 #[test]
 fn fetch_user_info() {
     let client = reqwest::Client::new();
@@ -30,3 +44,20 @@ fn fetch_user_info() {
     assert_eq!(user.avatar, 0);
     assert_eq!(user.score, 1000);
 }
+
+#[test]
+fn sign() {
+    let client = reqwest::Client::new();
+    let body = client
+        .put("http://127.0.0.1:32123/api/user/sign")
+        .header("Authorization", login())
+        .send()
+        .unwrap()
+        .json::<JsonBody<SignResult>>()
+        .unwrap();
+    assert!(body.is_success());
+
+    let sign_result = body.data().unwrap();
+    assert!(sign_result.score > 1000);
+    assert!(sign_result.remaining_time > 0);
+}