atlasexec: expose exit code (#68)

atlasexec/atlas.go

@@ -404,7 +404,7 @@ func (c *Client) SchemaInspect(ctx context.Context, params *SchemaInspectParams)
 		args = append(args, "--dev-url", params.DevURL)
 	}
 	switch {
-	case params.Format == "sql": 
+	case params.Format == "sql":
 		args = append(args, "--format", "{{ sql . }}")
 	case params.Format != "":
 		args = append(args, "--format", params.Format)
@@ -650,6 +650,14 @@ func (e *Error) Error() string {
 	return e.Stdout
 }
 
+// ExitCode returns the exit code of the command.
+func (e *Error) ExitCode() int {
+	if e.err == nil {
+		return new(exec.ExitError).ExitCode()
+	}
+	return e.err.ExitCode()
+}
+
 func (e *Error) Unwrap() error {
 	return e.err
 }

atlasexec/atlas_test.go

@@ -24,6 +24,13 @@ import (
 	"github.com/stretchr/testify/require"
 )
 
+func TestError(t *testing.T) {
+	err := atlasexec.Error{}
+	require.NotPanics(t, func() {
+		err.ExitCode()
+	})
+}
+
 func Test_NewClient(t *testing.T) {
 	execPath, err := exec.LookPath("atlas")
 	require.NoError(t, err)