| // Copyright 2023 The Go 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 swarmclient |
| |
| import ( |
| "context" |
| "os" |
| "testing" |
| |
| "github.com/google/go-cmp/cmp" |
| ) |
| |
| func TestListSwarmingBots(t *testing.T) { |
| contents, err := os.ReadFile("testdata/bb-sample.cfg") |
| if err != nil { |
| t.Fatalf("os.ReadFile() = nil, %s", err) |
| } |
| ctx := context.Background() |
| client := NewMemoryConfigClient(ctx, []*ConfigEntry{ |
| &ConfigEntry{"cr-buildbucket.cfg", contents}, |
| }) |
| bots, err := client.ListSwarmingBots(ctx) |
| if err != nil { |
| t.Fatalf("ListSwarmingBots() = nil, %s", err) |
| } |
| wantLen := 21 |
| if len(bots) != wantLen { |
| t.Errorf("len(bots) = %d; want %d", len(bots), wantLen) |
| } |
| bot := bots[0] |
| if bot.BucketName != "ci" { |
| t.Errorf("bot.BucketName = %q, want %q", bot.BucketName, "ci") |
| } |
| wantDimensions := []string{"cpu:x86-64", "os:Linux", "pool:luci.golang.ci"} |
| if diff := cmp.Diff(bot.Dimensions, wantDimensions); diff != "" { |
| t.Errorf("bot.Dimensions mismatch (-want +got): \n%s", diff) |
| } |
| wantHost := "chromium-swarm.appspot.com" |
| if bot.Host != wantHost { |
| t.Errorf("bot.Host = %q, want %q", bot.Host, wantHost) |
| } |
| wantName := "go1.20-darwin-amd64" |
| if bot.Name != wantName { |
| t.Errorf("bot.Name = %q, want %q", bot.Name, wantName) |
| } |
| } |
