resultstore/context_helpers.go

// Copyright 2018 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package resultstore

import (
	"context"

	"fuchsia.googlesource.com/infra/infra/fxctx"
)

const (
	// TestUUIDKey references a UUID to use for testing.
	TestUUIDKey = fxctx.ContextKey("test-uuid")

	// AuthTokenKey references the auth token to use in ResultStore requests.
	AuthTokenKey = fxctx.ContextKey("auth-token")
)

// TestUUID gets the test UUID, if present. Errors if the value is missing or cannot be
// read from the Context.
func TestUUID(ctx context.Context) (string, error) {
	var uuid string
	err := fxctx.Get(ctx, TestUUIDKey, &uuid)
	return uuid, err
}

// SetTestUUID sets a UUID to use for testing.
func SetTestUUID(ctx context.Context, uuid string) (context.Context, error) {
	return fxctx.Set(ctx, TestUUIDKey, uuid)
}

// AuthToken gets the auth token to use in ResultStore requests.
func AuthToken(ctx context.Context) (string, error) {
	var token string
	err := fxctx.Get(ctx, AuthTokenKey, &token)
	return token, err
}

// SetAuthToken sets the auth token to use in ResultStore requests.
func SetAuthToken(ctx context.Context, value string) (context.Context, error) {
	return fxctx.Set(ctx, AuthTokenKey, value)
}