tests

Author: progrium

Makefile

@@ -20,6 +20,13 @@ deps:
 example: build
 	./build/Darwin/sshfront -d -p 2222 -k ~/.ssh/id_rsa example/helloworld
 
+test:
+	docker build -t $(NAME)-tests tests
+	docker run --rm \
+		-v $(PWD)/tests:/tests \
+		-v $(PWD)/build/Linux/sshfront:/bin/sshfront \
+		$(NAME)-tests \
+		basht /tests/*.bash
 
 release:
 	rm -rf release && mkdir release

tests/Dockerfile

@@ -0,0 +1,5 @@
+FROM alpine
+RUN apk --update add expect bash openssh curl \
+  && ssh-keygen -t rsa -N "" -f /root/.ssh/id_rsa \
+  && curl -Ls https://github.com/progrium/basht/releases/download/v0.1.0/basht_0.1.0_Linux_x86_64.tgz \
+    | tar -zxC /bin

tests/functional.bash

@@ -0,0 +1,80 @@
+
+write-script() {
+  declare script="$1"
+  echo "#!/bin/sh"
+  echo "$script"
+}
+
+setup-daemon() {
+  write-script "true" > /tmp/auth
+  write-script "true" > /tmp/handler
+  chmod +x /tmp/auth /tmp/handler
+  sshfront -a /tmp/auth /tmp/handler &
+  sleep 1
+  expect <<-EOF
+    set timeout 1
+    spawn ssh test@localhost
+    expect {
+      timeout   { exit 1 }
+      eof       { exit 1 }
+      "Are you sure" {
+        send "yes\r"
+        sleep 1
+        exit 0
+      }
+    }
+    exit 1
+EOF
+  echo
+}
+setup-daemon
+
+T_user-env() {
+  write-script "env" > /tmp/handler
+  expect <<-EOF
+    set timeout 1
+    spawn ssh foobar@localhost
+    expect {
+      timeout       { exit 1 }
+      eof           { exit 1 }
+      "USER=foobar" { exit 0 }
+    }
+    exit 1
+EOF
+}
+
+
+T_echo-handler() {
+  write-script 'echo $@' > /tmp/handler
+  expect <<-EOF
+    set timeout 1
+    spawn ssh test@localhost foo bar baz
+    expect {
+      timeout       { exit 1 }
+      eof           { exit 1 }
+      "foo bar baz" { exit 0 }
+    }
+    exit 1
+EOF
+}
+
+T_interactive-handler() {
+  write-script 'exec bash' > /tmp/handler
+  expect <<-EOF
+    set timeout 1
+    spawn ssh test@localhost
+    expect {
+      timeout       { exit 1 }
+      eof           { exit 1 }
+      "bash-4.3#" {
+        send "echo hello\r"
+        expect {
+          timeout   { exit 1 }
+          eof       { exit 1 }
+          "hello"   { exit 0 }
+        }
+      }
+    }
+    exit 1
+EOF
+}