Skip to content

Commit

Permalink
assert called once is a function these days
Browse files Browse the repository at this point in the history
  • Loading branch information
YaphetKG committed Jan 3, 2024
1 parent 0a7c368 commit a863d38
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions tests/unit/test_cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,31 +25,31 @@ def test_dug_cli_parser():
@patch('dug.cli.crawl')
def test_dug_cli_main_crawl(mock_crawl):
main(["crawl", "somefile.csv", "--parser", "topmedtag"])
assert mock_crawl.called_once()
mock_crawl.assert_called_once()

@mark.cli
@patch('dug.cli.crawl')
def test_dug_cli_main_extract_dug_elements(mock_crawl):
main(["crawl", "somefile.csv", "--parser", "topmedtag", "-x"])
assert mock_crawl.called_once()
mock_crawl.assert_called_once()
assert mock_crawl.call_args_list[0].args[0].extract_dug_elements

@mark.cli
@patch('dug.cli.crawl')
def test_dug_cli_main_extract_dug_elements_none(mock_crawl):
main(["crawl", "somefile.csv", "--parser", "topmedtag"])
assert mock_crawl.called_once()
mock_crawl.assert_called_once()
assert not mock_crawl.call_args_list[0].args[0].extract_dug_elements

@mark.cli
@patch('dug.cli.crawl')
def test_dug_cli_main_annotator(mock_crawl):
main(["crawl", "somefile.csv","--parser", "topmedtag", "--annotator", "annotator-monarch"])
assert mock_crawl.called_once()
mock_crawl.assert_called_once()

@mark.cli
@patch('dug.cli.search')
def test_dug_cli_main_search(mock_search):
# mock_search.search.return_value = "Searching!"
main(["search", "-q", "heart attack", "-t", "variables", "-k", "namespace=default"])
assert mock_search.called_once()
mock_search.assert_called_once()

0 comments on commit a863d38

Please sign in to comment.