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_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)