Function TestMCPRegex

coderd/aibridged/mcp_internal_test.go:13–62

(t *testing.T)

)

func TestMCPRegex(t *testing.T) {
	t.Parallel()

	cases := []struct {
		name                  string
		allowRegex, denyRegex string
		expectedErr           error
	}{
		{
			name:        "invalid allow regex",
			allowRegex:  `\`,
			expectedErr: ErrCompileRegex,
		},
		{
			name:        "invalid deny regex",
			denyRegex:   `+`,
			expectedErr: ErrCompileRegex,
		},
		{
			name: "valid empty",
		},
		{
			name:       "valid",
			allowRegex: "(allowed|allowed2)",
			denyRegex:  ".*disallowed.*",
		},
	}

	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()

			logger := testutil.Logger(t)
			f := NewMCPProxyFactory(logger, otel.Tracer("aibridged_test"), nil)

			_, err := f.newStreamableHTTPServerProxy(&proto.MCPServerConfig{
				Id:             "mock",
				Url:            "mock/mcp",
				ToolAllowRegex: tc.allowRegex,
				ToolDenyRegex:  tc.denyRegex,
			}, "")

			if tc.expectedErr == nil {
				require.NoError(t, err)
			} else {
				require.ErrorIs(t, err, tc.expectedErr)
			}
		})
	}
}