Skip to content

Commit

Permalink
feat(amt): adds execute support for setup
Browse files Browse the repository at this point in the history
  • Loading branch information
@zaidusmani26
zaidusmani26 committed Oct 25, 2023
1 parent 91bdaba commit 187c8d1
Show file tree
Hide file tree
Showing 4 changed files with 93 additions and 37 deletions.
2 changes: 1 addition & 1 deletion pkg/amt/setupandconfiguration/service.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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
}
Expand Down
104 changes: 69 additions & 35 deletions pkg/amt/setupandconfiguration/service_test.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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 {
Expand All @@ -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},
Expand All @@ -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)
Expand Down
22 changes: 22 additions & 0 deletions pkg/wsmantesting/responses/amt/setupandconfiguration/enumerate.xml
View file
Original file line number Diff line number Diff line change
@@ -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>
2 changes: 1 addition & 1 deletion pkg/wsmantesting/responses/amt/setupandconfiguration/get.xml
View file
Original file line number Diff line number Diff line change
@@ -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"
Expand Down

0 comments on commit 187c8d1

Please sign in to comment.