David Walsh
106 lines
4.5 KiB
TypeScript
106 lines
4.5 KiB
TypeScript
import * as chai from 'chai';
|
|
import 'mocha';
|
|
|
|
import { TokenAdjacencyGraphBuilder } from '../../src/asset-swapper/utils/token_adjacency_graph';
|
|
|
|
import { chaiSetup } from './utils/chai_setup';
|
|
|
|
chaiSetup.configure();
|
|
const expect = chai.expect;
|
|
|
|
describe('TokenAdjacencyGraphBuilder and TokenAdjacencyGraph', () => {
|
|
describe('constructor', () => {
|
|
it('sanitizes passed default tokens to lower case', async () => {
|
|
const graph = new TokenAdjacencyGraphBuilder(['DEFAULT_1', 'DEFAULT_2']).build();
|
|
|
|
expect(graph.getAdjacentTokens('random_token')).to.deep.eq(['default_1', 'default_2']);
|
|
});
|
|
});
|
|
|
|
describe('add', () => {
|
|
it('adds a new token path to the graph', async () => {
|
|
const graph = new TokenAdjacencyGraphBuilder(['default_1', 'default_2']).add('token_a', 'token_b').build();
|
|
|
|
expect(graph.getAdjacentTokens('token_a')).to.deep.eq(['default_1', 'default_2', 'token_b']);
|
|
});
|
|
|
|
it('adds lower-cased token path to the graph', async () => {
|
|
const graph = new TokenAdjacencyGraphBuilder(['default_1', 'default_2']).add('TOKEN_A', 'TOKEN_B').build();
|
|
|
|
expect(graph.getAdjacentTokens('token_a')).to.deep.eq(['default_1', 'default_2', 'token_b']);
|
|
});
|
|
|
|
it('ignores an existing to token', async () => {
|
|
const graph = new TokenAdjacencyGraphBuilder().add('token_a', 'token_b').add('token_a', 'token_b').build();
|
|
|
|
expect(graph.getAdjacentTokens('token_a')).to.deep.eq(['token_b']);
|
|
});
|
|
});
|
|
|
|
describe('addBidirectional', () => {
|
|
it('adds a bidirectional path to the graph', async () => {
|
|
const graph = new TokenAdjacencyGraphBuilder(['default_1']).addBidirectional('token_a', 'token_b').build();
|
|
|
|
expect(graph.getAdjacentTokens('token_a')).to.deep.eq(['default_1', 'token_b']);
|
|
expect(graph.getAdjacentTokens('token_b')).to.deep.eq(['default_1', 'token_a']);
|
|
});
|
|
});
|
|
|
|
describe('addCompleteSubgraph', () => {
|
|
it('adds a complete subgraph to the graph', async () => {
|
|
const graph = new TokenAdjacencyGraphBuilder(['default_1'])
|
|
.addCompleteSubgraph(['token_a', 'token_b', 'token_c', 'token_d'])
|
|
.build();
|
|
|
|
expect(graph.getAdjacentTokens('token_a')).to.deep.eq(['default_1', 'token_b', 'token_c', 'token_d']);
|
|
expect(graph.getAdjacentTokens('token_b')).to.deep.eq(['default_1', 'token_a', 'token_c', 'token_d']);
|
|
expect(graph.getAdjacentTokens('token_c')).to.deep.eq(['default_1', 'token_a', 'token_b', 'token_d']);
|
|
expect(graph.getAdjacentTokens('token_d')).to.deep.eq(['default_1', 'token_a', 'token_b', 'token_c']);
|
|
});
|
|
});
|
|
|
|
describe('tap', () => {
|
|
it('applies callback correctly', async () => {
|
|
const graph = new TokenAdjacencyGraphBuilder(['default_1'])
|
|
.tap((g) => {
|
|
g.add('token_a', 'token_b');
|
|
g.add('token_c', 'token_d');
|
|
})
|
|
.build();
|
|
|
|
expect(graph.getAdjacentTokens('token_a')).to.deep.eq(['default_1', 'token_b']);
|
|
expect(graph.getAdjacentTokens('token_c')).to.deep.eq(['default_1', 'token_d']);
|
|
});
|
|
});
|
|
|
|
describe('getIntermediateTokens', () => {
|
|
it('returns intermediate tokens without a duplicate ', async () => {
|
|
const graph = new TokenAdjacencyGraphBuilder(['default_1'])
|
|
.add('token_a', 'token_b')
|
|
.add('token_c', 'token_b')
|
|
.build();
|
|
|
|
expect(graph.getIntermediateTokens('token_a', 'token_c')).to.deep.eq(['default_1', 'token_b']);
|
|
});
|
|
|
|
it('returns intermediate tokens after lower-casing taker and maker tokens', async () => {
|
|
const graph = new TokenAdjacencyGraphBuilder(['default_1'])
|
|
.add('token_a', 'token_b')
|
|
.add('token_c', 'token_d')
|
|
.build();
|
|
|
|
expect(graph.getIntermediateTokens('TOKEN_a', 'token_C')).to.deep.eq(['default_1', 'token_b', 'token_d']);
|
|
});
|
|
|
|
it('returns intermediate tokens excluding taker token or maker token ', async () => {
|
|
const graph = new TokenAdjacencyGraphBuilder(['default_1'])
|
|
.addBidirectional('token_a', 'token_b')
|
|
.addBidirectional('token_b', 'token_c')
|
|
.addBidirectional('token_c', 'token_a')
|
|
.build();
|
|
|
|
expect(graph.getIntermediateTokens('token_a', 'token_c')).to.deep.eq(['default_1', 'token_b']);
|
|
});
|
|
});
|
|
});
|