Skip to content

Commit

Permalink
Improving error messages in windows command execution (#2054)
Browse files Browse the repository at this point in the history
improving error messages from executing commands on windows
  • Loading branch information
ramiro-gamarra authored and jpayne3506 committed Sep 11, 2023
1 parent 31aed4f commit 7d4fcf3
Show file tree
Hide file tree
Showing 2 changed files with 110 additions and 8 deletions.
18 changes: 10 additions & 8 deletions platform/os_windows.go
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,9 @@ import (
"time"

"github.com/Azure/azure-container-networking/log"
"github.com/Azure/azure-container-networking/platform/windows/adapter"
"github.com/Azure/azure-container-networking/platform/windows/adapter/mellanox"
"github.com/pkg/errors"
"golang.org/x/sys/windows"
)

Expand Down Expand Up @@ -97,20 +100,19 @@ func GetLastRebootTime() (time.Time, error) {
}

func (p *execClient) ExecuteCommand(command string) (string, error) {
log.Printf("[Azure-Utils] %s", command)
log.Printf("[Azure-Utils] ExecuteCommand: %q", command)

var stderr, stdout bytes.Buffer

var stderr bytes.Buffer
var out bytes.Buffer
cmd := exec.Command("cmd", "/c", command)
cmd.Stderr = &stderr
cmd.Stdout = &out
cmd.Stdout = &stdout

err := cmd.Run()
if err != nil {
return "", fmt.Errorf("%s:%s", err.Error(), stderr.String())
if err := cmd.Run(); err != nil {
return "", errors.Wrapf(err, "ExecuteCommand failed. stdout: %q, stderr: %q", stdout.String(), stderr.String())
}

return out.String(), nil
return stdout.String(), nil
}

func SetOutboundSNAT(subnet string) error {
Expand Down
100 changes: 100 additions & 0 deletions platform/os_windows_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,100 @@
package platform

import (
"errors"
"os/exec"
"testing"

"github.com/Azure/azure-container-networking/platform/windows/adapter/mocks"
"github.com/golang/mock/gomock"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)

var errTestFailure = errors.New("test failure")

// Test if hasNetworkAdapter returns false on actual error or empty adapter name(an error)
func TestHasNetworkAdapterReturnsError(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()

mockNetworkAdapter := mocks.NewMockNetworkAdapter(ctrl)
mockNetworkAdapter.EXPECT().GetAdapterName().Return("", errTestFailure)

result := hasNetworkAdapter(mockNetworkAdapter)
assert.False(t, result)
}

// Test if hasNetworkAdapter returns false on actual error or empty adapter name(an error)
func TestHasNetworkAdapterAdapterReturnsEmptyAdapterName(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()

mockNetworkAdapter := mocks.NewMockNetworkAdapter(ctrl)
mockNetworkAdapter.EXPECT().GetAdapterName().Return("Ethernet 3", nil)

result := hasNetworkAdapter(mockNetworkAdapter)
assert.True(t, result)
}

// Test if updatePriorityVLANTagIfRequired returns error on getting error on calling getpriorityvlantag
func TestUpdatePriorityVLANTagIfRequiredReturnsError(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()

mockNetworkAdapter := mocks.NewMockNetworkAdapter(ctrl)
mockNetworkAdapter.EXPECT().GetPriorityVLANTag().Return(0, errTestFailure)
result := updatePriorityVLANTagIfRequired(mockNetworkAdapter, 3)
assert.EqualError(t, result, "error while getting Priority VLAN Tag value: test failure")
}

// Test if updatePriorityVLANTagIfRequired returns nil if currentval == desiredvalue (SetPriorityVLANTag not being called)
func TestUpdatePriorityVLANTagIfRequiredIfCurrentValEqualDesiredValue(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()

mockNetworkAdapter := mocks.NewMockNetworkAdapter(ctrl)
mockNetworkAdapter.EXPECT().GetPriorityVLANTag().Return(4, nil)
result := updatePriorityVLANTagIfRequired(mockNetworkAdapter, 4)
assert.NoError(t, result)
}

// Test if updatePriorityVLANTagIfRequired returns nil if SetPriorityVLANTag being called to set value
func TestUpdatePriorityVLANTagIfRequiredIfCurrentValNotEqualDesiredValAndSetReturnsNoError(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()

mockNetworkAdapter := mocks.NewMockNetworkAdapter(ctrl)
mockNetworkAdapter.EXPECT().GetPriorityVLANTag().Return(1, nil)
mockNetworkAdapter.EXPECT().SetPriorityVLANTag(2).Return(nil)
result := updatePriorityVLANTagIfRequired(mockNetworkAdapter, 2)
assert.NoError(t, result)
}

// Test if updatePriorityVLANTagIfRequired returns error if SetPriorityVLANTag throwing error

func TestUpdatePriorityVLANTagIfRequiredIfCurrentValNotEqualDesiredValAndSetReturnsError(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()

mockNetworkAdapter := mocks.NewMockNetworkAdapter(ctrl)
mockNetworkAdapter.EXPECT().GetPriorityVLANTag().Return(1, nil)
mockNetworkAdapter.EXPECT().SetPriorityVLANTag(5).Return(errTestFailure)
result := updatePriorityVLANTagIfRequired(mockNetworkAdapter, 5)
assert.EqualError(t, result, "error while setting Priority VLAN Tag value: test failure")
}

func TestExecuteCommand(t *testing.T) {
out, err := NewExecClient().ExecuteCommand("dir")
require.NoError(t, err)
require.NotEmpty(t, out)
}

func TestExecuteCommandError(t *testing.T) {
_, err := NewExecClient().ExecuteCommand("dontaddtopath")
require.Error(t, err)

var xErr *exec.ExitError
assert.ErrorAs(t, err, &xErr)
assert.Equal(t, 1, xErr.ExitCode())
}

0 comments on commit 7d4fcf3

Please sign in to comment.