From 187c8d13e66b976c5550cbb558a2ad4ee7fb35bb Mon Sep 17 00:00:00 2001
From: Zaid <zaid.usmani@intel.com>
Date: Wed, 25 Oct 2023 16:17:11 -0700
Subject: [PATCH] feat(amt): adds execute support for setup

---
 pkg/amt/setupandconfiguration/service.go      |   2 +-
 pkg/amt/setupandconfiguration/service_test.go | 104 ++++++++++++------
 .../amt/setupandconfiguration/enumerate.xml   |  22 ++++
 .../amt/setupandconfiguration/get.xml         |   2 +-
 4 files changed, 93 insertions(+), 37 deletions(-)
 create mode 100644 pkg/wsmantesting/responses/amt/setupandconfiguration/enumerate.xml

diff --git a/pkg/amt/setupandconfiguration/service.go b/pkg/amt/setupandconfiguration/service.go
index bf86eb2d..14536f8a 100644
--- a/pkg/amt/setupandconfiguration/service.go
+++ b/pkg/amt/setupandconfiguration/service.go
@@ -28,7 +28,7 @@ type (
 	}
 	Body struct {
 		XMLName           		  xml.Name          	  		`xml:"Body"`
-		Setup	  				  Setup 				    	`xml:"AMT_AuthorizationService"`
+		Setup	  				  Setup 				    	`xml:"AMT_SetupAndConfigurationService"`
 		
 		EnumerateResponse 		  common.EnumerateResponse
 	}
diff --git a/pkg/amt/setupandconfiguration/service_test.go b/pkg/amt/setupandconfiguration/service_test.go
index 68a36d32..03d6a619 100644
--- a/pkg/amt/setupandconfiguration/service_test.go
+++ b/pkg/amt/setupandconfiguration/service_test.go
@@ -7,25 +7,59 @@ package setupandconfiguration
 
 import (
 	"encoding/xml"
+	"fmt"
+	"io"
+	"os"
+	"strings"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
 
 	"github.com/open-amt-cloud-toolkit/go-wsman-messages/internal/message"
 	"github.com/open-amt-cloud-toolkit/go-wsman-messages/pkg/wsmantesting"
-	"github.com/open-amt-cloud-toolkit/go-wsman-messages/pkg/wsman"
-	//"github.com/open-amt-cloud-toolkit/go-wsman-messages/pkg/common"
+	//"github.com/open-amt-cloud-toolkit/go-wsman-messages/pkg/wsman"
+	"github.com/open-amt-cloud-toolkit/go-wsman-messages/pkg/common"
 )
 
