Merge pull request #2

adiciona teste unitario

.gitignore

@@ -0,0 +1,2 @@
+.coverage
+coverage/

dados_publicos_cnpj/__main__.py

@@ -1,19 +1,10 @@
-import aiohttp
 import asyncio
 import click
 
-
-URL = "https://arquivos.receitafederal.gov.br/dados/cnpj/dados_abertos_cnpj/"
+from dados_publicos_cnpj.extract.util import check_url_availability
 
 
-async def check_url_availability(url):
-    async with aiohttp.ClientSession() as session:
-        try:
-            async with session.get(url) as response:
-                return 200 <= response.status < 300
-
-        except aiohttp.ClientError:
-            return False
+URL = "https://arquivos.receitafederal.gov.br/dados/cnpj/dados_abertos_cnpj/"
 
 
 @click.group()

dados_publicos_cnpj/extract/util.py

@@ -0,0 +1,11 @@
+import aiohttp
+
+
+async def check_url_availability(url):
+    async with aiohttp.ClientSession() as session:
+        try:
+            async with session.get(url) as response:
+                return 200 <= response.status < 300
+
+        except aiohttp.ClientError:
+            return False

makefile

@@ -1,3 +1,7 @@
 
 verifica:
 	@poetry run python -m dados_publicos_cnpj verifica
+
+unit_tests:
+	@poetry run pytest
+	@poetry run coverage html

pyproject.toml

@@ -11,6 +11,25 @@ aiohttp = "^3.11.2"
 click = "^8.1.7"
 
 
+[tool.poetry.group.dev.dependencies]
+pytest = "^8.3.3"
+pytest-cov = "^6.0.0"
+pytest-asyncio = "^0.24.0"
+
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
+
+[tool.pytest.ini_options]
+pythonpath = "."
+addopts = [
+    "-s",
+    "-x",
+    "-vv",
+    "--doctest-modules",
+    "--cov=dados_publicos_cnpj",
+    "--cov-report=term",
+]
+
+[tool.coverage.html]
+directory = "coverage"

tests/unit/extract/test_util.py

@@ -0,0 +1,10 @@
+import pytest
+
+from dados_publicos_cnpj.extract.util import check_url_availability
+
+
+@pytest.mark.asyncio
+async def test_check_url_availability():
+    url = "https://google.com"
+    result = await check_url_availability(url)
+    assert result