old_contents: # -*- coding: utf-8 -*-
  from __future__ import unicode_literals
  from __future__ import print_function
  
  import os
  import fixture_utils as fixutils
  import docker_utils as dutils
  import steps.wrappers as wrappers
  
  
  DEBUG_ON_ERROR = False
  
  
  def before_all(context):
      """
      Set env parameters.
      """
      global DEBUG_ON_ERROR
      DEBUG_ON_ERROR = context.config.userdata.getbool('DEBUG_ON_ERROR')
  
      os.environ['LINES'] = "50"
      os.environ['COLUMNS'] = "120"
      os.environ['PAGER'] = 'cat'
  
      context.data_dir = fixutils.get_data_dir()
      context.fixture_lines = fixutils.read_fixture_files()
      context.client = dutils.init_docker_client()
      dutils.pull_required_images(context.client)
      context.exit_sent = False
      context.has_containers = False
  
  
  def after_scenario(context, _):
      """
      Cleans up after each test complete.
      """
      if hasattr(context, 'cli'):
          if context.has_containers:
              # force remove all containers that are still running.
              wrappers.expect_exact(context, 'wharfee> ')
              print('\nCleaning up containers...',)
              context.cli.sendline('rm -f --all')
              wrappers.expect_exact(context, ['Removed: ', 'There are no'])
              print('Cleaned up.')
              context.has_containers = False
  
          if not context.exit_sent:
              # Terminate the cli nicely.
              context.cli.terminate()
  
  
  def after_step(_, step):
      if DEBUG_ON_ERROR and step.status == 'failed':
          import ipdb
          ipdb.post_mortem(step.exc_traceback)
new_contents: # -*- coding: utf-8 -*-
  from __future__ import unicode_literals
  from __future__ import print_function
  
  import os
  import fixture_utils as fixutils
  import docker_utils as dutils
  import steps.wrappers as wrappers
  
  
  DEBUG_ON_ERROR = False
  
  
  def before_all(context):
      """
      Set env parameters.
      """
      global DEBUG_ON_ERROR
      DEBUG_ON_ERROR = context.config.userdata.getbool('DEBUG_ON_ERROR')
  
      os.environ['LINES'] = "50"
      os.environ['COLUMNS'] = "120"
      os.environ['PAGER'] = 'cat'
  
      context.data_dir = fixutils.get_data_dir()
      context.fixture_lines = fixutils.read_fixture_files()
      context.client = dutils.init_docker_client(timeout=10)
      dutils.pull_required_images(context.client)
      context.exit_sent = False
      context.has_containers = False
  
  
  def after_scenario(context, _):
      """
      Cleans up after each test complete.
      """
      if hasattr(context, 'cli'):
          if context.has_containers:
              # force remove all containers that are still running.
              wrappers.expect_exact(context, 'wharfee> ')
              print('\nCleaning up containers...',)
              context.cli.sendline('rm -f --all')
              wrappers.expect_exact(context, ['Removed: ', 'There are no'])
              print('Cleaned up.')
              context.has_containers = False
  
          if not context.exit_sent:
              # Terminate the cli nicely.
              context.cli.terminate()
  
  
  def after_step(_, step):
      if DEBUG_ON_ERROR and step.status == 'failed':
          import ipdb
          ipdb.post_mortem(step.exc_traceback)
subject: Increase timeout to 10 for tests.