func TestGetTaskError(t *testing.T) {
	_, err := tc.GetTask("magicfoobar")

	if !(err != nil) {
		t.Errorf("Error expected in task fetching")
	}
}
// XXX: This is a terrible test since it will eventually fail but the docker
// worker tests will ensure this will work...
func TestGetTask(t *testing.T) {
	task, err := tc.GetTask("6dS50KOBRjCZX2L-ql3atQ")

	if err != nil {
		t.Fatalf("Error fetching task: %s", err)
	}

	if task.Scopes[0] != "docker-worker:cache:tc-vcs" {
		t.Errorf("Could not fetch task...")
	}
}
Beispiel #3
0
func main() {
	// Parse the docopt string and exit on any error or help message.
	arguments, err := docopt.Parse(usage, nil, true, version, false, true)

	taskId := arguments["<taskId>"].(string)
	port, err := strconv.Atoi(arguments["--port"].(string))
	if err != nil {
		log.Fatalf("Failed to convert port to integer")
	}

	// Parse out additional scopes to add...
	var additionalScopes []string
	if arguments["<scope>"] != nil {
		additionalScopes = arguments["<scope>"].([]string)
	} else {
		additionalScopes = make([]string, 0)
	}

	// Client is is required but has a default.
	clientId := arguments["--client-id"]
	if clientId == nil || clientId == "" {
		clientId = os.Getenv("TASKCLUSTER_CLIENT_ID")
	}

	// Access token is also required but has a default.
	accessToken := arguments["--access-token"]
	if accessToken == nil || accessToken == "" {
		accessToken = os.Getenv("TASKCLUSTER_ACCESS_TOKEN")
	}

	log.Printf("%v - %v", clientId, accessToken)

	// Ensure we have credentials our auth proxy is pretty much useless without
	// it.
	if accessToken == "" || clientId == "" {
		log.Fatalf(
			"Credentials must be passed via environment variables or flags...",
		)
	}

	// Fetch the task to get the scopes we should be using...
	task, err := tc.GetTask(taskId)

	if err != nil {
		log.Fatalf("Could not fetch taskcluster task '%s' : %s", taskId, err)
	}

	scopes := append(additionalScopes, task.Scopes...)

	log.Println("Proxy with scopes: ", scopes)

	routes := Routes{
		Scopes:      scopes,
		ClientId:    clientId.(string),
		AccessToken: accessToken.(string),
	}

	startError := http.ListenAndServe(fmt.Sprintf(":%d", port), routes)
	if startError != nil {
		log.Fatal(startError)
	}
}