From fa199838a930b259ea9353e2af306fbe02c5a6c1 Mon Sep 17 00:00:00 2001
From: tengge <930372551@qq.com>
Date: Wed, 27 May 2020 21:38:28 +0800
Subject: [PATCH] TestCanInitialize

---
 .../server/middleware_validate_token_test.go  | 61 ++++++++++++++++++-
 1 file changed, 59 insertions(+), 2 deletions(-)

diff --git a/server/server/middleware_validate_token_test.go b/server/server/middleware_validate_token_test.go
index ee1e7382..5dc46e7c 100644
--- a/server/server/middleware_validate_token_test.go
+++ b/server/server/middleware_validate_token_test.go
@@ -5,6 +5,9 @@ import (
 	"os"
 	"strings"
 	"testing"
+
+	"github.com/tengge1/shadoweditor/server/system"
+	"go.mongodb.org/mongo-driver/bson"
 )
 
 func TestValidateTokenMiddleware(t *testing.T) {
@@ -16,16 +19,70 @@ func TestCanInitialize(t *testing.T) {
 	if err != nil {
 		t.Error(err)
 	}
-
 	mong, err := Mongo()
 	if err != nil {
 		t.Error(err)
 	}
-
 	_, err = mong.ListCollectionNames()
 	if err != nil {
 		t.Error(err)
 	}
+	// save old config
+	oldInitialized := false
+	config := system.Config{}
+	find, err := mong.FindOne(ConfigCollectionName, bson.M{}, &config)
+	if err != nil {
+		t.Error(err)
+	}
+	if find {
+		oldInitialized = config.Initialized
+	}
+	// set Initialized to false
+	if find {
+		update := bson.M{
+			"$set": bson.M{
+				"Initialized": false,
+			},
+		}
+		if _, err = mong.UpdateOne(ConfigCollectionName, bson.M{}, update); err != nil {
+			t.Error(err)
+		}
+	} else {
+		config.Initialized = false
+		if _, err = mong.InsertOne(ConfigCollectionName, config); err != nil {
+			t.Error(err)
+		}
+	}
+	if canInitialize() != true {
+		t.Errorf("expect true, got false")
+	}
+	// set Initialized to true
+	update := bson.M{
+		"$set": bson.M{
+			"Initialized": true,
+		},
+	}
+	if _, err = mong.UpdateOne(ConfigCollectionName, bson.M{}, update); err != nil {
+		t.Error(err)
+	}
+	if canInitialize() != false {
+		t.Errorf("expect false, got true")
+	}
+	// restore initial record
+	if find {
+		update := bson.M{
+			"$set": bson.M{
+				"Initialized": oldInitialized,
+			},
+		}
+		if _, err = mong.UpdateOne(ConfigCollectionName, bson.M{}, update); err != nil {
+			t.Error(err)
+		}
+	} else {
+		if _, err = mong.DeleteAll(ConfigCollectionName); err != nil {
+			t.Error(err)
+		}
+	}
 }
 
 func TestLogAPI(t *testing.T) {