Test: filesystem/Preview

models/policy.go

@@ -68,7 +68,6 @@ func GetPolicyByID(ID interface{}) (Policy, error) {
 }
 
 // IsDirectlyPreview 返回此策略下文件是否可以直接预览（不需要重定向）
-// TODO 测试
 func (policy *Policy) IsDirectlyPreview() bool {
 	return policy.Type == "local"
 }

models/policy_test.go

@@ -133,3 +133,11 @@ func TestPolicy_GenerateFileName(t *testing.T) {
 	testPolicy.FileNameRule = "{uid}123{originname}"
 	asserts.Equal("1123{filename}{.suffix}", testPolicy.GenerateFileName(1, ""))
 }
+
+func TestPolicy_IsDirectlyPreview(t *testing.T) {
+	asserts := assert.New(t)
+	policy := Policy{Type: "local"}
+	asserts.True(policy.IsDirectlyPreview())
+	policy.Type = "remote"
+	asserts.False(policy.IsDirectlyPreview())
+}