From 603c65063b18fdaee251d40bdbe88df2f49735ef Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?S=C3=B8ren=20L=2E=20Hansen?= <sorenisanerd@gmail.com>
Date: Fri, 4 Jun 2021 13:34:26 -0700
Subject: [PATCH] Add backend tests

---
 backend/localcommand/local_command_test.go | 88 ++++++++++++++++++++++
 1 file changed, 88 insertions(+)
 create mode 100644 backend/localcommand/local_command_test.go

diff --git a/backend/localcommand/local_command_test.go b/backend/localcommand/local_command_test.go
new file mode 100644
index 0000000..ddda029
--- /dev/null
+++ b/backend/localcommand/local_command_test.go
@@ -0,0 +1,88 @@
+package localcommand
+
+import (
+	"bytes"
+	"reflect"
+	"testing"
+	"time"
+)
+
+func TestNewFactory(t *testing.T) {
+	factory, err := NewFactory("/bin/false", []string{}, &Options{CloseSignal: 123, CloseTimeout: 321})
+	if err != nil {
+		t.Errorf("NewFactory() returned error")
+		return
+	}
+	if factory.command != "/bin/false" {
+		t.Errorf("factory.command = %v, expected %v", factory.command, "/bin/false")
+	}
+	if !reflect.DeepEqual(factory.argv, []string{}) {
+		t.Errorf("factory.argv = %v, expected %v", factory.argv, []string{})
+	}
+	if !reflect.DeepEqual(factory.options, &Options{CloseSignal: 123, CloseTimeout: 321}) {
+		t.Errorf("factory.options = %v, expected %v", factory.options, &Options{})
+	}
+
+	slave, _ := factory.New(nil)
+	lcmd := slave.(*LocalCommand)
+	if lcmd.closeSignal != 123 {
+		t.Errorf("lcmd.closeSignal = %v, expected %v", lcmd.closeSignal, 123)
+	}
+	if lcmd.closeTimeout != time.Second*321 {
+		t.Errorf("lcmd.closeTimeout = %v, expected %v", lcmd.closeTimeout, time.Second*321)
+	}
+}
+
+func TestFactoryNew(t *testing.T) {
+	factory, err := NewFactory("/bin/cat", []string{}, &Options{})
+	if err != nil {
+		t.Errorf("NewFactory() returned error")
+		return
+	}
+
+	slave, err := factory.New(nil)
+	if err != nil {
+		t.Errorf("factory.New() returned error")
+		return
+	}
+
+	writeBuf := []byte("foobar\n")
+	n, err := slave.Write(writeBuf)
+	if err != nil {
+		t.Errorf("write() failed: %v", err)
+		return
+	}
+	if n != 7 {
+		t.Errorf("Unexpected write length. n = %d, expected n = %d", n, 7)
+		return
+	}
+
+	// Local echo is on, so we get the output twice:
+	// Once because we're "typing" it, and once more
+	// repeated back to us by `cat`. Also, \r\n
+	// because we're a terminal.
+	expectedBuf := []byte("foobar\r\nfoobar\r\n")
+	readBuf := make([]byte, 1024)
+	var totalRead int
+	for totalRead < 16 {
+		n, err = slave.Read(readBuf[totalRead:])
+		if err != nil {
+			t.Errorf("read() failed: %v", err)
+			return
+		}
+		totalRead += n
+	}
+	if totalRead != 16 {
+		t.Errorf("Unexpected read length. totalRead = %d, expected totalRead = %d", totalRead, 16)
+		return
+	}
+	if !bytes.Equal(readBuf[:totalRead], expectedBuf) {
+		t.Errorf("unexpected output from slave: got %v, expected %v", readBuf[:totalRead], expectedBuf)
+	}
+	err = slave.Close()
+	if err != nil {
+		t.Errorf("close() failed: %v", err)
+		return
+	}
+
+}