mirror of https://github.com/ghostfolio/ghostfolio
Thomas Kaul
3 days ago
committed by
GitHub
GitHub
15 changed files with 263 additions and 136 deletions
@ -0,0 +1,35 @@ |
|||||
|
const NodeEnvironment = require('jest-environment-node').TestEnvironment; |
||||
|
|
||||
|
// Jest gives each test file a copy of `process`, so a test cannot change the
|
||||
|
// time zone at run time. This environment sets `TZ` on the real process of the
|
||||
|
// worker, which resets the internal date cache of Node.
|
||||
|
class TimeZoneEnvironment extends NodeEnvironment { |
||||
|
constructor(config, context) { |
||||
|
super(config, context); |
||||
|
|
||||
|
this.previousTimeZone = process.env.TZ; |
||||
|
this.timeZone = config.projectConfig.testEnvironmentOptions?.timeZone; |
||||
|
} |
||||
|
|
||||
|
async setup() { |
||||
|
if (this.timeZone) { |
||||
|
process.env.TZ = this.timeZone; |
||||
|
} |
||||
|
|
||||
|
await super.setup(); |
||||
|
} |
||||
|
|
||||
|
async teardown() { |
||||
|
try { |
||||
|
await super.teardown(); |
||||
|
} finally { |
||||
|
if (this.previousTimeZone === undefined) { |
||||
|
delete process.env.TZ; |
||||
|
} else { |
||||
|
process.env.TZ = this.previousTimeZone; |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
module.exports = TimeZoneEnvironment; |
||||
@ -0,0 +1,70 @@ |
|||||
|
/** |
||||
|
* @jest-environment <rootDir>/jest-environment-tz.js |
||||
|
* @jest-environment-options {"timeZone": "America/New_York"} |
||||
|
*/ |
||||
|
import { DataGatheringItem } from '@ghostfolio/api/services/interfaces/interfaces'; |
||||
|
|
||||
|
import { Job } from 'bull'; |
||||
|
|
||||
|
import { DataGatheringProcessor } from './data-gathering.processor'; |
||||
|
|
||||
|
describe('DataGatheringProcessor in a time zone behind UTC', () => { |
||||
|
let dataGatheringProcessor: DataGatheringProcessor; |
||||
|
let dataProviderService: { getHistoricalRaw: jest.Mock }; |
||||
|
let marketDataService: { replaceForSymbol: jest.Mock; updateMany: jest.Mock }; |
||||
|
|
||||
|
beforeAll(() => { |
||||
|
// 2026-08-23 21:30 in New York, but already 2026-08-24 in UTC
|
||||
|
jest.useFakeTimers().setSystemTime(new Date('2026-08-24T01:30:00.000Z')); |
||||
|
}); |
||||
|
|
||||
|
beforeEach(() => { |
||||
|
dataProviderService = { |
||||
|
getHistoricalRaw: jest.fn().mockResolvedValue({ |
||||
|
'COINGECKO-bitcoin': { |
||||
|
'2026-08-21': { marketPrice: 5 }, |
||||
|
'2026-08-22': { marketPrice: 6 }, |
||||
|
'2026-08-23': { marketPrice: 7 } |
||||
|
} |
||||
|
}) |
||||
|
}; |
||||
|
marketDataService = { |
||||
|
replaceForSymbol: jest.fn(), |
||||
|
updateMany: jest.fn() |
||||
|
}; |
||||
|
|
||||
|
dataGatheringProcessor = new DataGatheringProcessor( |
||||
|
null, |
||||
|
dataProviderService as any, |
||||
|
marketDataService as any, |
||||
|
null |
||||
|
); |
||||
|
}); |
||||
|
|
||||
|
afterAll(() => { |
||||
|
jest.useRealTimers(); |
||||
|
}); |
||||
|
|
||||
|
it('gathers up to the last complete UTC day and does not shift the dates', async () => { |
||||
|
await dataGatheringProcessor.gatherHistoricalMarketData({ |
||||
|
data: { |
||||
|
dataSource: 'COINGECKO', |
||||
|
// The queue enqueues dates at midnight (UTC), not at local midnight
|
||||
|
date: new Date('2026-08-21T00:00:00.000Z').toISOString(), |
||||
|
symbol: 'bitcoin' |
||||
|
} |
||||
|
} as unknown as Job<DataGatheringItem>); |
||||
|
|
||||
|
const { data } = marketDataService.updateMany.mock.calls[0][0]; |
||||
|
|
||||
|
expect( |
||||
|
data.map(({ date, marketPrice }) => { |
||||
|
return { date, marketPrice }; |
||||
|
}) |
||||
|
).toEqual([ |
||||
|
{ date: new Date('2026-08-21T00:00:00.000Z'), marketPrice: 5 }, |
||||
|
{ date: new Date('2026-08-22T00:00:00.000Z'), marketPrice: 6 }, |
||||
|
{ date: new Date('2026-08-23T00:00:00.000Z'), marketPrice: 7 } |
||||
|
]); |
||||
|
}); |
||||
|
}); |
||||
Loading…
Reference in new issue