import { Map } from 'immutable';
import * as Selectors from './portSelectors';
import { defaultState } from '../reducers/flow.reducer';
import { LinkRecord } from '../constants/flowdesigner.model';
import { PORT_SINK, PORT_SOURCE } from '../constants/flowdesigner.constants';
import { Port } from '../api';
import {
	State,
	Id,
	LinkRecord as LinkRecordType,
	PortRecord as PortRecordType,
} from '../customTypings/index.d';

const port1 = Port.create('id1', 'nodeId1', 0, PORT_SINK, 'reactComponentType');
const port2 = Port.create('id2', 'nodeId1', 0, PORT_SOURCE, 'reactComponentType');
const port3 = Port.create('id3', 'nodeId2', 0, PORT_SINK, 'reactComponentType');
const port4 = Port.create('id4', 'nodeId2', 0, PORT_SOURCE, 'reactComponentType');

const givenState: Partial<State> = defaultState
	.set(
		'links',
		Map<Id, LinkRecordType>().set(
			'id1',
			new LinkRecord({
				id: 'id1',
				source: 'id1',
				target: 'id2',
			}),
		),
	)
	.set(
		'ports',
		Map<Id, PortRecordType>()
			.set('id1', port1)
			.set('id2', port2)
			.set('id3', port3)
			.set('id4', port4),
	);

describe('getEmitterPorts', () => {
	it('return a map with port id2 && id4', () => {
		// given
		// when
		const result = Selectors.getEmitterPorts(givenState);
		// expect
		expect(result.has('id2')).toBe(true);
		expect(result.has('id4')).toBe(true);
	});
});

describe('getSinkPorts', () => {
	it('return a map with port id1 & id3', () => {
		// given
		// when
		const result = Selectors.getSinkPorts(givenState);
		// expect
		expect(result.has('id1')).toBe(true);
		expect(result.has('id3')).toBe(true);
	});
});

describe('getEmitterPortsForNode', () => {
	it('return a map with port id2', () => {
		// given
		// when
		const result = Selectors.getEmitterPortsForNode(givenState)('nodeId1');
		// expect
		expect(result.has('id2')).toBe(true);
	});
});

describe('getSinkPortsForNode', () => {
	it('return a map with port id1', () => {
		// given
		// when
		const result = Selectors.getSinkPortsForNode(givenState)('nodeId1');
		// expect
		expect(result.has('id1')).toBe(true);
	});
});