Fixes config tests

features/steps/core.py

@@ -2,9 +2,7 @@ from behave import *
 from jrnl import cli, install, Journal, util
 from dateutil import parser as date_parser
 import os
-import sys
 import json
-import pytz
 import keyring
 keyring.set_keyring(keyring.backends.file.PlaintextKeyring())
 try:
@@ -13,8 +11,9 @@ except ImportError:
     from cStringIO import StringIO
 import tzlocal
 
+
 def _parse_args(command):
-    nargs=[]
+    nargs = []
     concats = []
     for a in command.split()[1:]:
         if a.startswith("'"):
@@ -27,16 +26,16 @@ def _parse_args(command):
             nargs.append(a)
     return nargs
 
+
 def read_journal(journal_name="default"):
-    with open(install.CONFIG_FILE_PATH) as config_file:
-        config = json.load(config_file)
+    config = util.load_config(install.CONFIG_FILE_PATH)
     with open(config['journals'][journal_name]) as journal_file:
         journal = journal_file.read()
     return journal
 
+
 def open_journal(journal_name="default"):
-    with open(install.CONFIG_FILE_PATH) as config_file:
-        config = json.load(config_file)
+    config = util.load_config(install.CONFIG_FILE_PATH)
     journal_conf = config['journals'][journal_name]
     if type(journal_conf) is dict:  # We can override the default config on a by-journal basis
         config.update(journal_conf)
@@ -44,11 +43,13 @@ def open_journal(journal_name="default"):
         config['journal'] = journal_conf
     return Journal.Journal(**config)
 
+
 @given('we use the config "{config_file}"')
 def set_config(context, config_file):
     full_path = os.path.join("features/configs", config_file)
     install.CONFIG_FILE_PATH = os.path.abspath(full_path)
 
+
 @when('we run "{command}" and enter')
 @when('we run "{command}" and enter "{inputs}"')
 def run_with_input(context, command, inputs=None):
@@ -62,6 +63,7 @@ def run_with_input(context, command, inputs=None):
     except SystemExit as e:
         context.exit_status = e.code
 
+
 @when('we run "{command}"')
 def run(context, command):
     args = _parse_args(command)
@@ -71,23 +73,28 @@ def run(context, command):
     except SystemExit as e:
         context.exit_status = e.code
 
+
 @when('we set the keychain password of "{journal}" to "{password}"')
 def set_keychain(context, journal, password):
     keyring.set_password('jrnl', journal, password)
 
+
 @then('we should get an error')
 def has_error(context):
     assert context.exit_status != 0, context.exit_status
 
+
 @then('we should get no error')
 def no_error(context):
     assert context.exit_status is 0, context.exit_status
 
+
 @then('the output should be parsable as json')
 def check_output_json(context):
     out = context.stdout_capture.getvalue()
     assert json.loads(out), out
 
+
 @then('"{field}" in the json output should have {number:d} elements')
 @then('"{field}" in the json output should have 1 element')
 def check_output_field(context, field, number=1):
@@ -96,6 +103,7 @@ def check_output_field(context, field, number=1):
     assert field in out_json, [field, out_json]
     assert len(out_json[field]) == number, len(out_json[field])
 
+
 @then('"{field}" in the json output should not contain "{key}"')
 def check_output_field_not_key(context, field, key):
     out = context.stdout_capture.getvalue()
@@ -103,6 +111,7 @@ def check_output_field_not_key(context, field, key):
     assert field in out_json
     assert key not in out_json[field]
 
+
 @then('"{field}" in the json output should contain "{key}"')
 def check_output_field_key(context, field, key):
     out = context.stdout_capture.getvalue()
@@ -120,6 +129,7 @@ def check_output(context, text=None):
     for line_text, line_out in zip(text, out):
         assert line_text.strip() == line_out.strip(), [line_text.strip(), line_out.strip()]
 
+
 @then('the output should contain "{text}" in the local time')
 def check_output_time_inline(context, text):
     out = context.stdout_capture.getvalue()
@@ -129,6 +139,7 @@ def check_output_time_inline(context, text):
     local_date = date.strftime("%Y-%m-%d %H:%M")
     assert local_date in out, local_date
 
+
 @then('the output should contain "{text}"')
 def check_output_inline(context, text):
     out = context.stdout_capture.getvalue()
@@ -136,6 +147,7 @@ def check_output_inline(context, text):
         out = out.decode('utf-8')
     assert text in out
 
+
 @then('the output should not contain "{text}"')
 def check_output_not_inline(context, text):
     out = context.stdout_capture.getvalue()
@@ -143,22 +155,26 @@ def check_output_not_inline(context, text):
         out = out.decode('utf-8')
     assert text not in out
 
+
 @then('we should see the message "{text}"')
 def check_message(context, text):
     out = context.messages.getvalue()
     assert text in out, [text, out]
 
+
 @then('we should not see the message "{text}"')
 def check_not_message(context, text):
     out = context.messages.getvalue()
     assert text not in out, [text, out]
 
+
 @then('the journal should contain "{text}"')
 @then('journal "{journal_name}" should contain "{text}"')
 def check_journal_content(context, text, journal_name="default"):
     journal = read_journal(journal_name)
     assert text in journal, journal
 
+
 @then('journal "{journal_name}" should not exist')
 def journal_doesnt_exist(context, journal_name="default"):
     with open(install.CONFIG_FILE_PATH) as config_file:
@@ -166,6 +182,7 @@ def journal_doesnt_exist(context, journal_name="default"):
     journal_path = config['journals'][journal_name]
     assert not os.path.exists(journal_path)
 
+
 @then('the config should have "{key}" set to "{value}"')
 @then('the config for journal "{journal}" should have "{key}" set to "{value}"')
 def config_var(context, key, value, journal=None):
@@ -175,21 +192,22 @@ def config_var(context, key, value, journal=None):
         "int": int,
         "str": str
     }[t](value)
-    with open(install.CONFIG_FILE_PATH) as config_file:
-        config = json.load(config_file)
+    config = util.load_config(install.CONFIG_FILE_PATH)
     if journal:
         config = config["journals"][journal]
     assert key in config
     assert config[key] == value
 
+
 @then('the journal should have {number:d} entries')
 @then('the journal should have {number:d} entry')
 @then('journal "{journal_name}" should have {number:d} entries')
 @then('journal "{journal_name}" should have {number:d} entry')
-def check_journal_content(context, number, journal_name="default"):
+def check_journal_entries(context, number, journal_name="default"):
     journal = open_journal(journal_name)
     assert len(journal.entries) == number
 
+
 @then('fail')
 def debug_fail(context):
     assert False