eigent/test/unit/components/resolveContextConnector.test.ts

66 lines
2.1 KiB
TypeScript

// ========= Copyright 2025-2026 @ Eigent.ai All Rights Reserved. =========
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// ========= Copyright 2025-2026 @ Eigent.ai All Rights Reserved. =========
import {
resolveContextConnector,
type ContextConnector,
} from '@/components/Session/SidePanel/sections/buildContextItems';
import { describe, expect, it } from 'vitest';
const notion: ContextConnector = {
service: 'notion',
displayName: 'Notion',
};
const slack: ContextConnector = {
service: 'slack',
displayName: 'Slack',
};
describe('resolveContextConnector', () => {
it('matches a provider-prefixed toolkit by name', () => {
expect(
resolveContextConnector('NotionMCPToolkit', 'search', '', [notion, slack])
).toBe(notion);
});
it('matches a provider by method when the toolkit is generic', () => {
expect(
resolveContextConnector('MCPToolkit', 'slack_send_message', '', [
notion,
slack,
])
).toBe(slack);
});
it('assumes the only connector for a connector gateway call', () => {
expect(
resolveContextConnector('ConnectorGateway', 'call', '', [slack])
).toBe(slack);
expect(
resolveContextConnector('connector_gateway', 'call', '', [slack])
).toBe(slack);
});
it('stays generic for an unidentified MCP call even with one connector', () => {
expect(
resolveContextConnector('MCPToolkit', 'call', '', [slack])
).toBeNull();
});
it('stays generic when a gateway call cannot pick between connectors', () => {
expect(
resolveContextConnector('ConnectorGateway', 'call', '', [notion, slack])
).toBeNull();
});
});