Mock stdin

This commit is contained in:
Manuel Ebert 2013-07-19 12:44:39 +02:00
parent d4cbd4f944
commit 6abd7d5b68
5 changed files with 43 additions and 44 deletions

View file

@ -4,7 +4,7 @@ Feature: Basic reading and writing to a journal
Given we use the config "basic.json"
When we run "jrnl -n 2"
Then we should get no error
And the output should be
and the output should be
"""
2013-06-09 15:39 My first entry.
| Everything is alright
@ -27,3 +27,10 @@ Feature: Basic reading and writing to a journal
When we run "jrnl -n 1"
Then the output should contain "🌞"
and the output should contain "🐘"
Scenario: Writing an entry at the prompt
Given we use the config "basic.json"
When we run "jrnl" and enter "25 jul 2013: I saw Elvis. He's alive."
Then we should get no error
and the journal should contain "2013-07-25 09:00 I saw Elvis."
and the journal should contain "He's alive."

View file

@ -1,16 +1,36 @@
from behave import *
from jrnl import Journal, jrnl
from jrnl import jrnl
import os
import sys
import json
import StringIO
def read_journal(journal_name="default"):
with open(jrnl.CONFIG_PATH) as config_file:
config = json.load(config_file)
with open(config['journals'][journal_name]) as journal_file:
journal = journal_file.read()
return journal
@given('we use the config "{config_file}"')
def set_config(context, config_file):
full_path = os.path.join("features/configs", config_file)
jrnl.CONFIG_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):
text = inputs or context.text
args = command.split()[1:]
buffer = StringIO.StringIO(text.strip())
jrnl.util.STDIN = buffer
jrnl.cli(args)
@when('we run "{command}"')
def run(context, command):
args = command.split()[1:]
jrnl.cli(args)
jrnl.cli(args or None)
@then('we should get no error')
def no_error(context):
@ -24,7 +44,14 @@ def check_output(context):
assert line_text.strip() == line_out.strip()
@then('the output should contain "{text}"')
def check_output(context, text):
def check_output_inline(context, text):
out = context.stdout_capture.getvalue()
print out
assert text in 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