add support of team external group

team_external_group.go

@@ -0,0 +1,43 @@
+package gapi
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+)
+
+// TeamGroup represents a Grafana TeamGroup.
+type TeamGroup struct {
+	OrgID   int64  `json:"orgId,omitempty"`
+	TeamID  int64  `json:"teamId,omitempty"`
+	GroupID string `json:"groupID,omitempty"`
+}
+
+// TeamGroups fetches and returns the list of Grafana team group whose Team ID it's passed.
+func (c *Client) TeamGroups(id int64) ([]TeamGroup, error) {
+	teamGroups := make([]TeamGroup, 0)
+	err := c.request("GET", fmt.Sprintf("/api/teams/%d/groups", id), nil, nil, &teamGroups)
+	if err != nil {
+		return teamGroups, err
+	}
+
+	return teamGroups, nil
+}
+
+// NewTeamGroup creates a new Grafana Team Group .
+func (c *Client) NewTeamGroup(id int64, groupID string) error {
+	dataMap := map[string]string{
+		"groupId": groupID,
+	}
+	data, err := json.Marshal(dataMap)
+	if err != nil {
+		return err
+	}
+
+	return c.request("POST", fmt.Sprintf("/api/teams/%d/groups", id), nil, bytes.NewBuffer(data), nil)
+}
+
+// DeleteTeam deletes the Grafana team whose ID it's passed.
+func (c *Client) DeleteTeamGroup(id int64, groupID string) error {
+	return c.request("DELETE", fmt.Sprintf("/api/teams/%d/groups/%s", id, groupID), nil, nil, nil)
+}

team_external_group_test.go

@@ -0,0 +1,70 @@
+package gapi
+
+import (
+	"testing"
+
+	"github.com/gobs/pretty"
+)
+
+const (
+	getTeamGroupsJSON = `
+[
+  {
+    "orgId": 1,
+    "teamId": 1,
+    "groupId": "test"
+  }
+]
+`
+	createdTeamGroupJSON = `
+{
+  "message":"Group added to Team"
+}
+`
+
+	deletedTeamGroupJSON = `
+{
+  "message":"Team Group removed"
+}
+`
+)
+
+func TestTeamGroups(t *testing.T) {
+	server, client := gapiTestTools(t, 200, getTeamGroupsJSON)
+	defer server.Close()
+
+	teamID := int64(1)
+	teamGroups, err := client.TeamGroups(teamID)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	t.Log(pretty.PrettyFormat(teamGroups))
+
+	if len(teamGroups) != 1 {
+		t.Error("Length of returned teamGroups should be 1")
+	}
+	if teamGroups[0].TeamID != 1 || teamGroups[0].OrgID != 1 || teamGroups[0].GroupID != "test" {
+		t.Error("Not correctly parsing returned teamGroups.")
+	}
+}
+
+func TestNewTeamGroup(t *testing.T) {
+	server, client := gapiTestTools(t, 200, createdTeamGroupJSON)
+	defer server.Close()
+
+	err := client.NewTeamGroup(int64(1), "test")
+	if err != nil {
+		t.Fatal(err)
+	}
+}
+
+func TestDeleteTeamGroup(t *testing.T) {
+	server, client := gapiTestTools(t, 200, deletedTeamGroupJSON)
+	defer server.Close()
+
+	err := client.DeleteTeamGroup(int64(1), "test")
+	if err != nil {
+		t.Fatal(err)
+	}
+}