Emit a better error if the job API token is missing

Co-Authored-By: Josh Deprez <[email protected]>

jobapi/client.go

@@ -14,7 +14,10 @@ const (
 	redactionsURL = "http://job/api/current-job/v0/redactions"
 )
 
-var errNoSocketEnv = errors.New("BUILDKITE_AGENT_JOB_API_SOCKET empty or undefined")
+var (
+	errNoJobAPISocketEnv = errors.New("BUILDKITE_AGENT_JOB_API_SOCKET empty or undefined")
+	errNoJobAPITokenEnv  = errors.New("BUILDKITE_AGENT_JOB_API_TOKEN empty or undefined")
+)
 
 // Client connects to the Job API.
 type Client struct {
@@ -36,12 +39,14 @@ func NewDefaultClient(ctx context.Context) (*Client, error) {
 func DefaultSocketPath() (path, token string, err error) {
 	path = os.Getenv("BUILDKITE_AGENT_JOB_API_SOCKET")
 	if path == "" {
-		return "", "", errNoSocketEnv
+		return "", "", errNoJobAPISocketEnv
 	}
+
 	token = os.Getenv("BUILDKITE_AGENT_JOB_API_TOKEN")
 	if token == "" {
-		return "", "", errNoSocketEnv
+		return "", "", errNoJobAPITokenEnv
 	}
+
 	return path, token, nil
 }
 

jobapi/client_test.go

@@ -130,7 +130,18 @@ func TestClient_NoSocket(t *testing.T) {
 	// This may be set if the test is being run by a buildkite agent!
 	os.Unsetenv("BUILDKITE_AGENT_JOB_API_SOCKET")
 	_, err := NewDefaultClient(ctx)
-	assert.ErrorIs(t, err, errNoSocketEnv, "NewDefaultClient() error = %v, want %v", err, errNoSocketEnv)
+	assert.ErrorIs(t, err, errNoJobAPISocketEnv, "NewDefaultClient() error = %v, want %v", err, errNoJobAPISocketEnv)
+}
+
+func TestClient_Token(t *testing.T) {
+	ctx, canc := context.WithTimeout(context.Background(), 10*time.Second)
+	t.Cleanup(canc)
+
+	t.Setenv("BUILDKITE_AGENT_JOB_API_SOCKET", "/tmp/fake-socket")
+	// This may be set if the test is being run by a buildkite agent!
+	os.Unsetenv("BUILDKITE_AGENT_JOB_API_TOKEN")
+	_, err := NewDefaultClient(ctx)
+	assert.ErrorIs(t, err, errNoJobAPITokenEnv, "NewDefaultClient() error = %v, want %v", err, errNoJobAPITokenEnv)
 }
 
 func TestClientEnvGet(t *testing.T) {