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,
 | |
|     });
 | |
|   });
 | |
| });
 |