Merge pull request #4 from joonas/add-list-teams

Add ListTeams method

concourse/client.go

@@ -31,6 +31,7 @@ type Client interface {
 	GetInfo() (atc.Info, error)
 	GetCLIReader(arch, platform string) (io.ReadCloser, http.Header, error)
 	ListPipelines() ([]atc.Pipeline, error)
+	ListTeams() ([]atc.Team, error)
 
 	Team(teamName string) Team
 }

concourse/concoursefakes/fake_client.go

@@ -163,6 +163,13 @@ type FakeClient struct {
 		result1 []atc.Pipeline
 		result2 error
 	}
+	ListTeamsStub        func() ([]atc.Team, error)
+	listTeamsMutex       sync.RWMutex
+	listTeamsArgsForCall []struct{}
+	listTeamsReturns     struct {
+		result1 []atc.Team
+		result2 error
+	}
 	TeamStub        func(teamName string) concourse.Team
 	teamMutex       sync.RWMutex
 	teamArgsForCall []struct {
@@ -734,6 +741,31 @@ func (fake *FakeClient) ListPipelinesReturns(result1 []atc.Pipeline, result2 err
 	}{result1, result2}
 }
 
+func (fake *FakeClient) ListTeams() ([]atc.Team, error) {
+	fake.listTeamsMutex.Lock()
+	fake.listTeamsArgsForCall = append(fake.listTeamsArgsForCall, struct{}{})
+	fake.recordInvocation("ListTeams", []interface{}{})
+	fake.listTeamsMutex.Unlock()
+	if fake.ListTeamsStub != nil {
+		return fake.ListTeamsStub()
+	}
+	return fake.listTeamsReturns.result1, fake.listTeamsReturns.result2
+}
+
+func (fake *FakeClient) ListTeamsCallCount() int {
+	fake.listTeamsMutex.RLock()
+	defer fake.listTeamsMutex.RUnlock()
+	return len(fake.listTeamsArgsForCall)
+}
+
+func (fake *FakeClient) ListTeamsReturns(result1 []atc.Team, result2 error) {
+	fake.ListTeamsStub = nil
+	fake.listTeamsReturns = struct {
+		result1 []atc.Team
+		result2 error
+	}{result1, result2}
+}
+
 func (fake *FakeClient) Team(teamName string) concourse.Team {
 	fake.teamMutex.Lock()
 	fake.teamArgsForCall = append(fake.teamArgsForCall, struct {
@@ -806,6 +838,8 @@ func (fake *FakeClient) Invocations() map[string][][]interface{} {
 	defer fake.getCLIReaderMutex.RUnlock()
 	fake.listPipelinesMutex.RLock()
 	defer fake.listPipelinesMutex.RUnlock()
+	fake.listTeamsMutex.RLock()
+	defer fake.listTeamsMutex.RUnlock()
 	fake.teamMutex.RLock()
 	defer fake.teamMutex.RUnlock()
 	return fake.invocations

concourse/teams.go

@@ -69,3 +69,14 @@ func (team *team) DestroyTeam(teamName string) error {
 
 	return err
 }
+
+func (client *client) ListTeams() ([]atc.Team, error) {
+	var teams []atc.Team
+	err := client.connection.Send(internal.Request{
+		RequestName: atc.ListTeams,
+	}, &internal.Response{
+		Result: &teams,
+	})
+
+	return teams, err
+}

concourse/teams_test.go

@@ -149,4 +149,40 @@ var _ = Describe("ATC Handler Teams", func() {
 			})
 		})
 	})
+
+	Describe("ListTeams", func() {
+		var expectedTeams []atc.Team
+
+		BeforeEach(func() {
+			expectedURL := "/api/v1/teams"
+
+			expectedTeams = []atc.Team{
+				{
+					ID:   1,
+					Name: "main",
+				},
+				{
+					ID:   2,
+					Name: "a-team",
+				},
+				{
+					ID:   3,
+					Name: "b-team",
+				},
+			}
+
+			atcServer.AppendHandlers(
+				ghttp.CombineHandlers(
+					ghttp.VerifyRequest("GET", expectedURL),
+					ghttp.RespondWithJSONEncoded(http.StatusOK, expectedTeams),
+				),
+			)
+		})
+
+		It("returns all of the teams", func() {
+			teams, err := client.ListTeams()
+			Expect(err).NotTo(HaveOccurred())
+			Expect(teams).To(Equal(expectedTeams))
+		})
+	})
 })