sudacode/ghostfolio
1
0
Fork 0
Code Issues Pull Requests Actions Packages Releases Activity

Refactoring

Browse Source
This commit is contained in:
Thomas 2021-07-28 15:54:37 +02:00
parent 39cba0a8eb
commit e7ef1d426e
1 changed files with 25 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

67
apps/api/src/app/core/portfolio-calculator.spec.ts
View File

@ -15,32 +15,14 @@ import {
differenceInCalendarDays,
endOfDay,
isBefore,
isSameDay,
parse
} from 'date-fns';
function toYearMonthDay(date: Date) {
const year = date.getFullYear();
const month = date.getMonth() + 1;
const day = date.getDate();
return [year, month, day];
}
function dateEqual(date1: Date, date2: Date) {
const date1Converted = toYearMonthDay(date1);
const date2Converted = toYearMonthDay(date2);
return (
date1Converted[0] === date2Converted[0] &&
date1Converted[1] === date2Converted[1] &&
date1Converted[2] === date2Converted[2]
);
}
function mockGetValue(symbol: string, date: Date) {
const today = new Date();
if (symbol === 'VTI') {
if (dateEqual(today, date)) {
if (isSameDay(today, date)) {
return { marketPrice: 213.32 };
} else {
const startDate = parse('2019-02-01', 'yyyy-MM-dd', new Date());
@ -54,10 +36,10 @@ function mockGetValue(symbol: string, date: Date) {
} else if (symbol === 'AMZN') {
return { marketPrice: 2021.99 };
} else if (symbol === 'TSLA') {
if (dateEqual(parse('2021-07-26', 'yyyy-MM-dd', new Date()), date)) {
if (isSameDay(parse('2021-07-26', 'yyyy-MM-dd', new Date()), date)) {
return { marketPrice: 657.62 };
}
if (dateEqual(parse('2021-01-02', 'yyyy-MM-dd', new Date()), date)) {
if (isSameDay(parse('2021-01-02', 'yyyy-MM-dd', new Date()), date)) {
return { marketPrice: 666.66 };
}
return { marketPrice: 0 };
@ -72,9 +54,9 @@ jest.mock('@ghostfolio/api/app/core/current-rate.service', () => {
CurrentRateService: jest.fn().mockImplementation(() => {
return {
getValue: ({
currency,
date,
symbol,
currency,
userCurrency
}: GetValueParams) => {
return Promise.resolve(mockGetValue(symbol, date));
@ -144,13 +126,13 @@ describe('PortfolioCalculator', () => {
const orders = [
...ordersVTI,
{
currency: Currency.USD,
date: '2021-02-01',
name: 'Vanguard Total Stock Market Index Fund ETF Shares',
quantity: new Big('20'),
symbol: 'VTI',
type: OrderType.Buy,
unitPrice: new Big('197.15'),
currency: Currency.USD
unitPrice: new Big('197.15')
}
];
const portfolioCalculator = new PortfolioCalculator(
@ -166,12 +148,12 @@ describe('PortfolioCalculator', () => {
date: '2019-02-01',
items: [
{
currency: Currency.USD,
firstBuyDate: '2019-02-01',
investment: new Big('1443.8'),
name: 'Vanguard Total Stock Market Index Fund ETF Shares',
quantity: new Big('10'),
symbol: 'VTI',
investment: new Big('1443.8'),
currency: Currency.USD,
firstBuyDate: '2019-02-01',
transactionCount: 1
}
]
@ -180,12 +162,12 @@ describe('PortfolioCalculator', () => {
date: '2019-08-03',
items: [
{
currency: Currency.USD,
firstBuyDate: '2019-02-01',
investment: new Big('2923.7'),
name: 'Vanguard Total Stock Market Index Fund ETF Shares',
quantity: new Big('20'),
symbol: 'VTI',
investment: new Big('2923.7'),
currency: Currency.USD,
firstBuyDate: '2019-02-01',
transactionCount: 2
}
]
@ -194,12 +176,12 @@ describe('PortfolioCalculator', () => {
date: '2020-02-02',
items: [
{
currency: Currency.USD,
firstBuyDate: '2019-02-01',
investment: new Big('652.55'),
name: 'Vanguard Total Stock Market Index Fund ETF Shares',
quantity: new Big('5'),
symbol: 'VTI',
investment: new Big('652.55'),
currency: Currency.USD,
firstBuyDate: '2019-02-01',
transactionCount: 3
}
]
@ -208,12 +190,12 @@ describe('PortfolioCalculator', () => {
date: '2021-02-01',
items: [
{
currency: Currency.USD,
firstBuyDate: '2019-02-01',
investment: new Big('6627.05'),
name: 'Vanguard Total Stock Market Index Fund ETF Shares',
quantity: new Big('35'),
symbol: 'VTI',
investment: new Big('6627.05'),
currency: Currency.USD,
firstBuyDate: '2019-02-01',
transactionCount: 5
}
]
@ -222,12 +204,12 @@ describe('PortfolioCalculator', () => {
date: '2021-08-01',
items: [
{
currency: Currency.USD,
firstBuyDate: '2019-02-01',
investment: new Big('8403.95'),
name: 'Vanguard Total Stock Market Index Fund ETF Shares',
quantity: new Big('45'),
symbol: 'VTI',
investment: new Big('8403.95'),
currency: Currency.USD,
firstBuyDate: '2019-02-01',
transactionCount: 6
}
]
@ -239,13 +221,13 @@ describe('PortfolioCalculator', () => {
const orders = [
...ordersVTI,
{
currency: Currency.USD,
date: '2019-09-01',
name: 'Amazon.com, Inc.',
quantity: new Big('5'),
symbol: 'AMZN',
type: OrderType.Buy,
unitPrice: new Big('2021.99'),
currency: Currency.USD
unitPrice: new Big('2021.99')
}
];
const portfolioCalculator = new PortfolioCalculator(
@ -1463,6 +1445,7 @@ describe('PortfolioCalculator', () => {
});
});
});
const ordersMixedSymbols: PortfolioOrder[] = [
{
date: '2017-01-03',