+type MockClient struct {
+}
+
+const (
+	EnvelopeResponse = `<a:Envelope xmlns:a="http://www.w3.org/2003/05/soap-envelope" x-mlns:b="http://schemas.xmlsoap.org/ws/2004/08/addressing" xmlns:c="http://schemas.dmtf.org/wbem/wsman/1/wsman.xsd" xmlns:d="http://schemas.xmlsoap.org/ws/2005/02/trust" xmlns:e="http://docs.oasis-open.org/wss/2004/01/oasis-200401-wss-wssecurity-secext-1.0.xsd" xmlns:f="http://schemas.dmtf.org/wbem/wsman/1/cimbinding.xsd" xmlns:g="http://intel.com/wbem/wscim/1/amt-schema/1/AMT_RedirectionService" xmlns:h="http://schemas.dmtf.org/wbem/wscim/1/common" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"><a:Header><b:To>http://schemas.xmlsoap.org/ws/2004/08/addressing/role/anonymous</b:To><b:RelatesTo>0</b:RelatesTo><b:Action a:mustUnderstand="true">`
+	GetBody          = `<g:AMT_RedirectionService><g:CreationClassName>AMT_RedirectionService</g:CreationClassName><g:ElementName>Intel(r) AMT Redirection Service</g:ElementName><g:Name>Intel(r) AMT Redirection Service</g:Name><g:SystemCreationClassName>CIM_ComputerSystem</g:SystemCreationClassName><g:SystemName>ManagedSystem</g:SystemName></g:AMT_RedirectionService`
+)
+
+var currentMessage = ""
+
+func (c *MockClient) Post(msg string) ([]byte, error) {
+	// read an xml file from disk:
+	xmlFile, err := os.Open("../../wsmantesting/responses/amt/setupandconfiguration/" + strings.ToLower(currentMessage) + ".xml")
+	if err != nil {
+		fmt.Println("Error opening file:", err)
+		return nil, err
+	}
+	defer xmlFile.Close()
+	// read file into string
+	xmlData, err := io.ReadAll(xmlFile)
+	if err != nil {
+		fmt.Println("Error reading file:", err)
+		return nil, err
+	}
+	// strip carriage returns and new line characters
+	xmlData = []byte(strings.ReplaceAll(string(xmlData), "\r\n", ""))
+
+	// Simulate a successful response for testing.
+	return []byte(xmlData), nil
+}
 func TestAMT_SetupAndConfigurationService(t *testing.T) {
 	messageID := 0
 	resourceUriBase := "http://intel.com/wbem/wscim/1/amt-schema/1/"
 	wsmanMessageCreator := message.NewWSManMessageCreator(resourceUriBase)
-	//client := MockClient{} // wsman.NewClient("http://localhost:16992/wsman", "admin", "P@ssw0rd", true)
+	client := MockClient{} // wsman.NewClient("http://localhost:16992/wsman", "admin", "P@ssw0rd", true)
 	//elementUnderTest := NewServiceWithClient(wsmanMessageCreator, &client)
 	// enumerationId := ""
-	client := wsman.NewClient("http://localhost:16992/wsman", "admin", "Intel123!", true)
-	elementUnderTest := NewSetupAndConfigurationServiceWithClient(wsmanMessageCreator, client)
+	//client := wsman.NewClient("http://localhost:16992/wsman", "admin", "Intel123!", true)
+	elementUnderTest := NewSetupAndConfigurationServiceWithClient(wsmanMessageCreator, &client)
 
 	t.Run("amt_* Tests", func(t *testing.T) {
 		tests := []struct {
@@ -42,44 +76,44 @@ func TestAMT_SetupAndConfigurationService(t *testing.T) {
 				"AMT_SetupAndConfigurationService", 
 				wsmantesting.GET, "", 
 				func() (Response, error) {
-					//currentMessage = "Get"
+					currentMessage = "Get"
 					return elementUnderTest.Get()
 				},
 				Body{
 					XMLName: xml.Name{Space: "http://www.w3.org/2003/05/soap-envelope", Local: "Body"},
 					Setup: Setup{
-						CreationClassName: "",
-						ElementName: "",
-						EnabledState: 0,
-						Name: "",
-						PasswordModel: 0,
-						ProvisioningMode: 0,
-						ProvisioningServerOTP: "",
-						ProvisioningState: 0,
-						RequestedState: 0,
-						SystemCreationClassName: "",
-						SystemName: "",
-						ZeroTouchConfigurationEnabled: false,
+						CreationClassName: "AMT_SetupAndConfigurationService",
+						ElementName: "Intel(r) AMT Setup and Configuration Service",
+						EnabledState: 5,
+						Name: "Intel(r) AMT Setup and Configuration Service",
+						PasswordModel: 1,
+						ProvisioningMode: 1,
+						ProvisioningServerOTP: "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA=",
+						ProvisioningState: 2,
+						RequestedState: 12,
+						SystemCreationClassName: "CIM_ComputerSystem",
+						SystemName: "Intel(r) AMT",
+						ZeroTouchConfigurationEnabled: true,
 					},
 				},
 			},
 			//ENUMERATES
-			// {
-			// 	"should create a valid AMT_SetupAndConfigurationService Enumerate wsman message", 
-			// 	"AMT_SetupAndConfigurationService", 
-			// 	wsmantesting.ENUMERATE, 
-			// 	wsmantesting.ENUMERATE_BODY, 
-			// 	func() (Response, error) {
-			// 		//client.CurrentMessage = "Enumerate"
-			// 		return elementUnderTest.Enumerate()
-			// 	}, 
-			// 	Body{
-			// 		XMLName: xml.Name{Space: "http://www.w3.org/2003/05/soap-envelope", Local: "Body"},
-			// 		EnumerateResponse: common.EnumerateResponse{
-			// 			EnumerationContext: "5C000000-0000-0000-0000-000000000000",
-			// 		},
-			// 	},
-			// },
+			{
+				"should create a valid AMT_SetupAndConfigurationService Enumerate wsman message", 
+				"AMT_SetupAndConfigurationService", 
+				wsmantesting.ENUMERATE, 
+				wsmantesting.ENUMERATE_BODY, 
+				func() (Response, error) {
+					currentMessage = "Enumerate"
+					return elementUnderTest.Enumerate()
+				}, 
+				Body{
+					XMLName: xml.Name{Space: "http://www.w3.org/2003/05/soap-envelope", Local: "Body"},
+					EnumerateResponse: common.EnumerateResponse{
+						EnumerationContext: "D3000000-0000-0000-0000-000000000000",
+					},
+				},
+			},
 			//PULLS
 			// {"should create a valid AMT_SetupAndConfigurationService Pull wsman message", "AMT_SetupAndConfigurationService", wsmantesting.PULL, wsmantesting.PULL_BODY, func() string { return elementUnderTest.Pull(wsmantesting.EnumerationContext) }},
 			// {"should create a valid AMT_SetupAndConfigurationService CommitChanges wsman message", "AMT_SetupAndConfigurationService", string(actions.CommitChanges), `<h:CommitChanges_INPUT xmlns:h="http://intel.com/wbem/wscim/1/amt-schema/1/AMT_SetupAndConfigurationService"></h:CommitChanges_INPUT>`, elementUnderTest.CommitChanges},
@@ -93,7 +127,7 @@ func TestAMT_SetupAndConfigurationService(t *testing.T) {
 				expectedXMLInput := wsmantesting.ExpectedResponse(messageID, resourceUriBase, test.method, test.action, "", test.body)
 				messageID++
 				response, err := test.responseFunc()
-				println(response.XMLOutput)
+				//println(response.XMLOutput)
 				assert.NoError(t, err)
 				assert.Equal(t, expectedXMLInput, response.XMLInput)
 				assert.Equal(t, test.expectedResponse, response.Body)
diff --git a/pkg/wsmantesting/responses/amt/setupandconfiguration/enumerate.xml b/pkg/wsmantesting/responses/amt/setupandconfiguration/enumerate.xml
new file mode 100644
index 00000000..03c9fef2
--- /dev/null
+++ b/pkg/wsmantesting/responses/amt/setupandconfiguration/enumerate.xml
@@ -0,0 +1,22 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<a:Envelope xmlns:a="http://www.w3.org/2003/05/soap-envelope"
+    xmlns:b="http://schemas.xmlsoap.org/ws/2004/08/addressing"
+    xmlns:c="http://schemas.dmtf.org/wbem/wsman/1/wsman.xsd"
+    xmlns:d="http://schemas.xmlsoap.org/ws/2005/02/trust"
+    xmlns:e="http://docs.oasis-open.org/wss/2004/01/oasis-200401-wss-wssecurity-secext-1.0.xsd"
+    xmlns:f="http://schemas.dmtf.org/wbem/wsman/1/cimbinding.xsd"
+    xmlns:g="http://schemas.xmlsoap.org/ws/2004/09/enumeration"
+    xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
+    <a:Header>
+        <b:To>http://schemas.xmlsoap.org/ws/2004/08/addressing/role/anonymous</b:To>
+        <b:RelatesTo>0</b:RelatesTo>
+        <b:Action a:mustUnderstand="true">http://schemas.xmlsoap.org/ws/2004/09/enumeration/EnumerateResponse</b:Action>
+        <b:MessageID>uuid:00000000-8086-8086-8086-000000000322</b:MessageID>
+        <c:ResourceURI>http://intel.com/wbem/wscim/1/amt-schema/1/AMT_SetupAndConfigurationService</c:ResourceURI>
+    </a:Header>
+    <a:Body>
+        <g:EnumerateResponse>
+            <g:EnumerationContext>D3000000-0000-0000-0000-000000000000</g:EnumerationContext>
+        </g:EnumerateResponse>
+    </a:Body>
+</a:Envelope>
\ No newline at end of file
diff --git a/pkg/wsmantesting/responses/amt/setupandconfiguration/get.xml b/pkg/wsmantesting/responses/amt/setupandconfiguration/get.xml
index d54b88a4..ae6c23bf 100644
--- a/pkg/wsmantesting/responses/amt/setupandconfiguration/get.xml
+++ b/pkg/wsmantesting/responses/amt/setupandconfiguration/get.xml
@@ -1,4 +1,4 @@
-<?xml version="1.0" encoding="UTF-8"?>
+<?xml version="1.0" encoding="UTF-8"?> 
 <a:Envelope xmlns:a="http://www.w3.org/2003/05/soap-envelope"
     xmlns:b="http://schemas.xmlsoap.org/ws/2004/08/addressing"
     xmlns:c="http://schemas.dmtf.org/wbem/wsman/1/wsman.xsd"