refactor tests, add utils

test_extract_otp_secret_keys_unittest.py

@@ -19,9 +19,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
 import unittest
-import csv
-import json
-import os
+from utils import read_csv, read_json, remove_file
 
 import extract_otp_secret_keys
 
@@ -55,25 +53,5 @@ class TestExtract(unittest.TestCase):
         remove_file('test_example_output.json')
 
 
-def remove_file(filename):
-    if os.path.exists(filename): os.remove(filename)
-
-
-def read_csv(filename):
-    """Returns a list of lines."""
-    with open(filename, "r") as infile:
-        lines = []
-        reader = csv.reader(infile)
-        for line in reader:
-            lines.append(line)
-        return lines
-
-
-def read_json(filename):
-    """Returns a list or a dictionary."""
-    with open(filename, "r") as infile:
-        return json.load(infile)
-
-
 if __name__ == '__main__':
     unittest.main()