pikers
/
piker
3
0
Fork 0
Code Issues 2 Pull Requests 5 Packages Projects Releases Wiki Activity

Merge pull request #42 from pikers/multi_ticker_add 

Multi ticker add
kivy_mainline_and_py3.8
goodboy 2018-04-25 15:20:05 -04:00 committed by GitHub
parent ec2884a556 c8539b7373
commit 735fcb94c2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
piker/cli.py
View File

@ -206,11 +206,12 @@ def load(ctx, data):
@watchlists.command(help='add ticker to watchlist')
@click.argument('name', nargs=1, required=True)
@click.argument('ticker_name', nargs=1, required=True)
@click.argument('ticker_names', nargs=-1, required=True)
@click.pass_context
def add(ctx, name, ticker_name):
watchlist = wl.add_ticker(name, ticker_name,
ctx.obj['watchlist'])
def add(ctx, name, ticker_names):
for ticker in ticker_names:
watchlist = wl.add_ticker(
name, ticker, ctx.obj['watchlist'])
wl.write_to_file(watchlist, ctx.obj['path'])

12
tests/test_cli.py
View File

@ -150,9 +150,15 @@ def test_dump_watchlists(capfd, piker_dir, ex_watchlists):
assert out.strip() == expected_out
def test_ticker_added_to_watchlists(capfd, piker_dir, ex_watchlists):
ex_watchlists['pharma'].append('CRACK')
run(f"piker watchlists -d {piker_dir} add pharma CRACK")
@pytest.mark.parametrize(
'tickers', [('CRACK',), ('CRACK', 'SUIT',)]
)
def test_ticker_added_to_watchlists(capfd, piker_dir, ex_watchlists, tickers):
"""Verify that single or multi-ticker lists can be added.
"""
for ticker in tickers:
ex_watchlists['pharma'].append(ticker)
run(f"piker watchlists -d {piker_dir} add pharma {' '.join(tickers)}")
out = wl.ensure_watchlists(piker_dir)
assert out == ex_watchlists