test: add tests for fetch stock

seriaati · Oct 17, 2023 · 0e2b547 · 0e2b547
1 parent 5bd87f4
commit 0e2b547
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/test_crawl.py b/tests/test_crawl.py
@@ -91,6 +91,12 @@ async def test_fetch_stock():
     client = stock_crawl.StockCrawl()
     stock = await client.fetch_stock("2330")
     assert stock is not None
+    stock = await client.fetch_stock("台積電")
+    assert stock is not None
+    stock = await client.fetch_stock("0")
+    assert stock is None
+    stock = await client.fetch_stock("不存在")
+    assert stock is None
     await client.close()