mirror of
https://github.com/onyx-and-iris/vbantxt.git
synced 2024-11-22 09:50:50 +00:00
43 lines
750 B
Go
43 lines
750 B
Go
package vbantxt_test
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
_ "embed"
|
|
"testing"
|
|
|
|
"github.com/onyx-and-iris/vbantxt"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
//go:embed testdata/vm.txt
|
|
var vm []byte
|
|
|
|
//go:embed testdata/matrix.txt
|
|
var matrix []byte
|
|
|
|
func run(t *testing.T, client *vbantxt.VbanTxt, script []byte) {
|
|
t.Helper()
|
|
|
|
r := bytes.NewReader(script)
|
|
scanner := bufio.NewScanner(r)
|
|
for scanner.Scan() {
|
|
err := client.Send(scanner.Text())
|
|
require.NoError(t, err)
|
|
}
|
|
}
|
|
|
|
func TestSendVm(t *testing.T) {
|
|
client, err := vbantxt.New("vm.local", 6980, "onyx")
|
|
require.NoError(t, err)
|
|
|
|
run(t, client, vm)
|
|
}
|
|
|
|
func TestSendMatrix(t *testing.T) {
|
|
client, err := vbantxt.New("vm.local", 6990, "onyx")
|
|
require.NoError(t, err)
|
|
|
|
run(t, client, matrix)
|
|
}
|