import { CONSTANTS } from "../../src/Constants"; import { Player } from "../../src/Player"; import { Company } from "../../src/Company/Company"; import { Server } from "../../src/Server/Server"; import { buyStock, sellStock, shortStock, sellShort } from "../../src/StockMarket/BuyingAndSelling"; import { Order } from "../../src/StockMarket/Order"; import { forecastForecastChangeFromCompanyWork, forecastForecastChangeFromHack, influenceStockThroughCompanyWork, influenceStockThroughServerGrow, influenceStockThroughServerHack, } from "../../src/StockMarket/PlayerInfluencing"; import { processOrders, IProcessOrderRefs } from "../../src/StockMarket/OrderProcessing"; import { Stock, StockForecastInfluenceLimit } from "../../src/StockMarket/Stock"; import { cancelOrder, deleteStockMarket, initStockMarket, initSymbolToStockMap, placeOrder, processStockPrices, StockMarket, SymbolToStockMap, } from "../../src/StockMarket/StockMarket"; import { forecastChangePerPriceMovement, getBuyTransactionCost, getSellTransactionGain, processTransactionForecastMovement, } from "../../src/StockMarket/StockMarketHelpers"; import { CompanyName, OrderType, PositionType } from "../../src/Enums"; jest.mock(`!!raw-loader!../NetscriptDefinitions.d.ts`, () => "", { virtual: true, }); // jest.mock("../src/ui/React/createPopup.tsx", () => ({ // createPopup: jest.fn(), // })); describe("Stock Market Tests", function () { const commission = CONSTANTS.StockMarketCommission; // Generic Stock object that can be used by each test let stock: Stock; const ctorParams = { b: true, initPrice: 10e3, marketCap: 5e9, mv: 2, name: "MockStock", otlkMag: 20, spreadPerc: 1, shareTxForMovement: 5e3, symbol: "mock", }; beforeEach(function () { function construct(): void { stock = new Stock(ctorParams); } expect(construct).not.toThrow(); }); describe("Stock Class", function () { describe("constructor", function () { it("should have default parameters", function () { function construct(): void { new Stock(); // eslint-disable-line no-new } expect(construct).not.toThrow(); const defaultStock = new Stock(); expect(defaultStock).not.toEqual(null); expect(defaultStock.name).toEqual(""); }); it("should properly initialize props from parameters", function () { expect(stock.name).toEqual(ctorParams.name); expect(stock.symbol).toEqual(ctorParams.symbol); expect(stock.price).toEqual(ctorParams.initPrice); expect(stock.lastPrice).toEqual(ctorParams.initPrice); expect(stock.b).toEqual(ctorParams.b); expect(stock.mv).toEqual(ctorParams.mv); expect(stock.shareTxForMovement).toEqual(ctorParams.shareTxForMovement); expect(stock.shareTxUntilMovement).toEqual(ctorParams.shareTxForMovement); expect(stock.maxShares).toBeLessThan(stock.totalShares); expect(stock.spreadPerc).toEqual(ctorParams.spreadPerc); expect(stock.otlkMag).toEqual(ctorParams.otlkMag); expect(stock.otlkMagForecast).toEqual(ctorParams.b ? 50 + ctorParams.otlkMag : 50 - ctorParams.otlkMag); }); it("should properly initialize props from range-values", function () { const params = { b: true, initPrice: { max: 10e3, min: 1e3, }, marketCap: 5e9, mv: { divisor: 100, max: 150, min: 50, }, name: "MockStock", otlkMag: 10, spreadPerc: { divisor: 10, max: 10, min: 1, }, shareTxForMovement: { max: 10e3, min: 5e3, }, symbol: "mock", }; function construct(): void { new Stock(params); // eslint-disable-line no-new } expect(construct).not.toThrow(); const stock = new Stock(params); expect(stock).not.toEqual(null); expect(stock.price).toBeGreaterThanOrEqual(params.initPrice.min); expect(stock.price).toBeLessThanOrEqual(params.initPrice.max); expect(stock.mv).toBeGreaterThanOrEqual(params.mv.min / params.mv.divisor); expect(stock.mv).toBeLessThanOrEqual(params.mv.max / params.mv.divisor); expect(stock.spreadPerc).toBeGreaterThanOrEqual(params.spreadPerc.min / params.spreadPerc.divisor); expect(stock.spreadPerc).toBeLessThanOrEqual(params.spreadPerc.max / params.spreadPerc.divisor); expect(stock.shareTxForMovement).toBeGreaterThanOrEqual(params.shareTxForMovement.min); expect(stock.shareTxForMovement).toBeLessThanOrEqual(params.shareTxForMovement.max); }); it("should round the 'totalShare' prop to the nearest 100k", function () { expect(stock.totalShares % 100e3).toEqual(0); }); }); describe("#changeForecastForecast()", function () { it("should get the stock's second-order forecast property", function () { stock.changeForecastForecast(99); expect(stock.otlkMagForecast).toEqual(99); stock.changeForecastForecast(1); expect(stock.otlkMagForecast).toEqual(1); }); it("should prevent values outside of 0-100", function () { stock.changeForecastForecast(101); expect(stock.otlkMagForecast).toEqual(100); stock.changeForecastForecast(-1); expect(stock.otlkMagForecast).toEqual(0); }); }); describe("#changePrice()", function () { it("should set both the last price and current price properties", function () { const newPrice = 20e3; stock.changePrice(newPrice); expect(stock.lastPrice).toEqual(ctorParams.initPrice); expect(stock.price).toEqual(newPrice); }); }); describe("#cycleForecast()", function () { it("should appropriately change the otlkMag by the given amount when b=true", function () { stock.getForecastIncreaseChance = () => { return 1; }; stock.cycleForecast(5); expect(stock.otlkMag).toEqual(ctorParams.otlkMag + 5); stock.getForecastIncreaseChance = () => { return 0; }; stock.cycleForecast(10); expect(stock.otlkMag).toEqual(ctorParams.otlkMag - 5); }); it("should NOT(!) the stock's 'b' property if it causes 'otlkMag' to go below 0", function () { stock.getForecastIncreaseChance = () => { return 0; }; stock.cycleForecast(25); expect(stock.otlkMag).toEqual(5); expect(stock.b).toEqual(false); }); }); describe("#cycleForecastForecast()", function () { it("should increase the stock's second-order forecast by a given amount", function () { const expected = [65, 75]; stock.cycleForecastForecast(5); expect(expected).toEqual(expect.arrayContaining([stock.otlkMagForecast])); }); }); describe("#flipForecastForecast()", function () { it("should flip the 'otlkMagForecast' property around 50", function () { stock.otlkMagForecast = 50; stock.flipForecastForecast(); expect(stock.otlkMagForecast).toEqual(50); stock.otlkMagForecast = 60; stock.flipForecastForecast(); expect(stock.otlkMagForecast).toEqual(40); stock.otlkMagForecast = 90; stock.flipForecastForecast(); expect(stock.otlkMagForecast).toEqual(10); stock.otlkMagForecast = 100; stock.flipForecastForecast(); expect(stock.otlkMagForecast).toEqual(0); stock.otlkMagForecast = 40; stock.flipForecastForecast(); expect(stock.otlkMagForecast).toEqual(60); stock.otlkMagForecast = 0; stock.flipForecastForecast(); expect(stock.otlkMagForecast).toEqual(100); stock.otlkMagForecast = 25; stock.flipForecastForecast(); expect(stock.otlkMagForecast).toEqual(75); }); }); describe("#getAbsoluteForecast()", function () { it("should return the absolute forecast on a 1-100 scale", function () { stock.b = true; stock.otlkMag = 10; expect(stock.getAbsoluteForecast()).toEqual(60); stock.b = false; expect(stock.getAbsoluteForecast()).toEqual(40); stock.otlkMag = 30; expect(stock.getAbsoluteForecast()).toEqual(20); stock.b = true; expect(stock.getAbsoluteForecast()).toEqual(80); stock.otlkMag = 0; expect(stock.getAbsoluteForecast()).toEqual(50); stock.b = false; expect(stock.getAbsoluteForecast()).toEqual(50); }); }); describe("#getAskPrice()", function () { it("should return the price increased by spread percentage", function () { const perc = stock.spreadPerc / 100; expect(perc).toBeLessThan(1); expect(perc).toBeGreaterThan(0); const expected = stock.price * (1 + perc); expect(stock.getAskPrice()).toEqual(expected); }); }); describe("#getBidPrice()", function () { it("should return the price decreased by spread percentage", function () { const perc = stock.spreadPerc / 100; expect(perc).toBeLessThan(1); expect(perc).toBeGreaterThan(0); const expected = stock.price * (1 - perc); expect(stock.getBidPrice()).toEqual(expected); }); }); describe("#getForecastIncreaseChance()", function () { it("should return the chance that the stock has of increasing in decimal form", function () { stock.b = true; stock.otlkMagForecast = 90; stock.otlkMag = 20; // Absolute forecast of 70 expect(stock.getForecastIncreaseChance()).toEqual(0.7); stock.otlkMag = 25; // Absolute forecast of 75 expect(stock.getForecastIncreaseChance()).toEqual(0.65); stock.otlkMagForecast = 100; stock.otlkMag = 0; // Absolute forecast of 50 expect(stock.getForecastIncreaseChance()).toEqual(0.95); stock.otlkMagForecast = 60; stock.otlkMag = 25; // Absolute forecast of 75 expect(stock.getForecastIncreaseChance()).toEqual(0.35); stock.otlkMagForecast = 10; expect(stock.getForecastIncreaseChance()).toEqual(0.05); stock.b = false; stock.otlkMagForecast = 90; stock.otlkMag = 20; // Absolute forecast of 30 expect(stock.getForecastIncreaseChance()).toEqual(0.95); stock.otlkMagForecast = 50; stock.otlkMag = 25; // Absolute forecast of 25 expect(stock.getForecastIncreaseChance()).toEqual(0.75); stock.otlkMagForecast = 100; stock.otlkMag = 0; // Absolute forecast of 50 expect(stock.getForecastIncreaseChance()).toEqual(0.95); stock.otlkMagForecast = 5; stock.otlkMag = 25; // Absolute forecast of 25 expect(stock.getForecastIncreaseChance()).toEqual(0.3); stock.otlkMagForecast = 10; expect(stock.getForecastIncreaseChance()).toEqual(0.35); stock.otlkMagForecast = 50; stock.otlkMag = 0; expect(stock.getForecastIncreaseChance()).toEqual(0.5); stock.otlkMagForecast = 25; stock.otlkMag = 5; // Asolute forecast of 45 expect(stock.getForecastIncreaseChance()).toEqual(0.3); }); }); describe("#influenceForecast()", function () { beforeEach(function () { stock.otlkMag = 10; }); it("should change the forecast's value towards 50", function () { stock.influenceForecast(2); expect(stock.otlkMag).toEqual(8); }); it("should not care about whether the stock is in bull or bear mode", function () { stock.b = true; stock.influenceForecast(1); expect(stock.otlkMag).toEqual(9); stock.b = false; stock.influenceForecast(2); expect(stock.otlkMag).toEqual(7); }); it("should not influence the forecast beyond the limit", function () { stock.influenceForecast(10); expect(stock.otlkMag).toEqual(StockForecastInfluenceLimit); stock.influenceForecast(10); expect(stock.otlkMag).toEqual(StockForecastInfluenceLimit); }); }); describe("#influenceForecastForecast()", function () { it("should change the second-order forecast's value towards 50", function () { stock.otlkMagForecast = 75; stock.influenceForecastForecast(15); expect(stock.otlkMagForecast).toEqual(60); stock.otlkMagForecast = 25; stock.influenceForecastForecast(15); expect(stock.otlkMagForecast).toEqual(40); }); it("should not change the second-order forecast past 50", function () { stock.otlkMagForecast = 40; stock.influenceForecastForecast(20); expect(stock.otlkMagForecast).toEqual(50); stock.otlkMagForecast = 60; stock.influenceForecastForecast(20); expect(stock.otlkMagForecast).toEqual(50); }); }); }); describe("StockMarket object", function () { describe("Initialization", function () { // Keeps track of initialized stocks. Contains their symbols const stocks: string[] = []; beforeEach(function () { expect(initStockMarket).not.toThrow(); expect(initSymbolToStockMap).not.toThrow(); }); it("should have Stock objects", function () { for (const prop in StockMarket) { const stock = StockMarket[prop]; if (stock instanceof Stock) { stocks.push(stock.symbol); } } // We'll just check that there are some stocks expect(stocks.length).toBeGreaterThan(1); }); it("should have an order book in the 'Orders' property", function () { expect(StockMarket).toHaveProperty("Orders"); const orderbook = StockMarket["Orders"]; for (const symbol of stocks) { const ordersForStock = orderbook[symbol]; expect(ordersForStock).toEqual([]); expect(ordersForStock.length).toEqual(0); } }); it("should have properties for managing game cycles", function () { expect(StockMarket).toHaveProperty("storedCycles"); expect(StockMarket["storedCycles"]).toEqual(0); expect(StockMarket).toHaveProperty("lastUpdate"); expect(StockMarket["lastUpdate"]).toEqual(0); expect(StockMarket).toHaveProperty("ticksUntilCycle"); expect(typeof StockMarket["ticksUntilCycle"]).toBe("number"); }); }); describe("Deletion", function () { it("should set StockMarket to be an empty object", function () { expect(StockMarket).not.toEqual({}); deleteStockMarket(); expect(StockMarket).toEqual({ lastUpdate: 0, Orders: {}, storedCycles: 0, ticksUntilCycle: 0, }); }); }); describe("processStockPrices()", function () { beforeEach(function () { deleteStockMarket(); initStockMarket(); initSymbolToStockMap(); }); it("should store cycles until it actually processes", function () { expect(StockMarket["storedCycles"]).toEqual(0); processStockPrices(10); expect(StockMarket["storedCycles"]).toEqual(10); }); it("should trigger a price update when it has enough cycles", function () { // Get the initial prices const initialValues: Record = {}; for (const stockName in StockMarket) { const stock = StockMarket[stockName]; if (!(stock instanceof Stock)) { continue; } initialValues[stock.symbol] = { ...stock }; } // Don't know or care how many exact cycles are required StockMarket.lastUpdate = new Date().getTime() - 5e3; processStockPrices(1e9); // Both price and 'otlkMag' should be different for (const stockName in StockMarket) { const stock = StockMarket[stockName]; if (!(stock instanceof Stock)) { continue; } const initValue = initialValues[stock.symbol]; expect(initValue.price).not.toEqual(stock.price); if (initValue.otlkMag === stock.otlkMag && initValue.b === stock.b) { throw new Error("expected either price or otlkMag to be different"); } } }); }); }); describe("StockToSymbolMap", function () { beforeEach(function () { deleteStockMarket(); initStockMarket(); initSymbolToStockMap(); }); it("should map stock symbols to their corresponding Stock Objects", function () { for (const stockName in StockMarket) { const stock = StockMarket[stockName]; if (!(stock instanceof Stock)) { continue; } expect(SymbolToStockMap[stock.symbol]).toEqual(stock); } }); }); describe("Transaction Cost Calculator Functions", function () { describe("getBuyTransactionCost()", function () { it("should fail on invalid 'stock' argument", function () { const res = getBuyTransactionCost({} as Stock, 10, PositionType.Long); expect(res).toEqual(null); }); it("should fail on invalid 'shares' arg", function () { let res = getBuyTransactionCost(stock, NaN, PositionType.Long); expect(res).toEqual(null); res = getBuyTransactionCost(stock, -1, PositionType.Long); expect(res).toEqual(null); }); it("should properly evaluate LONG transactions", function () { const shares = ctorParams.shareTxForMovement / 2; const res = getBuyTransactionCost(stock, shares, PositionType.Long); expect(res).toEqual(shares * stock.getAskPrice() + commission); }); it("should properly evaluate SHORT transactions", function () { const shares = ctorParams.shareTxForMovement / 2; const res = getBuyTransactionCost(stock, shares, PositionType.Short); expect(res).toEqual(shares * stock.getBidPrice() + commission); }); it("should cap the 'shares' argument at the stock's maximum number of shares", function () { const maxRes = getBuyTransactionCost(stock, stock.maxShares, PositionType.Long); const exceedRes = getBuyTransactionCost(stock, stock.maxShares * 10, PositionType.Long); expect(maxRes).toEqual(exceedRes); }); }); describe("getSellTransactionGain()", function () { it("should fail on invalid 'stock' argument", function () { const res = getSellTransactionGain({} as Stock, 10, PositionType.Long); expect(res).toEqual(null); }); it("should fail on invalid 'shares' arg", function () { let res = getSellTransactionGain(stock, NaN, PositionType.Long); expect(res).toEqual(null); res = getSellTransactionGain(stock, -1, PositionType.Long); expect(res).toEqual(null); }); it("should properly evaluate LONG transactionst", function () { const shares = ctorParams.shareTxForMovement / 2; const res = getSellTransactionGain(stock, shares, PositionType.Long); const expected = shares * stock.getBidPrice() - commission; expect(res).toEqual(expected); }); it("should properly evaluate SHORT transactions", function () { // We need to set this property in order to calculate gains from short position stock.playerAvgShortPx = stock.price * 2; const shares = ctorParams.shareTxForMovement / 2; const res = getSellTransactionGain(stock, shares, PositionType.Short); const expected = shares * stock.playerAvgShortPx + shares * (stock.playerAvgShortPx - stock.getAskPrice()) - commission; expect(res).toEqual(expected); }); it("should cap the 'shares' argument at the stock's maximum number of shares", function () { const maxRes = getSellTransactionGain(stock, stock.maxShares, PositionType.Long); const exceedRes = getSellTransactionGain(stock, stock.maxShares * 10, PositionType.Long); expect(maxRes).toEqual(exceedRes); }); }); }); describe("Forecast Movement Processor Function", function () { // N = 1 is the original forecast function getNthForecast(origForecast: number, n: number): number { return origForecast - forecastChangePerPriceMovement * (n - 1); } function getNthForecastForecast(origForecastForecast: number, n: number): number { if (stock.otlkMagForecast > 50) { const expected = origForecastForecast - forecastChangePerPriceMovement * (n - 1) * (stock.mv / 100); return expected < 50 ? 50 : expected; } else if (stock.otlkMagForecast < 50) { const expected = origForecastForecast + forecastChangePerPriceMovement * (n - 1) * (stock.mv / 100); return expected > 50 ? 50 : expected; } else { return 50; } } describe("processTransactionForecastMovement() for buy transactions", function () { const noMvmtShares = Math.round(ctorParams.shareTxForMovement / 2.2); const mvmtShares = ctorParams.shareTxForMovement * 3 + noMvmtShares; it("should do nothing on invalid 'stock' argument", function () { const oldTracker = stock.shareTxUntilMovement; processTransactionForecastMovement({} as Stock, mvmtShares); expect(stock.shareTxUntilMovement).toEqual(oldTracker); }); it("should do nothing on invalid 'shares' arg", function () { const oldTracker = stock.shareTxUntilMovement; processTransactionForecastMovement(stock, NaN); expect(stock.shareTxUntilMovement).toEqual(oldTracker); processTransactionForecastMovement(stock, -1); expect(stock.shareTxUntilMovement).toEqual(oldTracker); }); it("should properly evaluate a LONG transaction that doesn't trigger a forecast movement", function () { const oldForecast = stock.otlkMag; processTransactionForecastMovement(stock, noMvmtShares); expect(stock.otlkMag).toEqual(oldForecast); expect(stock.shareTxUntilMovement).toEqual(stock.shareTxForMovement - noMvmtShares); }); it("should properly evaluate a SHORT transaction that doesn't trigger a forecast movement", function () { const oldForecast = stock.otlkMag; processTransactionForecastMovement(stock, noMvmtShares); expect(stock.otlkMag).toEqual(oldForecast); expect(stock.shareTxUntilMovement).toEqual(stock.shareTxForMovement - noMvmtShares); }); it("should properly evaluate LONG transactions that triggers forecast movements", function () { const oldForecast = stock.otlkMag; const oldForecastForecast = stock.otlkMagForecast; processTransactionForecastMovement(stock, mvmtShares); expect(stock.otlkMag).toEqual(getNthForecast(oldForecast, 4)); expect(stock.otlkMagForecast).toEqual(getNthForecastForecast(oldForecastForecast, 4)); expect(stock.shareTxUntilMovement).toEqual(stock.shareTxForMovement - noMvmtShares); }); it("should properly evaluate SHORT transactions that triggers forecast movements", function () { const oldForecast = stock.otlkMag; const oldForecastForecast = stock.otlkMagForecast; processTransactionForecastMovement(stock, mvmtShares); expect(stock.otlkMag).toEqual(getNthForecast(oldForecast, 4)); expect(stock.otlkMagForecast).toEqual(getNthForecastForecast(oldForecastForecast, 4)); expect(stock.shareTxUntilMovement).toEqual(stock.shareTxForMovement - noMvmtShares); }); it("should properly evaluate LONG transactions of exactly 'shareTxForMovement' shares", function () { const oldForecast = stock.otlkMag; const oldForecastForecast = stock.otlkMagForecast; processTransactionForecastMovement(stock, stock.shareTxForMovement); expect(stock.otlkMag).toEqual(getNthForecast(oldForecast, 2)); expect(stock.otlkMagForecast).toEqual(getNthForecastForecast(oldForecastForecast, 2)); expect(stock.shareTxUntilMovement).toEqual(stock.shareTxForMovement); }); it("should properly evaluate LONG transactions that total to 'shareTxForMovement' shares", function () { const oldForecast = stock.otlkMag; const oldForecastForecast = stock.otlkMagForecast; processTransactionForecastMovement(stock, Math.round(stock.shareTxForMovement / 2)); expect(stock.shareTxUntilMovement).toBeLessThan(stock.shareTxForMovement); processTransactionForecastMovement(stock, stock.shareTxUntilMovement); expect(stock.otlkMag).toEqual(getNthForecast(oldForecast, 2)); expect(stock.otlkMagForecast).toEqual(getNthForecastForecast(oldForecastForecast, 2)); expect(stock.shareTxUntilMovement).toEqual(stock.shareTxForMovement); }); it("should properly evaluate LONG transactions that are a multiple of 'shareTxForMovement' shares", function () { const oldForecast = stock.otlkMag; const oldForecastForecast = stock.otlkMagForecast; processTransactionForecastMovement(stock, 3 * stock.shareTxForMovement); expect(stock.otlkMag).toEqual(getNthForecast(oldForecast, 4)); expect(stock.otlkMagForecast).toEqual(getNthForecastForecast(oldForecastForecast, 4)); expect(stock.shareTxUntilMovement).toEqual(stock.shareTxForMovement); }); it("should properly evaluate SHORT transactions of exactly 'shareTxForMovement' shares", function () { const oldForecast = stock.otlkMag; const oldForecastForecast = stock.otlkMagForecast; processTransactionForecastMovement(stock, stock.shareTxForMovement); expect(stock.otlkMag).toEqual(getNthForecast(oldForecast, 2)); expect(stock.otlkMagForecast).toEqual(getNthForecastForecast(oldForecastForecast, 2)); expect(stock.shareTxUntilMovement).toEqual(stock.shareTxForMovement); }); it("should properly evaluate SHORT transactions that total to 'shareTxForMovement' shares", function () { const oldForecast = stock.otlkMag; const oldForecastForecast = stock.otlkMagForecast; processTransactionForecastMovement(stock, Math.round(stock.shareTxForMovement / 2)); expect(stock.shareTxUntilMovement).toBeLessThan(stock.shareTxForMovement); processTransactionForecastMovement(stock, stock.shareTxUntilMovement); expect(stock.otlkMag).toEqual(getNthForecast(oldForecast, 2)); expect(stock.otlkMagForecast).toEqual(getNthForecastForecast(oldForecastForecast, 2)); expect(stock.shareTxUntilMovement).toEqual(stock.shareTxForMovement); }); it("should properly evaluate SHORT transactions that are a multiple of 'shareTxForMovement' shares", function () { const oldForecast = stock.otlkMag; const oldForecastForecast = stock.otlkMagForecast; processTransactionForecastMovement(stock, 3 * stock.shareTxForMovement); expect(stock.otlkMag).toEqual(getNthForecast(oldForecast, 4)); expect(stock.otlkMagForecast).toEqual(getNthForecastForecast(oldForecastForecast, 4)); expect(stock.shareTxUntilMovement).toEqual(stock.shareTxForMovement); }); }); describe("processTransactionForecastMovement() for sell transactions", function () { const noMvmtShares = Math.round(ctorParams.shareTxForMovement / 2.2); const mvmtShares = ctorParams.shareTxForMovement * 3 + noMvmtShares; it("should do nothing on invalid 'stock' argument", function () { const oldTracker = stock.shareTxUntilMovement; processTransactionForecastMovement({} as Stock, mvmtShares); expect(stock.shareTxUntilMovement).toEqual(oldTracker); }); it("should do nothing on invalid 'shares' arg", function () { const oldTracker = stock.shareTxUntilMovement; processTransactionForecastMovement(stock, NaN); expect(stock.shareTxUntilMovement).toEqual(oldTracker); processTransactionForecastMovement(stock, -1); expect(stock.shareTxUntilMovement).toEqual(oldTracker); }); it("should properly evaluate a LONG transaction that doesn't trigger a price movement", function () { const oldForecast = stock.otlkMag; processTransactionForecastMovement(stock, noMvmtShares); expect(stock.otlkMag).toEqual(oldForecast); expect(stock.shareTxUntilMovement).toEqual(stock.shareTxForMovement - noMvmtShares); }); it("should properly evaluate a SHORT transaction that doesn't trigger a price movement", function () { const oldForecast = stock.otlkMag; processTransactionForecastMovement(stock, noMvmtShares); expect(stock.otlkMag).toEqual(oldForecast); expect(stock.shareTxUntilMovement).toEqual(stock.shareTxForMovement - noMvmtShares); }); it("should properly evaluate LONG transactions that trigger price movements", function () { const oldForecast = stock.otlkMag; processTransactionForecastMovement(stock, mvmtShares); expect(stock.otlkMag).toEqual(getNthForecast(oldForecast, 4)); expect(stock.shareTxUntilMovement).toEqual(stock.shareTxForMovement - noMvmtShares); }); it("should properly evaluate SHORT transactions that trigger price movements", function () { const oldForecast = stock.otlkMag; processTransactionForecastMovement(stock, mvmtShares); expect(stock.otlkMag).toEqual(getNthForecast(oldForecast, 4)); expect(stock.shareTxUntilMovement).toEqual(stock.shareTxForMovement - noMvmtShares); }); it("should properly evaluate LONG transactions of exactly 'shareTxForMovement' shares", function () { const oldForecast = stock.otlkMag; processTransactionForecastMovement(stock, stock.shareTxForMovement); expect(stock.otlkMag).toEqual(getNthForecast(oldForecast, 2)); expect(stock.shareTxUntilMovement).toEqual(stock.shareTxForMovement); }); it("should properly evaluate LONG transactions that total to 'shareTxForMovement' shares", function () { const oldForecast = stock.otlkMag; processTransactionForecastMovement(stock, Math.round(stock.shareTxForMovement / 2)); expect(stock.shareTxUntilMovement).toBeLessThan(stock.shareTxForMovement); processTransactionForecastMovement(stock, stock.shareTxUntilMovement); expect(stock.otlkMag).toEqual(getNthForecast(oldForecast, 2)); expect(stock.shareTxUntilMovement).toEqual(stock.shareTxForMovement); }); it("should properly evaluate LONG transactions that are a multiple of 'shareTxForMovement' shares", function () { const oldForecast = stock.otlkMag; processTransactionForecastMovement(stock, 3 * stock.shareTxForMovement); expect(stock.otlkMag).toEqual(getNthForecast(oldForecast, 4)); expect(stock.shareTxUntilMovement).toEqual(stock.shareTxForMovement); }); it("should properly evaluate SHORT transactions of exactly 'shareTxForMovement' shares", function () { const oldForecast = stock.otlkMag; processTransactionForecastMovement(stock, stock.shareTxForMovement); expect(stock.otlkMag).toEqual(getNthForecast(oldForecast, 2)); expect(stock.shareTxUntilMovement).toEqual(stock.shareTxForMovement); }); it("should properly evaluate SHORT transactions that total to 'shareTxForMovement' shares", function () { const oldForecast = stock.otlkMag; processTransactionForecastMovement(stock, Math.round(stock.shareTxForMovement / 2)); expect(stock.shareTxUntilMovement).toBeLessThan(stock.shareTxForMovement); processTransactionForecastMovement(stock, stock.shareTxUntilMovement); expect(stock.otlkMag).toEqual(getNthForecast(oldForecast, 2)); expect(stock.shareTxUntilMovement).toEqual(stock.shareTxForMovement); }); it("should properly evaluate SHORT transactions that are a multiple of 'shareTxForMovement' shares", function () { const oldForecast = stock.otlkMag; processTransactionForecastMovement(stock, 3 * stock.shareTxForMovement); expect(stock.otlkMag).toEqual(getNthForecast(oldForecast, 4)); expect(stock.shareTxUntilMovement).toEqual(stock.shareTxForMovement); }); }); }); describe("Transaction (Buy/Sell) Functions", function () { const suppressDialogOpt = { suppressDialog: true }; describe("buyStock()", function () { it("should fail for invalid arguments", function () { expect(buyStock({} as Stock, 1, null, suppressDialogOpt)).toEqual(false); expect(buyStock(stock, 0, null, suppressDialogOpt)).toEqual(false); expect(buyStock(stock, -1, null, suppressDialogOpt)).toEqual(false); expect(buyStock(stock, NaN, null, suppressDialogOpt)).toEqual(false); }); it("should fail if player doesn't have enough money", function () { Player.setMoney(0); expect(buyStock(stock, 1, null, suppressDialogOpt)).toEqual(false); }); it("should not allow for transactions that exceed the maximum shares", function () { const maxShares = stock.maxShares; expect(buyStock(stock, maxShares + 1, null, suppressDialogOpt)).toEqual(false); }); it("should return true and properly update stock properties for successful transactions", function () { const shares = 1e3; const cost = getBuyTransactionCost(stock, shares, PositionType.Long); expect(cost).not.toBeNull(); // Checked above // eslint-disable-next-line @typescript-eslint/no-non-null-assertion Player.setMoney(cost!); expect(buyStock(stock, shares, null, suppressDialogOpt)).toEqual(true); expect(stock.playerShares).toEqual(shares); expect(stock.playerAvgPx).toBeGreaterThan(0); expect(Player.money).toEqual(0); }); }); describe("sellStock()", function () { it("should fail for invalid arguments", function () { expect(sellStock({} as Stock, 1, null, suppressDialogOpt)).toEqual(false); expect(sellStock(stock, 0, null, suppressDialogOpt)).toEqual(false); expect(sellStock(stock, -1, null, suppressDialogOpt)).toEqual(false); expect(sellStock(stock, NaN, null, suppressDialogOpt)).toEqual(false); }); it("should fail if player doesn't have any shares", function () { Player.setMoney(0); expect(sellStock(stock, 1, null, suppressDialogOpt)).toEqual(false); }); it("should not allow for transactions that exceed the maximum shares", function () { const maxShares = stock.maxShares; expect(sellStock(stock, maxShares + 1, null, suppressDialogOpt)).toEqual(false); }); it("should return true and properly update stock properties for successful transactions", function () { const shares = 1e3; stock.playerShares = shares; stock.playerAvgPx = stock.price; const gain = getSellTransactionGain(stock, shares, PositionType.Long); Player.setMoney(0); expect(sellStock(stock, shares, null, suppressDialogOpt)).toEqual(true); expect(stock.playerShares).toEqual(0); expect(stock.playerAvgPx).toEqual(0); expect(Player.money).toEqual(gain); }); it("should cap the number of sharse sold to however many the player owns", function () { const attemptedShares = 2e3; const actualShares = 1e3; stock.playerShares = actualShares; stock.playerAvgPx = stock.price; const gain = getSellTransactionGain(stock, actualShares, PositionType.Long); Player.setMoney(0); expect(sellStock(stock, attemptedShares, null, suppressDialogOpt)).toEqual(true); expect(stock.playerShares).toEqual(0); expect(stock.playerAvgPx).toEqual(0); expect(Player.money).toEqual(gain); }); it("should properly update stock properties for partial transactions", function () { const shares = 1e3; const origPrice = stock.price; stock.playerShares = 2 * shares; stock.playerAvgPx = origPrice; const gain = getSellTransactionGain(stock, shares, PositionType.Long); Player.setMoney(0); expect(sellStock(stock, shares, null, suppressDialogOpt)).toEqual(true); expect(stock.playerShares).toEqual(shares); expect(stock.playerAvgPx).toEqual(origPrice); expect(Player.money).toEqual(gain); }); }); describe("shortStock()", function () { it("should fail for invalid arguments", function () { expect(shortStock({} as Stock, 1, null, suppressDialogOpt)).toEqual(false); expect(shortStock(stock, 0, null, suppressDialogOpt)).toEqual(false); expect(shortStock(stock, -1, null, suppressDialogOpt)).toEqual(false); expect(shortStock(stock, NaN, null, suppressDialogOpt)).toEqual(false); }); it("should fail if player doesn't have enough money", function () { Player.setMoney(0); expect(shortStock(stock, 1, null, suppressDialogOpt)).toEqual(false); }); it("should not allow for transactions that exceed the maximum shares", function () { const maxShares = stock.maxShares; expect(shortStock(stock, maxShares + 1, null, suppressDialogOpt)).toEqual(false); }); it("should return true and properly update stock properties for successful transactions", function () { const shares = 1e3; const cost = getBuyTransactionCost(stock, shares, PositionType.Short); expect(cost).not.toBeNull(); // Checked above // eslint-disable-next-line @typescript-eslint/no-non-null-assertion Player.setMoney(cost!); expect(shortStock(stock, shares, null, suppressDialogOpt)).toEqual(true); expect(stock.playerShortShares).toEqual(shares); expect(stock.playerAvgShortPx).toBeGreaterThan(0); expect(Player.money).toEqual(0); }); }); describe("sellShort()", function () { it("should fail for invalid arguments", function () { expect(sellShort({} as Stock, 1, null, suppressDialogOpt)).toEqual(false); expect(sellShort(stock, 0, null, suppressDialogOpt)).toEqual(false); expect(sellShort(stock, -1, null, suppressDialogOpt)).toEqual(false); expect(sellShort(stock, NaN, null, suppressDialogOpt)).toEqual(false); }); it("should fail if player doesn't have any shares", function () { Player.setMoney(0); expect(sellShort(stock, 1, null, suppressDialogOpt)).toEqual(false); }); it("should not allow for transactions that exceed the maximum shares", function () { const maxShares = stock.maxShares; expect(sellShort(stock, maxShares + 1, null, suppressDialogOpt)).toEqual(false); }); it("should return true and properly update stock properties for successful transactions", function () { const shares = 1e3; stock.playerShortShares = shares; stock.playerAvgShortPx = stock.price; const gain = getSellTransactionGain(stock, shares, PositionType.Short); Player.setMoney(0); expect(sellShort(stock, shares, null, suppressDialogOpt)).toEqual(true); expect(stock.playerShortShares).toEqual(0); expect(stock.playerAvgShortPx).toEqual(0); expect(Player.money).toEqual(gain); }); it("should cap the number of sharse sold to however many the player owns", function () { const attemptedShares = 2e3; const actualShares = 1e3; stock.playerShortShares = actualShares; stock.playerAvgShortPx = stock.price; const gain = getSellTransactionGain(stock, actualShares, PositionType.Short); Player.setMoney(0); expect(sellShort(stock, attemptedShares, null, suppressDialogOpt)).toEqual(true); expect(stock.playerShortShares).toEqual(0); expect(stock.playerAvgShortPx).toEqual(0); expect(Player.money).toEqual(gain); }); it("should properly update stock properties for partial transactions", function () { const shares = 1e3; const origPrice = stock.price; stock.playerShortShares = 2 * shares; stock.playerAvgShortPx = origPrice; const gain = getSellTransactionGain(stock, shares, PositionType.Short); Player.setMoney(0); expect(sellShort(stock, shares, null, suppressDialogOpt)).toEqual(true); expect(stock.playerShortShares).toEqual(shares); expect(stock.playerAvgShortPx).toEqual(origPrice); expect(Player.money).toEqual(gain); }); }); }); describe("Order Class", function () { it("should throw on invalid arguments", function () { function invalid1(): Order { return new Order({} as string, 1, 1, OrderType.LimitBuy, PositionType.Long); } function invalid2(): Order { return new Order("FOO", "z" as any as number, 0, OrderType.LimitBuy, PositionType.Short); } function invalid3(): Order { return new Order("FOO", 1, {} as number, OrderType.LimitBuy, PositionType.Short); } function invalid4(): Order { return new Order("FOO", 1, NaN, OrderType.LimitBuy, PositionType.Short); } function invalid5(): Order { return new Order("FOO", NaN, 0, OrderType.LimitBuy, PositionType.Short); } expect(invalid1).toThrow(); expect(invalid2).toThrow(); expect(invalid3).toThrow(); expect(invalid4).toThrow(); expect(invalid5).toThrow(); }); }); describe("Order Placing & Processing", function () { beforeEach(function () { expect(initStockMarket).not.toThrow(); expect(initSymbolToStockMap).not.toThrow(); // Create an order book for our mock stock StockMarket["Orders"][stock.symbol] = []; }); describe("placeOrder()", function () { it("should return false when it's called with invalid arguments", function () { const invalid1 = placeOrder({} as Stock, 1, 1, OrderType.LimitBuy, PositionType.Long); const invalid2 = placeOrder(stock, "foo" as any as number, 2, OrderType.LimitBuy, PositionType.Long); const invalid3 = placeOrder(stock, 1, "foo" as any as number, OrderType.LimitBuy, PositionType.Long); expect(invalid1).toEqual(false); expect(invalid2).toEqual(false); expect(invalid3).toEqual(false); expect(StockMarket["Orders"][stock.symbol]).toEqual([]); }); it("should return true and update the order book for valid arguments", function () { const res = placeOrder(stock, 1e3, 9e3, OrderType.LimitBuy, PositionType.Long); expect(res).toEqual(true); expect(StockMarket["Orders"][stock.symbol]).toHaveLength(1); const order = StockMarket["Orders"][stock.symbol][0]; expect(order).toBeInstanceOf(Order); expect(order.stockSymbol).toEqual(ctorParams.symbol); expect(order.shares).toEqual(1e3); expect(order.price).toEqual(9e3); expect(order.type).toEqual(OrderType.LimitBuy); expect(order.pos).toEqual(PositionType.Long); }); }); describe("cancelOrder()", function () { beforeEach(function () { StockMarket["Orders"][stock.symbol] = []; const res = placeOrder(stock, 1e3, 9e3, OrderType.LimitBuy, PositionType.Long); expect(res).toEqual(true); expect(StockMarket["Orders"][stock.symbol]).toHaveLength(1); }); it("returns true & removes an Order from the order book", function () { const order = StockMarket["Orders"][stock.symbol][0]; const res = cancelOrder({ order }); expect(res).toEqual(true); expect(StockMarket["Orders"][stock.symbol]).toHaveLength(0); }); it("should also work when passing in order parameters separately", function () { const res = cancelOrder({ stock, shares: 1e3, price: 9e3, type: OrderType.LimitBuy, pos: PositionType.Long, }); expect(res).toEqual(true); expect(StockMarket["Orders"][stock.symbol]).toHaveLength(0); }); it("should return false and do nothing when the specified order doesn't exist", function () { // Same parameters, but its a different object const order = new Order(stock.symbol, 1e3, 9e3, OrderType.LimitBuy, PositionType.Long); const res = cancelOrder({ order }); expect(res).toEqual(false); expect(StockMarket["Orders"][stock.symbol]).toHaveLength(1); const res2 = cancelOrder({ stock, shares: 999, price: 9e3, type: OrderType.LimitBuy, pos: PositionType.Long, }); expect(res2).toEqual(false); expect(StockMarket["Orders"][stock.symbol]).toHaveLength(1); }); }); describe("processOrders()", function () { let processOrdersRefs: IProcessOrderRefs; beforeEach(function () { expect(initStockMarket).not.toThrow(); expect(initSymbolToStockMap).not.toThrow(); StockMarket[stock.name] = stock; SymbolToStockMap[stock.symbol] = stock; StockMarket["Orders"][stock.symbol] = []; stock.playerShares = 1e3; stock.playerShortShares = 1e3; Player.setMoney(100e9); processOrdersRefs = { stockMarket: StockMarket, symbolToStockMap: SymbolToStockMap, }; }); function checkThatOrderExists(placeOrderRes?: boolean): void { if (typeof placeOrderRes === "boolean") { expect(placeOrderRes).toEqual(true); } expect(StockMarket["Orders"][stock.symbol]).toHaveLength(1); } function checkThatOrderExecuted(): void { expect(StockMarket["Orders"][stock.symbol]).toHaveLength(0); } it("should execute LONG Limit Buy orders when price <= order price", function () { const res = placeOrder(stock, 1e3, 9e3, OrderType.LimitBuy, PositionType.Long); checkThatOrderExists(res); stock.changePrice(9e3); processOrders(stock, OrderType.LimitBuy, PositionType.Long, processOrdersRefs); checkThatOrderExecuted(); expect(stock.playerShares).toEqual(2e3); }); it("should execute SHORT Limit Buy Orders when price >= order price", function () { const res = placeOrder(stock, 1e3, 11e3, OrderType.LimitBuy, PositionType.Short); checkThatOrderExists(res); stock.changePrice(11e3); processOrders(stock, OrderType.LimitBuy, PositionType.Short, processOrdersRefs); checkThatOrderExecuted(); expect(stock.playerShortShares).toEqual(2e3); }); it("should execute LONG Limit Sell Orders when price >= order price", function () { const res = placeOrder(stock, 1e3, 11e3, OrderType.LimitSell, PositionType.Long); checkThatOrderExists(res); stock.changePrice(11e3); processOrders(stock, OrderType.LimitSell, PositionType.Long, processOrdersRefs); checkThatOrderExecuted(); expect(stock.playerShares).toEqual(0); }); it("should execute SHORT Limit Sell Orders when price <= order price", function () { const res = placeOrder(stock, 1e3, 9e3, OrderType.LimitSell, PositionType.Short); checkThatOrderExists(res); stock.changePrice(9e3); processOrders(stock, OrderType.LimitSell, PositionType.Short, processOrdersRefs); checkThatOrderExecuted(); expect(stock.playerShortShares).toEqual(0); }); it("should execute LONG Stop Buy Orders when price >= order price", function () { const res = placeOrder(stock, 1e3, 11e3, OrderType.StopBuy, PositionType.Long); checkThatOrderExists(res); stock.changePrice(11e3); processOrders(stock, OrderType.StopBuy, PositionType.Long, processOrdersRefs); checkThatOrderExecuted(); expect(stock.playerShares).toEqual(2e3); }); it("should execute SHORT Stop Buy Orders when price <= order price", function () { const res = placeOrder(stock, 1e3, 9e3, OrderType.StopBuy, PositionType.Short); checkThatOrderExists(res); stock.changePrice(9e3); processOrders(stock, OrderType.StopBuy, PositionType.Short, processOrdersRefs); checkThatOrderExecuted(); expect(stock.playerShortShares).toEqual(2e3); }); it("should execute LONG Stop Sell Orders when price <= order price", function () { const res = placeOrder(stock, 1e3, 9e3, OrderType.StopSell, PositionType.Long); checkThatOrderExists(res); stock.changePrice(9e3); processOrders(stock, OrderType.StopSell, PositionType.Long, processOrdersRefs); checkThatOrderExecuted(); expect(stock.playerShares).toEqual(0); }); it("should execute SHORT Stop Sell Orders when price >= order price", function () { const res = placeOrder(stock, 1e3, 11e3, OrderType.StopSell, PositionType.Short); checkThatOrderExists(res); stock.changePrice(11e3); processOrders(stock, OrderType.StopSell, PositionType.Short, processOrdersRefs); checkThatOrderExecuted(); expect(stock.playerShortShares).toEqual(0); }); it("should execute immediately if their conditions are satisfied", function () { placeOrder(stock, 1e3, 11e3, OrderType.LimitBuy, PositionType.Long); checkThatOrderExecuted(); expect(stock.playerShares).toEqual(2e3); }); }); }); // TODO describe("Player Influencing", function () { const server = new Server({ hostname: "mockserver", moneyAvailable: 1e6, organizationName: "MockStock", }); const company = new Company({ name: "MockStock" as CompanyName, info: "", companyPositions: [], expMultiplier: 1, salaryMultiplier: 1, jobStatReqOffset: 1, }); beforeEach(function () { expect(initStockMarket).not.toThrow(); expect(initSymbolToStockMap).not.toThrow(); StockMarket[stock.name] = stock; }); describe("influenceStockThroughServerHack()", function () { it("should decrease a stock's second-order forecast when all of its money is hacked", function () { const oldSecondOrderForecast = stock.otlkMagForecast; influenceStockThroughServerHack(server, server.moneyMax); expect(stock.otlkMagForecast).toEqual(oldSecondOrderForecast - forecastForecastChangeFromHack); }); it("should not decrease the stock's second-order forecast when no money is stolen", function () { const oldSecondOrderForecast = stock.otlkMagForecast; influenceStockThroughServerHack(server, 0); expect(stock.otlkMagForecast).toEqual(oldSecondOrderForecast); }); }); describe("influenceStockThroughServerGrow()", function () { it("should increase a stock's second-order forecast when all of its money is grown", function () { const oldSecondOrderForecast = stock.otlkMagForecast; influenceStockThroughServerGrow(server, server.moneyMax); expect(stock.otlkMagForecast).toEqual(oldSecondOrderForecast + forecastForecastChangeFromHack); }); it("should not increase the stock's second-order forecast when no money is grown", function () { const oldSecondOrderForecast = stock.otlkMagForecast; influenceStockThroughServerGrow(server, 0); expect(stock.otlkMagForecast).toEqual(oldSecondOrderForecast); }); }); describe("influenceStockThroughCompanyWork()", function () { it("should increase the server's second order forecast", function () { const oldSecondOrderForecast = stock.otlkMagForecast; // Use 1e3 for numCycles to force a change // (This may break later if numbers are rebalanced); influenceStockThroughCompanyWork(company, 1, 500); expect(stock.otlkMagForecast).toEqual(oldSecondOrderForecast + forecastForecastChangeFromCompanyWork); }); it("should be affected by performanceMult", function () { const oldSecondOrderForecast = stock.otlkMagForecast; // Use 1e3 for numCycles to force a change // (This may break later if numbers are rebalanced); influenceStockThroughCompanyWork(company, 4, 1e3); expect(stock.otlkMagForecast).toEqual(oldSecondOrderForecast + 4 * forecastForecastChangeFromCompanyWork); }); }); }); });