tlab-trading-toolkit/test/mocks/MockLivebook.ts

A trading toolkit for building advanced trading bots on the GDAX platform

/***************************************************************************************************************************
 * @license                                                                                                                *
 * Copyright 2017 Coinbase, Inc.                                                                                           *
 *                                                                                                                         *
 * 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 { LiveBookConfig, LiveOrderbook } from '../../src/core/LiveOrderbook';
import { StreamMessage } from '../../src/core/Messages';
import { StaticCommandSet } from '../../src/lib/StaticCommandSet';

export interface MockLivebook {
    liveBook: LiveOrderbook;
    messages: StaticCommandSet;
}

export function createMockLivebook(product: string, messages: StreamMessage[]): MockLivebook {
    const bookConfig: LiveBookConfig = {
        logger: null,
        strictMode: false,
        product: product
    };
    const liveBook = new LiveOrderbook(bookConfig);
    const messageStream = new StaticCommandSet(messages);
    messageStream.pipe(liveBook);
    return {
        liveBook: liveBook,
        messages: messageStream
    };
}