sub2api/backend/internal/config/validate_dingtalk_test.go at b19da9c7fea8a6ae075d2d265dec30c0ed75e8d9 · DaydreamCoding/sub2api · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
package config

import (
	"testing"

	"github.com/stretchr/testify/require"
)

func TestValidateDingTalkConfig_Disabled_Skip(t *testing.T) {
	require.NoError(t, ValidateDingTalkConfig(DingTalkConnectConfig{Enabled: false}))
}

func TestValidateDingTalkConfig_V4_DingTalkAppKind(t *testing.T) {
	err := ValidateDingTalkConfig(DingTalkConnectConfig{
		Enabled:               true,
		DingTalkAppKind:       "third_party_enterprise_app",
		CorpRestrictionPolicy: "none",
	})
	require.ErrorIs(t, err, ErrDingTalkV4InvalidAppKind)
}

func TestValidateDingTalkConfig_V1_InternalOnlyRequiresInternalAppType(t *testing.T) {
	err := ValidateDingTalkConfig(DingTalkConnectConfig{
		Enabled:               true,
		DingTalkAppKind:       "internal_app",
		AppType:               "public",
		CorpRestrictionPolicy: "internal_only",
		InternalCorpID:        "dingABC",
	})
	require.ErrorIs(t, err, ErrDingTalkV1AppTypeMismatch)
}

// TestValidateDingTalkConfig_V3_InternalOnlyAllowsEmptyCorpID 验证方案 A：
// internal_only 策略下，InternalCorpID="" 应通过校验（企业隔离由钉钉 AppType=internal 保证）。
func TestValidateDingTalkConfig_V3_InternalOnlyAllowsEmptyCorpID(t *testing.T) {
	err := ValidateDingTalkConfig(DingTalkConnectConfig{
		Enabled:               true,
		DingTalkAppKind:       "internal_app",
		AppType:               "internal",
		CorpRestrictionPolicy: "internal_only",
		InternalCorpID:        "",
	})
	require.NoError(t, err)
}

func TestValidateDingTalkConfig_HappyPath_None(t *testing.T) {
	require.NoError(t, ValidateDingTalkConfig(DingTalkConnectConfig{
		Enabled:               true,
		DingTalkAppKind:       "internal_app",
		AppType:               "public",
		CorpRestrictionPolicy: "none",
	}))
}