From 88b9707201dcb88c25a796990e4e74da3f9d1121 Mon Sep 17 00:00:00 2001
From: Andrei Stoica <andrei.stoica.365@gmail.com>
Date: Sat, 15 Oct 2022 14:27:40 -0400
Subject: [PATCH] added test for logging setup

---
 test/test_insert_sites.py | 37 +++++++++++++++++++++++++++++++++----
 1 file changed, 33 insertions(+), 4 deletions(-)

diff --git a/test/test_insert_sites.py b/test/test_insert_sites.py
index cba9585..2b01be0 100644
--- a/test/test_insert_sites.py
+++ b/test/test_insert_sites.py
@@ -1,10 +1,8 @@
-import inspect
 import json
 import os
 from recipe_graph import insert_sites
 from sqlalchemy import select
-from sqlalchemy.exc import SQLAlchemyError
-import sqlalchemy
+import logging
 
 import pytest
 
@@ -25,4 +23,35 @@ def json_file(json_data: list[dict]) -> str:
 
 def test_load_file(json_file: str, json_data):
     test_data = insert_sites.load_file(json_file)
-    assert test_data == json_data
\ No newline at end of file
+    assert test_data == json_data
+    
+
+def test_setup_argparser():
+    file_name = "test"
+    args = insert_sites.setup_argparser([file_name])
+    assert len(vars(args)) == 2
+    assert args.file == file_name
+    assert args.verbose == False
+
+
+    args = insert_sites.setup_argparser([file_name, "-v"])
+    assert args.file == file_name
+    assert args.verbose == True
+
+    args = insert_sites.setup_argparser([file_name, "--verbose"])
+    assert args.file == file_name
+    assert args.verbose == True
+
+def test_setup_logging():
+    args = insert_sites.setup_argparser(["test"])
+    logger = insert_sites.setup_logging(args)
+    assert logger.level == logging.WARNING
+
+    args = insert_sites.setup_argparser(["test", "-v"])
+    logger =  insert_sites.setup_logging(args)
+    assert logger.level == logging.INFO
+
+    args = insert_sites.setup_argparser(["test", "--verbose"])
+    logger =  insert_sites.setup_logging(args)
+    assert logger.level == logging.INFO
+    
\ No newline at end of file