From bbc4e3902316b7e47dc0770c085badd8d588d9c4 Mon Sep 17 00:00:00 2001
From: rawmind0 <rawmind@gmail.com>
Date: Tue, 29 Jan 2019 21:34:56 +0100
Subject: [PATCH] Added PatchVirtualServer and PatchVirtualAddress functions
 and tests to update VirtualServer and VirtualAddress objects using patch
 method

---
 ltm.go      | 16 ++++++++++++++--
 ltm_test.go | 32 ++++++++++++++++++++++++++++++++
 2 files changed, 46 insertions(+), 2 deletions(-)

diff --git a/ltm.go b/ltm.go
index 2c8564b..21a6185 100644
--- a/ltm.go
+++ b/ltm.go
@@ -1824,11 +1824,17 @@ func (b *BigIP) DeleteVirtualServer(name string) error {
 }
 
 // ModifyVirtualServer allows you to change any attribute of a virtual server. Fields that
-// can be modified are referenced in the VirtualServer struct.
+// can be modified are referenced in the VirtualServer struct. Set all the attributes.
 func (b *BigIP) ModifyVirtualServer(name string, config *VirtualServer) error {
 	return b.put(config, uriLtm, uriVirtual, name)
 }
 
+// PatchVirtualServer allows you to change any attribute of a virtual server. Fields that
+// can be modified are referenced in the VirtualServer struct. Sets only the attributes specified.
+func (b *BigIP) PatchVirtualServer(name string, config *VirtualServer) error {
+	return b.patch(config, uriLtm, uriVirtual, name)
+}
+
 // VirtualServerProfiles gets the profiles currently associated with a virtual server.
 func (b *BigIP) VirtualServerProfiles(vs string) (*Profiles, error) {
 	var p Profiles
@@ -1891,11 +1897,17 @@ func (b *BigIP) VirtualAddressStatus(vaddr, state string) error {
 }
 
 // ModifyVirtualAddress allows you to change any attribute of a virtual address. Fields that
-// can be modified are referenced in the VirtualAddress struct.
+// can be modified are referenced in the VirtualAddress struct. Sets all the attributes.
 func (b *BigIP) ModifyVirtualAddress(vaddr string, config *VirtualAddress) error {
 	return b.put(config, uriLtm, uriVirtualAddress, vaddr)
 }
 
+// PatchVirtualAddress allows you to change any attribute of a virtual address. Fields that
+// can be modified are referenced in the VirtualAddress struct. Sets only the attributes specified.
+func (b *BigIP) PatchVirtualAddress(vaddr string, config *VirtualAddress) error {
+	return b.patch(config, uriLtm, uriVirtualAddress, vaddr)
+}
+
 func (b *BigIP) DeleteVirtualAddress(vaddr string) error {
 	return b.delete(uriLtm, uriVirtualAddress, vaddr)
 }
diff --git a/ltm_test.go b/ltm_test.go
index 9003495..11cefda 100644
--- a/ltm_test.go
+++ b/ltm_test.go
@@ -100,6 +100,14 @@ func (s *LTMTestSuite) TestModifyVirtualAddress() {
 	assert.Equal(s.T(), "PUT", s.LastRequest.Method)
 }
 
+func (s *LTMTestSuite) TestPatchVirtualAddress() {
+	d := &VirtualAddress{}
+	s.Client.PatchVirtualAddress("address1", d)
+
+	assert.Equal(s.T(), fmt.Sprintf("/mgmt/tm/%s/%s/%s", uriLtm, uriVirtualAddress, "address1"), s.LastRequest.URL.Path)
+	assert.Equal(s.T(), "PATCH", s.LastRequest.Method)
+}
+
 func (s *LTMTestSuite) TestGetPolicies() {
 	s.ResponseFunc = func(w http.ResponseWriter, r *http.Request) {
 		w.Write([]byte(`{
@@ -468,6 +476,30 @@ func (s *LTMTestSuite) TestModifyVirtualServer() {
 
 }
 
+func (s *LTMTestSuite) TestPatchVirtualServer() {
+	vs := &VirtualServer{
+		Name: "test",
+		Profiles: []Profile{
+			Profile{Name: "/Common/tcp", Context: CONTEXT_CLIENT},
+			Profile{Name: "/Common/tcp", Context: CONTEXT_SERVER}},
+		//TODO: test more
+	}
+
+	s.Client.PatchVirtualServer("test", vs)
+
+	assert.Equal(s.T(), "PATCH", s.LastRequest.Method)
+	assert.Equal(s.T(), fmt.Sprintf("/mgmt/tm/%s/%s/test", uriLtm, uriVirtual), s.LastRequest.URL.Path)
+	assert.JSONEq(s.T(), `
+	{"name":"test",
+	"sourceAddressTranslation":{},
+	"profiles":[
+		{"name":"/Common/tcp","context":"clientside"},
+		{"name":"/Common/tcp","context":"serverside"}
+	]
+	}`, s.LastRequestBody)
+
+}
+
 func (s *LTMTestSuite) TestDeleteVirtualServer() {
 	s.Client.DeleteVirtualServer("/Common/test-vs")