From 71611af1a5364f7a4d517b2e7339f0921b54fcbb Mon Sep 17 00:00:00 2001
From: Max <shadow.iqka@qq.com>
Date: Wed, 28 Sep 2022 17:32:35 +0800
Subject: [PATCH] [fix] test data

---
 widgets/chart/process_test.go | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/widgets/chart/process_test.go b/widgets/chart/process_test.go
index 63f40570..5e5597aa 100644
--- a/widgets/chart/process_test.go
+++ b/widgets/chart/process_test.go
@@ -88,4 +88,34 @@ func load(t *testing.T) {
 		t.Fatal(err)
 	}
 	q.Load(config.Conf)
+	clear(t)
+	testData(t)
+}
+
+func testData(t *testing.T) {
+	pet := gou.Select("pet")
+	err := pet.Insert(
+		[]string{"name", "type", "status", "mode", "stay", "cost", "doctor_id"},
+		[][]interface{}{
+			{"Cookie", "cat", "checked", "enabled", 200, 105, 1},
+			{"Baby", "dog", "checked", "enabled", 186, 24, 1},
+			{"Poo", "others", "checked", "enabled", 199, 66, 1},
+		},
+	)
+	if err != nil {
+		t.Fatal(err)
+	}
+}
+
+func clear(t *testing.T) {
+	for _, m := range gou.Models {
+		err := m.DropTable()
+		if err != nil {
+			t.Fatal(err)
+		}
+		err = m.Migrate(true)
+		if err != nil {
+			t.Fatal(err)
+		}
+	}
 }