74 lines
1.7 KiB
Go
74 lines
1.7 KiB
Go
package sys_test
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/tetratelabs/wazero/internal/testing/require"
|
|
"github.com/tetratelabs/wazero/sys"
|
|
)
|
|
|
|
type notExitError struct {
|
|
exitCode uint32
|
|
}
|
|
|
|
func (e *notExitError) Error() string {
|
|
return "not exit error"
|
|
}
|
|
|
|
func TestIs(t *testing.T) {
|
|
err := sys.NewExitError(2)
|
|
tests := []struct {
|
|
name string
|
|
target error
|
|
matches bool
|
|
}{
|
|
{
|
|
name: "same object",
|
|
target: err,
|
|
matches: true,
|
|
},
|
|
{
|
|
name: "different exit code",
|
|
target: sys.NewExitError(1),
|
|
matches: false,
|
|
},
|
|
{
|
|
name: "different type",
|
|
target: ¬ExitError{
|
|
exitCode: 2,
|
|
},
|
|
matches: false,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
tc := tt
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
matches := errors.Is(err, tc.target)
|
|
require.Equal(t, tc.matches, matches)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestExitError_Error(t *testing.T) {
|
|
t.Run("timeout", func(t *testing.T) {
|
|
err := sys.NewExitError(sys.ExitCodeDeadlineExceeded)
|
|
require.Equal(t, sys.ExitCodeDeadlineExceeded, err.ExitCode())
|
|
require.EqualError(t, err, "module closed with context deadline exceeded")
|
|
require.ErrorIs(t, err, context.DeadlineExceeded, "exit code context deadline exceeded should work")
|
|
})
|
|
t.Run("cancel", func(t *testing.T) {
|
|
err := sys.NewExitError(sys.ExitCodeContextCanceled)
|
|
require.Equal(t, sys.ExitCodeContextCanceled, err.ExitCode())
|
|
require.EqualError(t, err, "module closed with context canceled")
|
|
require.ErrorIs(t, err, context.Canceled, "exit code context canceled should work")
|
|
})
|
|
t.Run("normal", func(t *testing.T) {
|
|
err := sys.NewExitError(123)
|
|
require.Equal(t, uint32(123), err.ExitCode())
|
|
require.EqualError(t, err, "module closed with exit_code(123)")
|
|
})
|
|
}
|