[#77829] devices: linux-client: shell: Update tests

Signed-off-by: Anna Roszkiewicz <aroszkiewicz@antmicro.com>

Author: anroszkiewicz

devices/linux-client/shell/runner_test.go

@@ -16,7 +16,7 @@ func TestShellRunnerBasic(t *testing.T) {
 	sr, err := NewShellRunner(1)
 	assert.NoError(t, err)
 
-	_, err = sr.Spawn(TestSessionUuid)
+	_, err = sr.Spawn(TestSessionUuid, "")
 	assert.NoError(t, err)
 }
 
@@ -29,11 +29,11 @@ func TestShellRunnerLimits(t *testing.T) {
 	assert.NoError(t, err)
 
 	for i := 0; i < LimitToTest; i += 1 {
-		_, err = sr.Spawn(strconv.Itoa(i))
+		_, err = sr.Spawn(strconv.Itoa(i), "")
 		assert.NoError(t, err)
 	}
 
-	_, err = sr.Spawn(uuid.NewString())
+	_, err = sr.Spawn(uuid.NewString(), "")
 	if assert.Error(t, err) {
 		assert.Equal(t, ErrSessionLimitReached, err)
 	}
@@ -46,17 +46,17 @@ func TestShellRunnerLimitsIncrementedAfterTermination(t *testing.T) {
 	id1 := uuid.NewString()
 	id2 := uuid.NewString()
 
-	_, err = sr.Spawn(id1)
+	_, err = sr.Spawn(id1, "")
 	assert.NoError(t, err)
 
-	_, err = sr.Spawn(id2)
+	_, err = sr.Spawn(id2, "")
 	if assert.Error(t, err) {
 		assert.Equal(t, ErrSessionLimitReached, err)
 	}
 
 	err = sr.Terminate(id1)
 	assert.NoError(t, err)
 
-	_, err = sr.Spawn(id2)
+	_, err = sr.Spawn(id2, "")
 	assert.NoError(t, err)
 }

devices/linux-client/shell/session_test.go

@@ -1,31 +1,33 @@
 package shell
 
 import (
+	"os"
 	"testing"
 
 	"github.com/google/uuid"
 	"github.com/stretchr/testify/assert"
 )
 
 func TestSessionSpawn(t *testing.T) {
-	_, err := NewShellSession(uuid.NewString())
+	_, err := NewShellSession(uuid.NewString(), "")
 	assert.NoError(t, err)
 }
 
 func TestSessionClose(t *testing.T) {
-	session, _ := NewShellSession(uuid.NewString())
+	session, _ := NewShellSession(uuid.NewString(), "")
 	err := session.Close()
 	assert.NoError(t, err)
 }
 
 func TestSessionCannotFindShell(t *testing.T) {
 	DefaultShellList = []string{}
-	_, err := NewShellSession(uuid.NewString())
+	os.Setenv("SHELL", "/this/path/is/invalid")
+	_, err := NewShellSession(uuid.NewString(), "")
 	assert.ErrorIs(t, err, ErrNoShellAvailable)
 
 	DefaultShellList = []string{
 		"/this/path/will/not/exist",
 	}
-	_, err = NewShellSession(uuid.NewString())
+	_, err = NewShellSession(uuid.NewString(), "")
 	assert.ErrorIs(t, err, ErrNoShellAvailable)
 }