feat: added mock server tests

Author: stainless-app[bot]

internal/mocktest/mocktest.go

@@ -0,0 +1,96 @@
+package mocktest
+
+import (
+	"context"
+	"fmt"
+	"net"
+	"net/http"
+	"net/url"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"runtime"
+	"strings"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+var mockServerURL *url.URL
+
+func init() {
+	mockServerURL, _ = url.Parse("http://localhost:4010")
+	if testURL := os.Getenv("TEST_API_BASE_URL"); testURL != "" {
+		if parsed, err := url.Parse(testURL); err == nil {
+			mockServerURL = parsed
+		}
+	}
+}
+
+// OnlyMockServerDialer only allows network connections to the mock server
+type OnlyMockServerDialer struct{}
+
+func (d *OnlyMockServerDialer) DialContext(ctx context.Context, network, address string) (net.Conn, error) {
+	if address == mockServerURL.Host {
+		return (&net.Dialer{}).DialContext(ctx, network, address)
+	}
+
+	return nil, fmt.Errorf("BLOCKED: connection to %s not allowed (only allowed: %s)", address, mockServerURL.Host)
+}
+
+func blockNetworkExceptMockServer() (http.RoundTripper, http.RoundTripper) {
+	restricted := &http.Transport{
+		DialContext: (&OnlyMockServerDialer{}).DialContext,
+	}
+
+	origClient, origDefault := http.DefaultClient.Transport, http.DefaultTransport
+	http.DefaultClient.Transport, http.DefaultTransport = restricted, restricted
+	return origClient, origDefault
+}
+
+func restoreNetwork(origClient, origDefault http.RoundTripper) {
+	http.DefaultClient.Transport, http.DefaultTransport = origClient, origDefault
+}
+
+// TestRunMockTestWithFlags runs a test against a mock server with the provided
+// CLI flags and ensures it succeeds
+func TestRunMockTestWithFlags(t *testing.T, flags ...string) {
+	origClient, origDefault := blockNetworkExceptMockServer()
+	defer restoreNetwork(origClient, origDefault)
+
+	// Check if mock server is running
+	conn, err := net.DialTimeout("tcp", mockServerURL.Host, 2*time.Second)
+	if err != nil {
+		require.Fail(t, "Mock server is not running on "+mockServerURL.Host+". Please start the mock server before running tests.")
+	} else {
+		conn.Close()
+	}
+
+	// Get the path to the main command
+	_, filename, _, ok := runtime.Caller(0)
+	require.True(t, ok, "Could not get current file path")
+	dirPath := filepath.Dir(filename)
+	project := filepath.Join(dirPath, "..", "..", "cmd", "...")
+
+	args := []string{"run", project, "--base-url", mockServerURL.String()}
+	args = append(args, flags...)
+
+	t.Logf("Testing command: stl %s", strings.Join(args[4:], " "))
+
+	cmd := exec.Command("go", args...)
+	output, err := cmd.CombinedOutput()
+	if err != nil {
+		assert.Fail(t, "Test failed", "Error: %v\nOutput: %s", err, output)
+	}
+
+	t.Logf("Test passed successfully with output:\n%s\n", output)
+}
+
+func TestFile(t *testing.T, contents string) string {
+	tmpDir := t.TempDir()
+	filename := filepath.Join(tmpDir, "file.txt")
+	require.NoError(t, os.WriteFile(filename, []byte(contents), 0644))
+	return filename
+}

internal/requestflag/requestflag.go

@@ -564,7 +564,7 @@ func (d *DateValue) Parse(s string) error {
 func (d *DateTimeValue) Parse(s string) error {
 	formats := []string{
 		time.RFC3339,
-		"2006-01-02T15:04:05Z07:00",
+		time.RFC3339Nano,
 		"2006-01-02T15:04:05",
 		"2006-01-02 15:04:05",
 		time.RFC1123,
@@ -584,6 +584,7 @@ func (d *DateTimeValue) Parse(s string) error {
 func (t *TimeValue) Parse(s string) error {
 	formats := []string{
 		"15:04:05",
+		"15:04:05.999999999Z07:00",
 		"3:04:05PM",
 		"3:04 PM",
 		"15:04",

pkg/cmd/build_test.go

@@ -0,0 +1,58 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+package cmd
+
+import (
+	"testing"
+
+	"github.com/stainless-api/stainless-api-cli/internal/mocktest"
+)
+
+func TestBuildsCreate(t *testing.T) {
+	t.Skip("Prism tests are disabled")
+	mocktest.TestRunMockTestWithFlags(
+		t,
+		"builds", "create",
+		"--project", "project",
+		"--revision", "string",
+		"--allow-empty",
+		"--branch", "branch",
+		"--commit-message", "commit_message",
+		"--target-commit-messages", "{cli: cli, csharp: csharp, go: go, java: java, kotlin: kotlin, node: node, openapi: openapi, php: php, python: python, ruby: ruby, terraform: terraform, typescript: typescript}",
+		"--target", "node",
+	)
+}
+
+func TestBuildsRetrieve(t *testing.T) {
+	t.Skip("Prism tests are disabled")
+	mocktest.TestRunMockTestWithFlags(
+		t,
+		"builds", "retrieve",
+		"--build-id", "buildId",
+	)
+}
+
+func TestBuildsList(t *testing.T) {
+	t.Skip("Prism tests are disabled")
+	mocktest.TestRunMockTestWithFlags(
+		t,
+		"builds", "list",
+		"--project", "project",
+		"--branch", "branch",
+		"--cursor", "cursor",
+		"--limit", "1",
+		"--revision", "string",
+	)
+}
+
+func TestBuildsCompare(t *testing.T) {
+	t.Skip("Prism tests are disabled")
+	mocktest.TestRunMockTestWithFlags(
+		t,
+		"builds", "compare",
+		"--base", "{branch: branch, revision: string, commit_message: commit_message}",
+		"--head", "{branch: branch, revision: string, commit_message: commit_message}",
+		"--project", "project",
+		"--target", "node",
+	)
+}

pkg/cmd/builddiagnostic_test.go

@@ -0,0 +1,22 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+package cmd
+
+import (
+	"testing"
+
+	"github.com/stainless-api/stainless-api-cli/internal/mocktest"
+)
+
+func TestBuildsDiagnosticsList(t *testing.T) {
+	t.Skip("Prism tests are disabled")
+	mocktest.TestRunMockTestWithFlags(
+		t,
+		"builds:diagnostics", "list",
+		"--build-id", "buildId",
+		"--cursor", "cursor",
+		"--limit", "1",
+		"--severity", "fatal",
+		"--targets", "targets",
+	)
+}

pkg/cmd/buildtargetoutput_test.go

@@ -0,0 +1,21 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+package cmd
+
+import (
+	"testing"
+
+	"github.com/stainless-api/stainless-api-cli/internal/mocktest"
+)
+
+func TestBuildsTargetOutputsRetrieve(t *testing.T) {
+	t.Skip("Prism tests are disabled")
+	mocktest.TestRunMockTestWithFlags(
+		t,
+		"builds:target-outputs", "retrieve",
+		"--build-id", "build_id",
+		"--target", "node",
+		"--type", "source",
+		"--output", "url",
+	)
+}

pkg/cmd/flagoptions.go

@@ -24,6 +24,7 @@ const (
 	EmptyBody BodyContentType = iota
 	MultipartFormEncoded
 	ApplicationJSON
+	ApplicationOctetStream
 )
 
 func flagOptions(
@@ -125,12 +126,23 @@ func flagOptions(
 			return nil, err
 		}
 		options = append(options, option.WithRequestBody(writer.FormDataContentType(), buf))
+
 	case ApplicationJSON:
 		bodyBytes, err := json.Marshal(bodyData)
 		if err != nil {
 			return nil, err
 		}
 		options = append(options, option.WithRequestBody("application/json", bodyBytes))
+
+	case ApplicationOctetStream:
+		if bodyBytes, ok := bodyData.([]byte); ok {
+			options = append(options, option.WithRequestBody("application/octet-stream", bodyBytes))
+		} else if bodyStr, ok := bodyData.(string); ok {
+			options = append(options, option.WithRequestBody("application/octet-stream", []byte(bodyStr)))
+		} else {
+			return nil, fmt.Errorf("Unsupported body for application/octet-stream: %v", bodyData)
+		}
+
 	default:
 		panic("Invalid body content type!")
 	}

pkg/cmd/org_test.go

@@ -0,0 +1,26 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+package cmd
+
+import (
+	"testing"
+
+	"github.com/stainless-api/stainless-api-cli/internal/mocktest"
+)
+
+func TestOrgsRetrieve(t *testing.T) {
+	t.Skip("Prism tests are disabled")
+	mocktest.TestRunMockTestWithFlags(
+		t,
+		"orgs", "retrieve",
+		"--org", "org",
+	)
+}
+
+func TestOrgsList(t *testing.T) {
+	t.Skip("Prism tests are disabled")
+	mocktest.TestRunMockTestWithFlags(
+		t,
+		"orgs", "list",
+	)
+}

pkg/cmd/project_test.go

@@ -0,0 +1,52 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+package cmd
+
+import (
+	"testing"
+
+	"github.com/stainless-api/stainless-api-cli/internal/mocktest"
+)
+
+func TestProjectsCreate(t *testing.T) {
+	t.Skip("Prism tests are disabled")
+	mocktest.TestRunMockTestWithFlags(
+		t,
+		"projects", "create",
+		"--display-name", "display_name",
+		"--org", "org",
+		"--revision", "{foo: {content: content}}",
+		"--slug", "slug",
+		"--target", "node",
+	)
+}
+
+func TestProjectsRetrieve(t *testing.T) {
+	t.Skip("Prism tests are disabled")
+	mocktest.TestRunMockTestWithFlags(
+		t,
+		"projects", "retrieve",
+		"--project", "project",
+	)
+}
+
+func TestProjectsUpdate(t *testing.T) {
+	t.Skip("Prism tests are disabled")
+	mocktest.TestRunMockTestWithFlags(
+		t,
+		"projects", "update",
+		"--project", "project",
+		"--display-name", "display_name",
+	)
+}
+
+func TestProjectsList(t *testing.T) {
+	t.Skip("Prism tests are disabled")
+	mocktest.TestRunMockTestWithFlags(
+		t,
+		"projects", "list",
+		"--cursor", "cursor",
+		"--limit", "1",
+		"--org", "org",
+	)
+}