def test_linechart_data(example_ledger):
with app.test_request_context():
g.conversion = "units"
data = example_ledger.charts.linechart(
"Assets:Testing:MultipleCommodities")
assert data == [
{
"date": datetime.date(2000, 1, 1),
"balance": {
"USD": D("100")
}
},
{
"date": datetime.date(2000, 1, 2),
"balance": {
"XYZ": D("1"),
"USD": D("50")
},
},
{
"date": datetime.date(2000, 1, 3),
"balance": {
"USD": 0,
"ABC": D("1"),
"XYZ": D("1")
},
},
]
def test_scrape(setup_app):
urls = URLQueue()
current = urls.stack[0]
@app.url_defaults
def collect_urls(endpoint, values):
if 'loop' in values:
values.pop('loop')
return
urls.append(endpoint, values, current)
test_app = app.test_client()
while urls.stack:
print(current)
rv = test_app.get(current)
assert rv.status_code in [200, 302]
current = urls.pop()
for url_values in urls.all:
for filters in filter_combinations:
values = url_values[1].copy()
values.update(filters)
with app.test_request_context(current):
app.preprocess_request()
url = url_for(url_values[0], **values)
print(url)
rv = test_app.get(url)
assert rv.status_code in [200, 302]
def test_linechart_data(example_ledger):
with app.test_request_context('/'):
g.conversion = 'units'
data = example_ledger.charts.linechart(
'Assets:Testing:MultipleCommodities')
assert data == [
{
'date': datetime.date(2000, 1, 1),
'balance': {
'USD': D('100')
}
},
{
'date': datetime.date(2000, 1, 2),
'balance': {
'XYZ': D('1'),
'USD': D('50')
},
},
{
'date': datetime.date(2000, 1, 3),
'balance': {
'USD': 0,
'ABC': D('1'),
'XYZ': D('1')
},
},
]
def test_scrape(setup_app):
urls = URLQueue()
current = urls.stack[0]
@app.url_defaults
def collect_urls(endpoint, values):
if 'loop' in values:
values.pop('loop')
return
urls.append(endpoint, values, current)
test_app = app.test_client()
while urls.stack:
print(current)
rv = test_app.get(current)
assert rv.status_code in [200, 302]
current = urls.pop()
for url_values in urls.all:
for filters in filter_combinations:
values = url_values[1].copy()
values.update(filters)
with app.test_request_context(current):
app.preprocess_request()
url = url_for(url_values[0], **values)
print(url)
rv = test_app.get(url)
assert rv.status_code in [200, 302]
def test_net_worth(example_ledger):
with app.test_request_context():
app.preprocess_request()
g.conversion = "USD"
data = example_ledger.charts.net_worth(Interval.MONTH)
assert data[-18]["date"] == datetime.date(2015, 1, 1)
assert data[-18]["balance"]["USD"] == D("39125.34004")
assert data[-1]["date"] == datetime.date(2016, 5, 10)
assert data[-1]["balance"]["USD"] == D("102327.53144")
def test_hierarchy(example_ledger):
with app.test_request_context('/'):
app.preprocess_request()
data = example_ledger.charts.hierarchy('Assets')
assert data['balance_children'] == {
'IRAUSD': D('7200.00'),
'USD': D('94320.27840'),
'VACHR': D('-82')
}
assert data['balance'] == {}
# Assets:US:ETrade
etrade = data['children'][0]['children'][2]
assert etrade['children'][4]['balance'] == {'USD': D('4899.98')}
assert etrade['balance_children'] == {'USD': D('23137.54')}
def test_hierarchy(example_ledger):
with app.test_request_context('/'):
app.preprocess_request()
data = example_ledger.charts.hierarchy('Assets')
assert data['balance_children'] == {
'IRAUSD': D('7200.00'),
'USD': D('94320.27840'),
'VACHR': D('-82')
}
assert data['balance'] == {}
# Assets:US:ETrade
etrade = data['children'][0]['children'][2]
assert etrade['children'][4]['balance'] == {'USD': D('4899.98')}
assert etrade['balance_children'] == {'USD': D('23137.54')}
def test_hierarchy(example_ledger):
with app.test_request_context("/"):
app.preprocess_request()
data = example_ledger.charts.hierarchy("Assets")
assert data["balance_children"] == {
"IRAUSD": D("7200.00"),
"USD": D("94320.27840"),
"VACHR": D("-82"),
}
assert data["balance"] == {}
# Assets:US:ETrade
etrade = data["children"][0]["children"][2]
assert etrade["children"][4]["balance"] == {"USD": D("4899.98")}
assert etrade["balance_children"] == {"USD": D("23137.54")}
def append(self, endpoint, values, current):
if endpoint in ['source', 'document', 'add_document'] or \
'REPLACEME' in values.values():
return
real_endpoint = endpoint
if 'report_name' in values:
endpoint = values['report_name']
value_keys = frozenset([key for key, value in values.items() if value])
if value_keys not in self.seen_values[endpoint]:
self.seen_values[endpoint].add(value_keys)
with app.test_request_context(current):
app.preprocess_request()
url = url_for(real_endpoint, loop=True, **values)
if url not in self.seen:
self.seen.add(url)
self.stack.append(url)
def append(self, endpoint, values, current):
if (endpoint in ['source', 'document', 'add_document'] or
'REPLACEME' in values.values()):
return
real_endpoint = endpoint
if 'report_name' in values:
endpoint = values['report_name']
filters = ['account', 'interval', 'payee', 'tag', 'time']
value_keys = frozenset([key for key, value in values.items()
if value and key not in filters])
if value_keys not in self.seen_values[endpoint]:
self.seen_values[endpoint].add(value_keys)
with app.test_request_context(current):
app.preprocess_request()
url = url_for(real_endpoint, loop=True, **values)
self.all.append((real_endpoint, values))
self.stack.append(url)
def append(self, endpoint, values, current):
if (endpoint in ['source', 'document', 'add_document']
or 'REPLACEME' in values.values()):
return
real_endpoint = endpoint
if 'report_name' in values:
endpoint = values['report_name']
filters = ['account', 'interval', 'payee', 'tag', 'time']
value_keys = frozenset([
key for key, value in values.items()
if value and key not in filters
])
if value_keys not in self.seen_values[endpoint]:
self.seen_values[endpoint].add(value_keys)
with app.test_request_context(current):
app.preprocess_request()
url = url_for(real_endpoint, loop=True, **values)
self.all.append((real_endpoint, values))
self.stack.append(url)
def test_interval_totals(small_example_ledger):
with app.test_request_context(''):
g.conversion = None
data = small_example_ledger.charts.interval_totals(
Interval.MONTH, 'Expenses')
assert data == [{
'date': datetime.date(2012, 11, 18),
'balance': {
'EUR': D('280.00')
},
'budgets': Counter({
'EUR': D('31.42857142857142857142857145')
})
}, {
'date': datetime.date(2012, 12, 1),
'balance': {
'EUR': D('80.00')
},
'budgets': Counter({
'EUR': D('48.57142857142857142857142861')
})
}]
def test_interval_totals(small_example_ledger):
with app.test_request_context(''):
g.conversion = None
data = small_example_ledger.charts.interval_totals(
Interval.MONTH, 'Expenses')
assert data == [{
'date':
datetime.date(2012, 11, 18),
'balance': {
'EUR': D('280.00')
},
'budgets':
Counter({'EUR': D('31.42857142857142857142857145')})
}, {
'date':
datetime.date(2012, 12, 1),
'balance': {
'EUR': D('80.00')
},
'budgets':
Counter({'EUR': D('48.57142857142857142857142861')})
}]
def test_interval_totals(small_example_ledger):
with app.test_request_context(""):
g.conversion = None
data = small_example_ledger.charts.interval_totals(
Interval.MONTH, "Expenses")
assert data == [
{
"date": datetime.date(2012, 11, 18),
"balance": {
"EUR": D("280.00")
},
"budgets": Counter({"EUR":
D("31.42857142857142857142857145")}),
},
{
"date": datetime.date(2012, 12, 1),
"balance": {
"EUR": D("80.00")
},
"budgets": Counter({"EUR":
D("48.57142857142857142857142861")}),
},
]
def test_linechart_data(example_ledger):
with app.test_request_context('/'):
g.conversion = 'units'
data = example_ledger.charts.linechart(
'Assets:Testing:MultipleCommodities')
assert data == [{
'date': datetime.date(2000, 1, 1),
'balance': {
'USD': D('100')
},
}, {
'date': datetime.date(2000, 1, 2),
'balance': {
'XYZ': D('1'),
'USD': D('50')
},
}, {
'date': datetime.date(2000, 1, 3),
'balance': {
'USD': 0,
'ABC': D('1'),
'XYZ': D('1')
},
}]