From 8d5be347621096645358642691b406fedb43fdc2 Mon Sep 17 00:00:00 2001
From: "R. Aidan Campbell" <aidan.campbell@aexp.com>
Date: Fri, 1 May 2020 09:58:10 -0700
Subject: [PATCH] Add support for modifying an incident status and assignees,
 as per
 https://developer.pagerduty.com/api-reference/reference/REST/openapiv3.json/paths/~1incidents/put.
  This resolves a regression from
 https://github.com/PagerDuty/go-pagerduty/pull/190

---
 incident.go      |   8 ++--
 incident_test.go | 112 ++++++++++++++++++++++++++++++++++++++++++++++-
 2 files changed, 116 insertions(+), 4 deletions(-)

diff --git a/incident.go b/incident.go
index 2ceffaf1..a6f14e15 100644
--- a/incident.go
+++ b/incident.go
@@ -139,9 +139,11 @@ type CreateIncidentOptions struct {
 
 // ManageIncidentsOptions is the structure used when PUTing updates to incidents to the ManageIncidents func
 type ManageIncidentsOptions struct {
-	ID     string `json:"id"`
-	Type   string `json:"type"`
-	Status string `json:"status"`
+	ID          string        `json:"id"`
+	Type        string        `json:"type"`
+	Status      string        `json:"status,omitempty"`
+	Priority    *APIReference `json:"priority,omitempty"`
+	Assignments []Assignee    `json:"assignments,omitempty"`
 }
 
 // MergeIncidentsOptions is the structure used when merging incidents with MergeIncidents func
diff --git a/incident_test.go b/incident_test.go
index 640f1717..b343a816 100644
--- a/incident_test.go
+++ b/incident_test.go
@@ -64,7 +64,7 @@ func TestIncident_Create(t *testing.T) {
 	testEqual(t, want, res)
 }
 
-func TestIncident_Manage(t *testing.T) {
+func TestIncident_Manage_status(t *testing.T) {
 	setup()
 	defer teardown()
 
@@ -102,6 +102,116 @@ func TestIncident_Manage(t *testing.T) {
 	testEqual(t, want, res)
 }
 
+func TestIncident_Manage_priority(t *testing.T) {
+	setup()
+	defer teardown()
+
+	mux.HandleFunc("/incidents", func(w http.ResponseWriter, r *http.Request) {
+		testMethod(t, r, "PUT")
+		w.Write([]byte(`{"incidents": [{"title": "foo", "id": "1", "priority": {"id": "PRIORITY_ID_HERE", "type": "priority_reference"}}]}`))
+	})
+	var listObj = APIListObject{Limit: 0, Offset: 0, More: false, Total: 0}
+	var client = &Client{apiEndpoint: server.URL, authToken: "foo", HTTPClient: defaultHTTPClient}
+	from := "foo@bar.com"
+
+	input := []ManageIncidentsOptions{
+		{
+			ID:   "1",
+			Type: "incident",
+			Priority: &APIReference{
+				ID:   "PRIORITY_ID_HERE",
+				Type: "priority_reference",
+			},
+		},
+	}
+
+	want := &ListIncidentsResponse{
+		APIListObject: listObj,
+		Incidents: []Incident{
+			{
+				Id:    "1",
+				Title: "foo",
+				Priority: &Priority{
+					APIObject: APIObject{
+						ID:   "PRIORITY_ID_HERE",
+						Type: "priority_reference",
+					},
+				},
+			},
+		},
+	}
+	res, err := client.ManageIncidents(from, input)
+
+	if err != nil {
+		t.Fatal(err)
+	}
+	testEqual(t, want, res)
+}
+
+func TestIncident_Manage_assignments(t *testing.T) {
+	setup()
+	defer teardown()
+
+	mux.HandleFunc("/incidents", func(w http.ResponseWriter, r *http.Request) {
+		testMethod(t, r, "PUT")
+		w.Write([]byte(`{"incidents": [{"title": "foo", "id": "1", "assignments": [{"assignee":{"id": "ASSIGNEE_ONE", "type": "user_reference"}},{"assignee":{"id": "ASSIGNEE_TWO", "type": "user_reference"}}]}]}`))
+	})
+	var listObj = APIListObject{Limit: 0, Offset: 0, More: false, Total: 0}
+	var client = &Client{apiEndpoint: server.URL, authToken: "foo", HTTPClient: defaultHTTPClient}
+	from := "foo@bar.com"
+
+	input := []ManageIncidentsOptions{
+		{
+			ID:   "1",
+			Type: "incident",
+			Assignments: []Assignee{
+				{
+					Assignee: APIObject{
+						ID:   "ASSIGNEE_ONE",
+						Type: "user_reference",
+					},
+				},
+				{
+					Assignee: APIObject{
+						ID:   "ASSIGNEE_TWO",
+						Type: "user_reference",
+					},
+				},
+			},
+		},
+	}
+
+	want := &ListIncidentsResponse{
+		APIListObject: listObj,
+		Incidents: []Incident{
+			{
+				Id:    "1",
+				Title: "foo",
+				Assignments: []Assignment{
+					{
+						Assignee: APIObject{
+							ID:   "ASSIGNEE_ONE",
+							Type: "user_reference",
+						},
+					},
+					{
+						Assignee: APIObject{
+							ID:   "ASSIGNEE_TWO",
+							Type: "user_reference",
+						},
+					},
+				},
+			},
+		},
+	}
+	res, err := client.ManageIncidents(from, input)
+
+	if err != nil {
+		t.Fatal(err)
+	}
+	testEqual(t, want, res)
+}
+
 func TestIncident_Merge(t *testing.T) {
 	setup()
 	defer teardown()