71 lines
2.1 KiB
TypeScript
71 lines
2.1 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 { ArrayUtil } from '../src/array';
|
|
|
|
describe('array', () => {
|
|
it('array2Map', () => {
|
|
const testItem1 = { id: '1', name: 'Alice', age: 20 };
|
|
const testItem2 = { id: '2', name: 'Bob', age: 25 };
|
|
const { array2Map } = ArrayUtil;
|
|
const array1 = [testItem1, testItem2];
|
|
|
|
const mapById = array2Map(array1, 'id');
|
|
expect(mapById).toEqual({
|
|
'1': testItem1,
|
|
'2': testItem2,
|
|
});
|
|
|
|
const mapByName = array2Map(array1, 'name', 'age');
|
|
expect(mapByName).toEqual({ Alice: 20, Bob: 25 });
|
|
|
|
const array = [testItem1, testItem2];
|
|
const mapByIdFunc = array2Map(
|
|
array,
|
|
'id',
|
|
item => `${item.name}-${item.age}`,
|
|
);
|
|
expect(mapByIdFunc).toEqual({ '1': 'Alice-20', '2': 'Bob-25' });
|
|
});
|
|
|
|
it('mapAndFilter', () => {
|
|
const { mapAndFilter } = ArrayUtil;
|
|
const array = [
|
|
{ id: 1, name: 'Alice', value: 100 },
|
|
{ id: 2, name: 'Bob', value: 200 },
|
|
];
|
|
|
|
// filter
|
|
const result1 = mapAndFilter(array, {
|
|
filter: item => item.name === 'Alice',
|
|
});
|
|
expect(result1).toEqual([{ id: 1, name: 'Alice', value: 100 }]);
|
|
|
|
// map
|
|
const result2 = mapAndFilter(array, {
|
|
map: item => ({ value: item.value }),
|
|
});
|
|
expect(result2).toEqual([{ value: 100 }, { value: 200 }]);
|
|
|
|
// filter & map
|
|
const result3 = mapAndFilter(array, {
|
|
filter: item => item.value > 100,
|
|
map: item => ({ id: item.id, name: item.name }),
|
|
});
|
|
expect(result3).toEqual([{ id: 2, name: 'Bob' }]);
|
|
});
|
|
});
|