import { ShortChannelId } from "@node-lightning/core" ; import { expect } from "chai" ; import { Channel } from "../lib/channel" ; import { Node } from "../lib/node" ; describe ( "Node class" , ( ) => { describe ( ".linkChannel" , ( ) => { it ( "should add the channel to list of channels" , ( ) => { const node = new Node ( ) ; const c = new Channel ( ) ; c . shortChannelId = new ShortChannelId ( 1288457 , 3 , 0 ) ; node . linkChannel ( c ) ; expect ( node . channels . size ) . to . equal ( 1 ) ; expect ( node . channels . get ( c . shortChannelId . toNumber ( ) ) ) . to . equal ( c ) ; } ) ; } ) ; describe ( ".unlinkChannel" , ( ) => { it ( "should remove the channel from the list of channels" , ( ) => { const c1 = new Channel ( ) ; c1 . shortChannelId = new ShortChannelId ( 1288457 , 3 , 0 ) ; const c2 = new Channel ( ) ; c2 . shortChannelId = new ShortChannelId ( 1288457 , 4 , 0 ) ; const node = new Node ( ) ; node . channels = new Map ( [ [ new ShortChannelId ( 1288457 , 3 , 0 ) . toNumber ( ) , c1 ] , [ new ShortChannelId ( 1288457 , 4 , 0 ) . toNumber ( ) , c2 ] , ] ) ; node . unlinkChannel ( c2 ) ; expect ( node . channels . size ) . to . equal ( 1 ) ; expect ( node . channels . get ( new ShortChannelId ( 1288457 , 3 , 0 ) . toNumber ( ) ) ) . to . equal ( c1 ) ; } ) ; } ) ; } ) ;