Move withcontext test to only run on linux as the sleep command is a unix command

Author: dylanhitt

command_linux_test.go

@@ -2,12 +2,14 @@ package cmd
 
 import (
 	"bytes"
-	"github.com/stretchr/testify/assert"
+	"context"
 	"io/ioutil"
 	"os"
 	"strings"
 	"testing"
 	"time"
+
+	"github.com/stretchr/testify/assert"
 )
 
 func TestCommand_ExecuteStderr(t *testing.T) {
@@ -138,3 +140,19 @@ func TestWithEnvironmentVariables(t *testing.T) {
 
 	assertEqualWithLineBreak(t, "value", c.Stdout())
 }
+
+func TestCommand_WithContext(t *testing.T) {
+	// ensure legacy timeout is honored
+	cmd := NewCommand("sleep 3;", WithTimeout(1*time.Second))
+	err := cmd.Execute()
+	assert.NotNil(t, err)
+	assert.Equal(t, "Command timed out after 1s", err.Error())
+
+	// set context timeout to 2 seconds to ensure
+	// context takes precedence over timeout
+	ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
+	defer cancel()
+	err = cmd.ExecuteContext(ctx)
+	assert.NotNil(t, err)
+	assert.Equal(t, "context deadline exceeded", err.Error())
+}

command_test.go

@@ -2,7 +2,6 @@ package cmd
 
 import (
 	"bytes"
-	"context"
 	"fmt"
 	"os"
 	"runtime"
@@ -131,19 +130,3 @@ func TestCommand_SetOptions(t *testing.T) {
 	assert.Equal(t, time.Duration(1000000000), c.Timeout)
 	assertEqualWithLineBreak(t, "test", writer.String())
 }
-
-func TestCommand_WithContext(t *testing.T) {
-	// ensure legacy timeout is honored
-	cmd := NewCommand("sleep 3;", WithTimeout(1*time.Second))
-	err := cmd.Execute()
-	assert.NotNil(t, err)
-	assert.Equal(t, "Command timed out after 1s", err.Error())
-
-	// set context timeout to 2 seconds to ensure
-	// context takes precedence over timeout
-	ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
-	defer cancel()
-	err = cmd.ExecuteContext(ctx)
-	assert.NotNil(t, err)
-	assert.Equal(t, "context deadline exceeded", err.Error())
-}