Fix new stuff from tests to work with TreeNodes
This commit is contained in:
parent
0300da9ec1
commit
e9d7f5f2b6
@ -1,6 +1,6 @@
|
|||||||
from io import StringIO
|
from io import StringIO
|
||||||
|
|
||||||
from tree import TreeNode
|
from psh.tree import TreeNode
|
||||||
|
|
||||||
|
|
||||||
class BaseCommand(object):
|
class BaseCommand(object):
|
||||||
|
|||||||
@ -1,6 +1,6 @@
|
|||||||
from psh.commands import BaseCommand, register_cmd
|
from psh.commands import BaseCommand, register_cmd
|
||||||
|
|
||||||
from tree import TreeNode
|
from psh.tree import TreeNode
|
||||||
|
|
||||||
|
|
||||||
@register_cmd("example")
|
@register_cmd("example")
|
||||||
|
|||||||
@ -2,8 +2,7 @@ import shlex
|
|||||||
|
|
||||||
from psh.formatters import Printer
|
from psh.formatters import Printer
|
||||||
from psh.commands import BaseCommand
|
from psh.commands import BaseCommand
|
||||||
|
from psh.tree import TreeNode
|
||||||
from tree import TreeNode
|
|
||||||
|
|
||||||
|
|
||||||
class RawCommand(BaseCommand):
|
class RawCommand(BaseCommand):
|
||||||
|
|||||||
@ -14,7 +14,7 @@ class TestFormatter(BaseCommand):
|
|||||||
def call(self):
|
def call(self):
|
||||||
input_generator = self.get_input_generator()
|
input_generator = self.get_input_generator()
|
||||||
for line in input_generator:
|
for line in input_generator:
|
||||||
self.buffer.write(line.decode('utf-8'))
|
self.buffer.write(line.data.decode('utf-8'))
|
||||||
return None
|
return None
|
||||||
|
|
||||||
def get_data(self):
|
def get_data(self):
|
||||||
|
|||||||
Reference in New Issue
Block a user