From fa7416452e3a722226484975160097cd238d7ed7 Mon Sep 17 00:00:00 2001
From: ysqi <devysq@gmail.com>
Date: Wed, 27 Apr 2016 22:26:32 +0800
Subject: [PATCH] Change set test mode way

---
 beego.go | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/beego.go b/beego.go
index db5db3f5..32b64f75 100644
--- a/beego.go
+++ b/beego.go
@@ -85,9 +85,11 @@ func initBeforeHTTPRun() {
 
 // TestBeegoInit is for test package init
 func TestBeegoInit(ap string) {
-	os.Setenv("BEEGO_RUNMODE", "test")
 	appConfigPath = filepath.Join(ap, "conf", "app.conf")
 	os.Chdir(ap)
-	LoadAppConfig(appConfigProvider, appConfigPath)
+	if err := LoadAppConfig(appConfigProvider, appConfigPath); err != nil {
+		panic(err)
+	}
+	BConfig.RunMode = "test"
 	initBeforeHTTPRun()
 }