import io

import spur
from nose.tools import istest, assert_equal

from toodlepip.consoles import Console


@istest
def console_writes_output_to_console():
    console, output = _create_local_console()
    console.run("Action", ["echo", "Go go go!"])
    assert b"Go go go!" in output.getvalue()


def _create_local_console():
    output = io.BytesIO()
    shell = spur.LocalShell()
    return Console(shell, output), output
    

import io

import spur
from nose.tools import istest, assert_equal

from toodlepip.consoles import Console


@istest
def console_writes_stdout_output_to_console():
    console, output = _create_local_console()
    console.run("Action", ["echo", "Go go go!"])
    assert b"Go go go!" in output.getvalue()


@istest
def console_writes_stderr_output_to_console():
    console, output = _create_local_console()
    console.run("Action", ["sh", "-c", "echo 'Go go go!' 1>&2"])
    assert b"Go go go!" in output.getvalue()


def _create_local_console():
    output = io.BytesIO()
    shell = spur.LocalShell()
    return Console(shell, output), output
    

Add test for stderr output from console
Add test for stderr output from console

  import io
  
  import spur
  from nose.tools import istest, assert_equal
  
  from toodlepip.consoles import Console
  
  
  @istest
- def console_writes_output_to_console():
+ def console_writes_stdout_output_to_console():
      console, output = _create_local_console()
      console.run("Action", ["echo", "Go go go!"])
+     assert b"Go go go!" in output.getvalue()
+ 
+ 
+ @istest
+ def console_writes_stderr_output_to_console():
+     console, output = _create_local_console()
+     console.run("Action", ["sh", "-c", "echo 'Go go go!' 1>&2"])
      assert b"Go go go!" in output.getvalue()
  
  
  def _create_local_console():
      output = io.BytesIO()
      shell = spur.LocalShell()
      return Console(shell, output), output
      
  
Add test for stderr output from console
## Code Before:
import io

import spur
from nose.tools import istest, assert_equal

from toodlepip.consoles import Console


@istest
def console_writes_output_to_console():
    console, output = _create_local_console()
    console.run("Action", ["echo", "Go go go!"])
    assert b"Go go go!" in output.getvalue()


def _create_local_console():
    output = io.BytesIO()
    shell = spur.LocalShell()
    return Console(shell, output), output
    

## Instruction:
Add test for stderr output from console
## Code After:
import io

import spur
from nose.tools import istest, assert_equal

from toodlepip.consoles import Console


@istest
def console_writes_stdout_output_to_console():
    console, output = _create_local_console()
    console.run("Action", ["echo", "Go go go!"])
    assert b"Go go go!" in output.getvalue()


@istest
def console_writes_stderr_output_to_console():
    console, output = _create_local_console()
    console.run("Action", ["sh", "-c", "echo 'Go go go!' 1>&2"])
    assert b"Go go go!" in output.getvalue()


def _create_local_console():
    output = io.BytesIO()
    shell = spur.LocalShell()
    return Console(shell, output), output
    
