fix tests

Author: pjz

mhi/main.py

@@ -1097,10 +1097,14 @@ def cmd_main():
     _cmd_dispatch(['mhi', cmd] + sys.argv[1:])
 
 
-def main():
+def init_config():
     global config, state
     cfgdir = os.environ.get('HOME', '')
     config = ConfigObj(infile=f"{cfgdir}/.mhirc", create_empty=True)
     state = ConfigObj(infile=f"{cfgdir}/.mhistate", create_empty=True)
+
+
+def main():
+    init_config()
     # main program
     _cmd_dispatch(sys.argv)

tests/test_consolidate.py

@@ -1,5 +1,5 @@
 
-from mhi import _consolidate
+from mhi.main import _consolidate
 
 def test_flatten():
     assert _consolidate([1,3,5,7,9]) == '1,3,5,7,9'

tests/test_fetch.py

@@ -7,13 +7,14 @@
 import email
 import imaplib
 
-import mhi
+from mhi import main as mhi
 #from click.testing import CliRunner # someday
-import six
 
 from imap4mock import ImapConnectionMock
 imaplib.IMAP4_SSL = ImapConnectionMock()
 
+mhi.init_config()
+
 reference_mail = {
     'headers': {
         'From': 'exampleFrom <[email protected]>',