Add a test for operation commands (#121)

pkg/cmd/resource/operation_test.go

@@ -1,9 +1,13 @@
 package resource
 
 import (
+	"io/ioutil"
+	"net/http"
+	"net/http/httptest"
 	"testing"
 
 	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
 	"github.com/stretchr/testify/require"
 
 	"github.com/stripe/stripe-cli/pkg/config"
@@ -24,3 +28,37 @@ func TestNewOperationCmd(t *testing.T) {
 	require.Equal(t, "operation", val)
 	require.Contains(t, oc.Cmd.UsageTemplate(), "<id>")
 }
+
+func TestRunOperationCmd(t *testing.T) {
+	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.WriteHeader(http.StatusOK)
+		body, err := ioutil.ReadAll(r.Body)
+		require.Nil(t, err)
+
+		require.Equal(t, http.MethodPost, r.Method)
+		require.Equal(t, "/v1/bars/bar_123", r.URL.Path)
+		require.Equal(t, "Bearer sk_test_1234", r.Header.Get("Authorization"))
+		require.Equal(t, "param1=value1&param2=value2", string(body))
+	}))
+	defer ts.Close()
+
+	viper.Reset()
+	viper.Set("api_key", "sk_test_1234")
+	parentCmd := &cobra.Command{Annotations: make(map[string]string)}
+	oc := NewOperationCmd(parentCmd, "foo", "/v1/bars/{id}", "post", &config.Config{})
+	oc.APIBaseURL = ts.URL
+
+	err := oc.runOperationCmd(oc.Cmd, []string{"bar_123", "param1=value1", "param2=value2"})
+
+	require.NoError(t, err)
+}
+
+func TestRunOperationCmd_NoAPIKey(t *testing.T) {
+	viper.Reset()
+	parentCmd := &cobra.Command{Annotations: make(map[string]string)}
+	oc := NewOperationCmd(parentCmd, "foo", "/v1/bars/{id}", "post", &config.Config{})
+
+	err := oc.runOperationCmd(oc.Cmd, []string{"bar_123", "param1=value1", "param2=value2"})
+
+	require.Error(t, err, "your API key has not been configured. Use `stripe login` to set your API key")
+}