Implement + test subshell expansion and statement evaluation
parent
ba1ad8f13a
commit
f6f3210165
@ -0,0 +1,89 @@
|
|||||||
|
package shell
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"bytes"
|
||||||
|
"io"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"git.snorba.art/hugo/nssh/commands/shell/builtins"
|
||||||
|
"github.com/stretchr/testify/suite"
|
||||||
|
)
|
||||||
|
|
||||||
|
type StatementTest struct {
|
||||||
|
suite.Suite
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *StatementTest) TestParse() {
|
||||||
|
parser := &BaseParser{&StatementParameters{}}
|
||||||
|
reader := bufio.NewReader(strings.NewReader("echo hello world!"))
|
||||||
|
|
||||||
|
iterator, err := NewCharIterator(reader)
|
||||||
|
t.NoError(err)
|
||||||
|
|
||||||
|
statement, err := parser.Parse(iterator, &CharCollection{})
|
||||||
|
t.ErrorIs(err, io.EOF)
|
||||||
|
|
||||||
|
t.IsType(&Word{}, statement.Tokens()[0])
|
||||||
|
t.IsType(&Whitespace{}, statement.Tokens()[1])
|
||||||
|
t.IsType(&Word{}, statement.Tokens()[2])
|
||||||
|
t.IsType(&Whitespace{}, statement.Tokens()[3])
|
||||||
|
t.IsType(&Word{}, statement.Tokens()[4])
|
||||||
|
|
||||||
|
t.Equal(5, len(statement.Tokens()))
|
||||||
|
|
||||||
|
t.Equal("echo", statement.Tokens()[0].String())
|
||||||
|
t.Equal(" ", statement.Tokens()[1].String())
|
||||||
|
t.Equal("hello", statement.Tokens()[2].String())
|
||||||
|
t.Equal(" ", statement.Tokens()[3].String())
|
||||||
|
t.Equal("world!", statement.Tokens()[4].String())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *StatementTest) TestEval() {
|
||||||
|
parser := &BaseParser{&StatementParameters{}}
|
||||||
|
reader := bufio.NewReader(strings.NewReader("echo hello world!"))
|
||||||
|
|
||||||
|
iterator, err := NewCharIterator(reader)
|
||||||
|
t.NoError(err)
|
||||||
|
|
||||||
|
statement, err := parser.Parse(iterator, &CharCollection{})
|
||||||
|
t.ErrorIs(err, io.EOF)
|
||||||
|
|
||||||
|
out := []byte{}
|
||||||
|
outBuf := bytes.NewBuffer(out)
|
||||||
|
state := NewShellState()
|
||||||
|
|
||||||
|
echo := &builtins.Echo{}
|
||||||
|
state.SetCommand(echo.Name(), echo)
|
||||||
|
|
||||||
|
state.Eval(statement.(Evalable), nil, outBuf, nil)
|
||||||
|
|
||||||
|
t.Equal("hello world!", string(outBuf.Bytes()))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *StatementTest) TestEvalWithSubshell() {
|
||||||
|
parser := &BaseParser{&StatementParameters{}}
|
||||||
|
reader := bufio.NewReader(strings.NewReader("echo \"$(echo hello world!)\""))
|
||||||
|
|
||||||
|
iterator, err := NewCharIterator(reader)
|
||||||
|
t.NoError(err)
|
||||||
|
|
||||||
|
statement, err := parser.Parse(iterator, &CharCollection{})
|
||||||
|
t.ErrorIs(err, io.EOF)
|
||||||
|
|
||||||
|
out := []byte{}
|
||||||
|
outBuf := bytes.NewBuffer(out)
|
||||||
|
state := NewShellState()
|
||||||
|
|
||||||
|
echo := &builtins.Echo{}
|
||||||
|
state.SetCommand(echo.Name(), echo)
|
||||||
|
|
||||||
|
state.Eval(statement.(Evalable), nil, outBuf, nil)
|
||||||
|
|
||||||
|
t.Equal("hello world!", string(outBuf.Bytes()))
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestStatementTest(t *testing.T) {
|
||||||
|
suite.Run(t, new(StatementTest))
|
||||||
|
}
|
@ -0,0 +1,59 @@
|
|||||||
|
package shell
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/suite"
|
||||||
|
)
|
||||||
|
|
||||||
|
type SubshellTest struct {
|
||||||
|
suite.Suite
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *SubshellTest) TestParse() {
|
||||||
|
parser := &BaseParser{&SubshellParameters{}}
|
||||||
|
reader := bufio.NewReader(strings.NewReader("(echo hello world!)"))
|
||||||
|
|
||||||
|
iterator, err := NewCharIterator(reader)
|
||||||
|
t.NoError(err)
|
||||||
|
|
||||||
|
token, err := parser.Parse(iterator, &CharCollection{})
|
||||||
|
t.NoError(err)
|
||||||
|
|
||||||
|
t.IsType(&Subshell{}, token)
|
||||||
|
t.IsType(&Statement{}, token.Tokens()[0])
|
||||||
|
t.Equal(1, len(token.Tokens()))
|
||||||
|
|
||||||
|
statement := token.Tokens()[0]
|
||||||
|
t.IsType(&Word{}, statement.Tokens()[0])
|
||||||
|
t.IsType(&Word{}, statement.Tokens()[2])
|
||||||
|
t.IsType(&Word{}, statement.Tokens()[4])
|
||||||
|
|
||||||
|
t.Equal(5, len(statement.Tokens()))
|
||||||
|
|
||||||
|
t.Equal("echo", statement.Tokens()[0].String())
|
||||||
|
t.Equal("hello", statement.Tokens()[2].String())
|
||||||
|
t.Equal("world!", statement.Tokens()[4].String())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *SubshellTest) TestParseMultipleStatements() {
|
||||||
|
parser := &BaseParser{&SubshellParameters{}}
|
||||||
|
reader := bufio.NewReader(strings.NewReader("(echo hello world!; echo tramp_exit_code $?)"))
|
||||||
|
|
||||||
|
iterator, err := NewCharIterator(reader)
|
||||||
|
t.NoError(err)
|
||||||
|
|
||||||
|
token, err := parser.Parse(iterator, &CharCollection{})
|
||||||
|
t.NoError(err)
|
||||||
|
|
||||||
|
t.Equal(2, len(token.Tokens()))
|
||||||
|
t.IsType(&Subshell{}, token)
|
||||||
|
t.IsType(&Statement{}, token.Tokens()[0])
|
||||||
|
t.IsType(&Statement{}, token.Tokens()[1])
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSubshellTest(t *testing.T) {
|
||||||
|
suite.Run(t, new(SubshellTest))
|
||||||
|
}
|
Loading…
Reference in New Issue