Test: get source URL of files

models/group.go

@@ -22,7 +22,6 @@ type Group struct {
 }
 
 // GetAria2Option 获取用户离线下载设备
-// TODO:测试
 func (group *Group) GetAria2Option() [3]bool {
 	if len(group.Aria2Option) != 5 {
 		return [3]bool{false, false, false}

models/migration.go

@@ -195,7 +195,8 @@ func addDefaultGroups() {
 	// 未找到初始游客用户组时，则创建
 	if gorm.IsRecordNotFoundError(err) {
 		defaultAdminGroup := Group{
-			Name: "游客",
+			Name:     "游客",
+			Policies: "[]",
 		}
 		if err := DB.Create(&defaultAdminGroup).Error; err != nil {
 			util.Log().Panic("无法创建初始游客用户组, %s", err)

models/setting_test.go

@@ -140,3 +140,29 @@ func TestIsTrueVal(t *testing.T) {
 	asserts.False(IsTrueVal("0"))
 	asserts.False(IsTrueVal("false"))
 }
+
+func TestGetSiteURL(t *testing.T) {
+	asserts := assert.New(t)
+
+	// 正常
+	{
+		err := cache.Deletes([]string{"siteURL"}, "setting_")
+		asserts.NoError(err)
+
+		mock.ExpectQuery("SELECT(.+)").WithArgs("siteURL").WillReturnRows(sqlmock.NewRows([]string{"id", "value"}).AddRow(1, "https://drive.cloudreve.org"))
+		siteURL := GetSiteURL()
+		asserts.NoError(mock.ExpectationsWereMet())
+		asserts.Equal("https://drive.cloudreve.org", siteURL.String())
+	}
+
+	// 失败 返回默认值
+	{
+		err := cache.Deletes([]string{"siteURL"}, "setting_")
+		asserts.NoError(err)
+
+		mock.ExpectQuery("SELECT(.+)").WithArgs("siteURL").WillReturnRows(sqlmock.NewRows([]string{"id", "value"}).AddRow(1, ":][\\/\\]sdf"))
+		siteURL := GetSiteURL()
+		asserts.NoError(mock.ExpectationsWereMet())
+		asserts.Equal("https://cloudreve.org", siteURL.String())
+	}
+}