From c2f3aaf6e8b765bc53732af3777b232327e267fb Mon Sep 17 00:00:00 2001
From: Aleksey Lobanov <alekseylobanov1@gmail.com>
Date: Fri, 25 Mar 2016 19:08:50 +0300
Subject: [PATCH] Added basic tests for GetKeyNumsFromOutput

---
 src/main_test.go | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)
 create mode 100644 src/main_test.go

diff --git a/src/main_test.go b/src/main_test.go
new file mode 100644
index 0000000..c205567
--- /dev/null
+++ b/src/main_test.go
@@ -0,0 +1,32 @@
+// main_test.go
+package main
+
+import (
+	"reflect"
+	"testing"
+)
+
+func TestGetKeyNumsFromOutput(t *testing.T) {
+	var buf []byte // = {"key press   41k ey press   41"}
+	var keyNums []uint8
+
+	const test1 = "key press 36\nkey release 41\nkey press   41"
+	var result1 = []uint8{36, 41}
+
+	const test2 = ""
+	var result2 = []uint8{}
+
+	// Test1. Simple
+	buf = []byte(test1)
+	keyNums = GetKeyNumsFromOutput(buf)
+	if !reflect.DeepEqual(keyNums, result1) {
+		t.Fail()
+	}
+
+	// Test2. Clear
+	buf = []byte(test2)
+	keyNums = GetKeyNumsFromOutput(buf)
+	if !reflect.DeepEqual(keyNums, result2) {
+		t.Fail()
+	}
+}