69 lines
		
	
	
		
			1.9 KiB
		
	
	
	
		
			TypeScript
		
	
	
	
			
		
		
	
	
			69 lines
		
	
	
		
			1.9 KiB
		
	
	
	
		
			TypeScript
		
	
	
	
/*
 | 
						|
 * Copyright 2025 coze-dev Authors
 | 
						|
 *
 | 
						|
 * 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.
 | 
						|
 */
 | 
						|
 
 | 
						|
import React, { type FC } from 'react';
 | 
						|
 | 
						|
import {
 | 
						|
  renderHook,
 | 
						|
  type WrapperComponent,
 | 
						|
} from '@testing-library/react-hooks';
 | 
						|
import { ToolGroupKey } from '@coze-agent-ide/tool-config';
 | 
						|
 | 
						|
import { useAbilityConfig } from '../../src/hooks/builtin/use-ability-config';
 | 
						|
import {
 | 
						|
  AbilityAreaContextProvider,
 | 
						|
  type ToolEntryCommonProps,
 | 
						|
  ToolView,
 | 
						|
  ToolKey,
 | 
						|
  AbilityScope,
 | 
						|
  GroupingContainer,
 | 
						|
} from '../../src';
 | 
						|
 | 
						|
vi.stubGlobal('IS_DEV_MODE', false);
 | 
						|
 | 
						|
type ITestComponentProps = ToolEntryCommonProps;
 | 
						|
 | 
						|
const TestComponent: FC<ITestComponentProps> = ({ title, toolKey }) => (
 | 
						|
  <div>
 | 
						|
    test{title} {toolKey}
 | 
						|
  </div>
 | 
						|
);
 | 
						|
 | 
						|
describe('useAbilityConfig', () => {
 | 
						|
  test('useAbilityConfig', () => {
 | 
						|
    const wrapper: WrapperComponent<{ children }> = ({ children }) => (
 | 
						|
      <AbilityAreaContextProvider mode={0}>
 | 
						|
        {children}
 | 
						|
        <ToolView>
 | 
						|
          <GroupingContainer
 | 
						|
            toolGroupKey={ToolGroupKey.CHARACTER}
 | 
						|
            title="testGroup"
 | 
						|
          >
 | 
						|
            <TestComponent toolKey={ToolKey.ONBOARDING} title="demo" />
 | 
						|
          </GroupingContainer>
 | 
						|
        </ToolView>
 | 
						|
      </AbilityAreaContextProvider>
 | 
						|
    );
 | 
						|
 | 
						|
    const { result } = renderHook(() => useAbilityConfig(), { wrapper });
 | 
						|
 | 
						|
    expect(result.current).toEqual({
 | 
						|
      abilityKey: ToolKey.ONBOARDING,
 | 
						|
      scope: AbilityScope.TOOL,
 | 
						|
    });
 | 
						|
  });
 | 
						|
});
 |