Skip to content

Commit

Permalink
🐛 fix: 修复折叠的 bug
Browse files Browse the repository at this point in the history
  • Loading branch information
@arvinxx
arvinxx committed Mar 18, 2021
1 parent f371e51 commit 9b7e5a4
Show file tree
Hide file tree
Showing 2 changed files with 423 additions and 16 deletions.
371 changes: 369 additions & 2 deletions packages/mindflow/src/utils/dataMap.test.ts
View file
Original file line number Diff line number Diff line change
@@ -1,5 +1,12 @@
import { mapColorToHex, getUncollapsedNode } from './dataMap';
import { MindflowData } from '@arvinxu/mindflow';
import type { BaseEdge, MindflowData } from '@arvinxu/mindflow';
import {
mapColorToHex,
getUncollapsedNode,
getUncollapsedEdge,
getUncollaspedData,
mapTypeToColor,
preprocessData,
} from './dataMap';

describe('mapColorToHex', () => {
it('返回红色', () => {
Expand All @@ -23,6 +30,19 @@ describe('mapColorToHex', () => {
it('返回紫色', () => {
expect(mapColorToHex('purple')).toEqual('#b37feb');
});
it('返回灰色', () => {
expect(mapColorToHex('gray')).toEqual('#8f8f8f');
expect(mapColorToHex('123')).toEqual('#8f8f8f');
});
});

test('mapTypeToColor', () => {
expect(mapTypeToColor('question')).toEqual('red');
expect(mapTypeToColor('action')).toEqual('green');
expect(mapTypeToColor('idea')).toEqual('yellow');
expect(mapTypeToColor('subject')).toEqual('blue');
expect(mapTypeToColor('subject')).toEqual('blue');
expect(mapTypeToColor('123')).toEqual('grey');
});

describe('getUncollapsedNode', () => {
Expand Down Expand Up @@ -106,4 +126,351 @@ describe('getUncollapsedNode', () => {
},
]);
});
it('多个节点', () => {
const data2: MindflowData = {
nodes: [
{
id: '1',
},
{
id: '2',
},
{
id: '3',
},
{
id: '4',
},
{
id: '5',
},
],
edges: [
{ source: '1', target: '2' },
{ source: '1', target: '3' },
{ source: '3', target: '4' },
{ source: '3', target: '5' },
],
};
const nodes = getUncollapsedNode(data2, ['1']);
expect(nodes).toEqual([
{
id: '1',
},
]);
});
});

describe('getUncollapsedEdge', () => {
it('返回正常', () => {
const edges: BaseEdge<any>[] = [
{
source: 'node1',
target: 'node2',
},
{
source: 'node1',
target: 'node3',
},
];

const nodes = getUncollapsedEdge(edges, ['node1']);
expect(nodes).toEqual([]);
});
it('多条边', () => {
const edges: BaseEdge<any>[] = [
{ source: '1', target: '2' },
{ source: '1', target: '3' },
{ source: '3', target: '4' },
{ source: '3', target: '5' },
];

const enableEdgeResult = getUncollapsedEdge(edges, ['1']);
expect(enableEdgeResult).toEqual([]);

const enableEdgeResult2 = getUncollapsedEdge(edges, ['3']);
expect(enableEdgeResult2).toEqual([
{ source: '1', target: '2' },
{ source: '1', target: '3' },
]);
});
it('多个折叠条边', () => {
const edges: BaseEdge<any>[] = [
{ source: '1', target: '2' },
{ source: '1', target: '3' },
{ source: '3', target: '4' },
{ source: '3', target: '5' },
{ source: '2', target: '6' },
{ source: '2', target: '7' },
];

const enableEdgeResult = getUncollapsedEdge(edges, ['2', '3']);
expect(enableEdgeResult).toEqual([
{ source: '1', target: '2' },
{ source: '1', target: '3' },
]);
});
it('折叠一个节点', () => {
const edges = [
{
source: 'node1',
target: 'node2',
},
{
source: 'node1',
target: 'node3',
},
{
source: 'node1',
target: 'node4',
},
{
source: 'node4',
target: 'node5',
},
{
source: 'node5',
target: 'node6',
},
{
source: 'node6',
target: 'node7',
},
{
source: 'node7',
target: 'node8',
},
];

const enableEdge = getUncollapsedEdge(edges, ['node1']);
expect(enableEdge).toEqual([]);
});
});

describe('getUncollapsedData', () => {
const data = {
nodes: [
{
id: 'node1',
},
{
id: 'node2',
},
{
id: 'node3',
},
{
id: 'node4',
},
{
id: 'node5',
},
{
id: 'node6',
},
{
id: 'node7',
},
{
id: 'node8',
},
],
edges: [
{
source: 'node1',
target: 'node2',
},
{
source: 'node1',
target: 'node3',
},
{
source: 'node1',
target: 'node4',
},
{
source: 'node4',
target: 'node5',
},
{
source: 'node5',
target: 'node6',
},
{
source: 'node6',
target: 'node7',
},
{
source: 'node7',
target: 'node8',
},
],
};

it('折叠 node1', () => {
const result = getUncollaspedData(data, ['node1']);
expect(result).toEqual({
edges: [],
nodes: [
{
id: 'node1',
},
],
});
});

it('折叠 node6', () => {
const result = getUncollaspedData(data, ['node6']);

expect(result).toEqual({
edges: [
{
source: 'node1',
target: 'node2',
},
{
source: 'node1',
target: 'node3',
},
{
source: 'node1',
target: 'node4',
},
{
source: 'node4',
target: 'node5',
},
{
source: 'node5',
target: 'node6',
},
],
nodes: [
{
id: 'node1',
},
{
id: 'node2',
},
{
id: 'node3',
},
{
id: 'node4',
},
{
id: 'node5',
},
{
id: 'node6',
},
],
});
});
});

describe('preprocessData', () => {
const data = {
nodes: [
{
id: 'node1',
},
{
id: 'node2',
},
{
id: 'node3',
},
{
id: 'node4',
},
{
id: 'node5',
},
],
edges: [
{
source: 'node1',
target: 'node2',
},
{
source: 'node1',
target: 'node3',
},
{
source: 'node1',
target: 'node4',
},
{
source: 'node4',
target: 'node5',
},
],
};

it('折叠 node4', () => {
const result = preprocessData(data, ['node4']);
expect(result).toEqual({
edges: [
{
source: 'node1',
target: 'node2',
},
{
source: 'node1',
target: 'node3',
},
{
source: 'node1',
target: 'node4',
},
],
nodes: [
{
component: 'mind-node',
data: {
collapsed: false,
leaf: false,
},
height: 40,
id: 'node1',
shape: 'react-shape',
width: 220,
},
{
component: 'mind-node',
data: {
collapsed: false,
leaf: true,
},
height: 40,
id: 'node2',
shape: 'react-shape',
width: 220,
},
{
component: 'mind-node',
data: {
collapsed: false,
leaf: true,
},
height: 40,
id: 'node3',
shape: 'react-shape',
width: 220,
},
{
component: 'mind-node',
data: {
collapsed: true,
leaf: false,
},
height: 40,
id: 'node4',
shape: 'react-shape',
width: 220,
},
],
});
});
});
Loading

0 comments on commit 9b7e5a4

Please sign in to comment.