From 6aad0527bda68de43aca7cabda33bf89c5e76814 Mon Sep 17 00:00:00 2001
From: James Rasell <jamesrasell@gmail.com>
Date: Tue, 3 Dec 2019 08:01:33 +0000
Subject: [PATCH] Add initial tests for scale state.

---
 pkg/state/scale_test.go | 63 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 63 insertions(+)
 create mode 100644 pkg/state/scale_test.go

diff --git a/pkg/state/scale_test.go b/pkg/state/scale_test.go
new file mode 100644
index 0000000..7ea0eb7
--- /dev/null
+++ b/pkg/state/scale_test.go
@@ -0,0 +1,63 @@
+package state
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestScaleDirection_String(t *testing.T) {
+	testCases := []struct {
+		inputScaleDirection ScaleDirection
+		expectedOutput      string
+		name                string
+	}{
+		{
+			inputScaleDirection: ScaleDirectionIn,
+			expectedOutput:      "in",
+		},
+		{
+			inputScaleDirection: ScaleDirectionOut,
+			expectedOutput:      "out",
+		},
+		{
+			inputScaleDirection: ScaleDirectionNone,
+			expectedOutput:      "none",
+		},
+	}
+
+	for _, tc := range testCases {
+		actualOutput := tc.inputScaleDirection.String()
+		assert.Equal(t, tc.expectedOutput, actualOutput, tc.name)
+	}
+}
+
+func TestScaleStatus_String(t *testing.T) {
+	testCases := []struct {
+		inputScaleStatus ScaleStatus
+		expectedOutput   string
+		name             string
+	}{
+		{
+			inputScaleStatus: ScaleStatusStarted,
+			expectedOutput:   "started",
+		},
+		{
+			inputScaleStatus: ScaleStatusInProgress,
+			expectedOutput:   "in-progress",
+		},
+		{
+			inputScaleStatus: ScaleStatusCompleted,
+			expectedOutput:   "completed",
+		},
+		{
+			inputScaleStatus: ScaleStatusFailed,
+			expectedOutput:   "failed",
+		},
+	}
+
+	for _, tc := range testCases {
+		actualOutput := tc.inputScaleStatus.String()
+		assert.Equal(t, tc.expectedOutput, actualOutput, tc.name)
+	}
+}