Merge pull request #106 from aliyun/update_alert

add alert field

client_alert.go

@@ -86,12 +86,13 @@ type AlertQuery struct {
 }
 
 type Notification struct {
-	Type       string   `json:"type"`
-	Content    string   `json:"content"`
-	EmailList  []string `json:"emailList,omitempty"`
-	Method     string   `json:"method,omitempty"`
-	MobileList []string `json:"mobileList,omitempty"`
-	ServiceUri string   `json:"serviceUri,omitempty"`
+	Type       string            `json:"type"`
+	Content    string            `json:"content"`
+	EmailList  []string          `json:"emailList,omitempty"`
+	Method     string            `json:"method,omitempty"`
+	MobileList []string          `json:"mobileList,omitempty"`
+	ServiceUri string            `json:"serviceUri,omitempty"`
+	Headers    map[string]string `json:"headers,omitempty"`
 }
 
 type Schedule struct {
@@ -208,7 +209,6 @@ func (c *Client) ListSavedSearch(project string, savedSearchName string, offset,
 	return listSavedSearch.Savedsearches, listSavedSearch.Total, listSavedSearch.Count, err
 }
 
-
 func (c *Client) ListSavedSearchV2(project string, savedSearchName string, offset, size int) (savedSearches []string, savedsearchItems []ResponseSavedSearchItem, total int, count int, err error) {
 	h := map[string]string{
 		"x-log-bodyrawsize": "0",
@@ -221,14 +221,14 @@ func (c *Client) ListSavedSearchV2(project string, savedSearchName string, offse
 	uri := "/savedsearches"
 	r, err := c.request(project, "GET", uri, h, nil)
 	if err != nil {
-		return nil, nil,0, 0, err
+		return nil, nil, 0, 0, err
 	}
 	defer r.Body.Close()
 
 	type ListSavedSearch struct {
-		Total         int      `json:"total"`
-		Count         int      `json:"count"`
-		Savedsearches []string `json:"savedsearches"`
+		Total            int                       `json:"total"`
+		Count            int                       `json:"count"`
+		Savedsearches    []string                  `json:"savedsearches"`
 		SavedsearchItems []ResponseSavedSearchItem `json:"savedsearchItems"`
 	}
 

client_alert_test.go

@@ -108,6 +108,13 @@ func (s *AlertTestSuite) createAlert() error {
 					Content:    "${alertName} triggered at ${firetime}",
 					MobileList: []string{"1234567891"},
 				},
+				{
+					Type:       NotificationTypeWebhook,
+					Method:     "OPTIONS",
+					Content:    "${alertName} triggered at ${firetime}",
+					Headers:    map[string]string{"content-type": "test", "name": "aliyun"},
+					ServiceUri: "https://www.aliyun.com/",
+				},
 			},
 			NotifyThreshold: 1,
 		},
@@ -166,6 +173,14 @@ func (s *AlertTestSuite) TestClient_GetAlert() {
 	getAlert, err := s.client.GetAlert(s.projectName, s.alertName)
 	s.Require().Nil(err)
 	s.Require().Equal(getAlert.Name, s.alertName)
+	s.Require().Equal(len(getAlert.Configuration.NotificationList), 3)
+	for _, v := range getAlert.Configuration.NotificationList {
+		if v.Type == NotificationTypeWebhook {
+			s.Require().Equal(v.Method, "OPTIONS")
+			s.Require().Equal(v.Headers, map[string]string{"content-type": "test", "name": "aliyun"})
+		}
+	}
+
 	err = s.client.DeleteAlert(s.projectName, s.alertName)
 	s.Require().Nil(err)
 }