From dfbf8703836171a299bca3df190bac89100e0c1d Mon Sep 17 00:00:00 2001
From: "Jonathan Leibiusky @xetorthio" <ionathan@gmail.com>
Date: Fri, 26 May 2017 10:01:38 -0300
Subject: [PATCH] Add test for `InstanceResizeTerminal`

---
 pwd/docker_mock_test.go |  8 ++++++--
 pwd/instance_test.go    | 35 +++++++++++++++++++++++++++++++++++
 2 files changed, 41 insertions(+), 2 deletions(-)
 create mode 100644 pwd/instance_test.go

diff --git a/pwd/docker_mock_test.go b/pwd/docker_mock_test.go
index b0720d4..9abfbbf 100644
--- a/pwd/docker_mock_test.go
+++ b/pwd/docker_mock_test.go
@@ -9,8 +9,9 @@ import (
 )
 
 type mockDocker struct {
-	createNetwork  func(string) error
-	connectNetwork func(container, network, ip string) (string, error)
+	createNetwork   func(string) error
+	connectNetwork  func(container, network, ip string) (string, error)
+	containerResize func(string, uint, uint) error
 }
 
 func (m *mockDocker) CreateNetwork(id string) error {
@@ -40,6 +41,9 @@ func (m *mockDocker) GetContainerStats(name string) (io.ReadCloser, error) {
 	return nil, nil
 }
 func (m *mockDocker) ContainerResize(name string, rows, cols uint) error {
+	if m.containerResize != nil {
+		return m.containerResize(name, rows, cols)
+	}
 	return nil
 }
 func (m *mockDocker) CreateAttachConnection(name string) (net.Conn, error) {
diff --git a/pwd/instance_test.go b/pwd/instance_test.go
new file mode 100644
index 0000000..99dda16
--- /dev/null
+++ b/pwd/instance_test.go
@@ -0,0 +1,35 @@
+package pwd
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestInstanceResizeTerminal(t *testing.T) {
+	resizedInstanceName := ""
+	resizedRows := uint(0)
+	resizedCols := uint(0)
+
+	docker := &mockDocker{}
+	docker.containerResize = func(name string, rows, cols uint) error {
+		resizedInstanceName = name
+		resizedRows = rows
+		resizedCols = cols
+
+		return nil
+	}
+
+	tasks := &mockTasks{}
+	broadcast := &mockBroadcast{}
+	storage := &mockStorage{}
+
+	p := NewPWD(docker, tasks, broadcast, storage)
+
+	err := p.InstanceResizeTerminal(&Instance{Name: "foobar"}, 24, 80)
+
+	assert.Nil(t, err)
+	assert.Equal(t, "foobar", resizedInstanceName)
+	assert.Equal(t, uint(24), resizedRows)
+	assert.Equal(t, uint(80), resizedCols)
+}