Skip to content

Commit

Permalink
feat(amt): add execute support for setup
Browse files Browse the repository at this point in the history
  • Loading branch information
@zaidusmani26
zaidusmani26 committed Oct 11, 2023
1 parent 5ec768a commit 91bdaba
Show file tree
Hide file tree
Showing 3 changed files with 188 additions and 19 deletions.
95 changes: 91 additions & 4 deletions pkg/amt/setupandconfiguration/service.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -7,15 +7,47 @@ package setupandconfiguration

import (
"encoding/xml"
"encoding/json"
"fmt"

"github.com/open-amt-cloud-toolkit/go-wsman-messages/internal/message"
"github.com/open-amt-cloud-toolkit/go-wsman-messages/pkg/amt/actions"
"github.com/open-amt-cloud-toolkit/go-wsman-messages/pkg/cim/models"
"github.com/open-amt-cloud-toolkit/go-wsman-messages/pkg/wsman"
"github.com/open-amt-cloud-toolkit/go-wsman-messages/pkg/common"
)

const AMT_SetupAndConfigurationService = "AMT_SetupAndConfigurationService"

type (
Response struct {
*wsman.Message
XMLName xml.Name `xml:"Envelope"`
Header message.Header `xml:"Header"`
Body Body `xml:"Body"`
}
Body struct {
XMLName xml.Name `xml:"Body"`
Setup Setup `xml:"AMT_AuthorizationService"`

EnumerateResponse common.EnumerateResponse
}

Setup struct{
CreationClassName string
ElementName string
EnabledState int
Name string
PasswordModel int
ProvisioningMode int
ProvisioningServerOTP string
ProvisioningState int
RequestedState int
SystemCreationClassName string
SystemName string
ZeroTouchConfigurationEnabled bool
}
)
type UnprovisionResponse struct {
XMLName xml.Name `xml:"Envelope"`
Header message.Header `xml:"Header"`
Expand Down Expand Up @@ -48,22 +80,77 @@ type SetupAndConfigurationService struct {
ZeroTouchConfigurationEnabled string
}
}

func (w *Response) JSON() string {
jsonOutput, err := json.Marshal(w.Body)
if err != nil {
return ""
}
return string(jsonOutput)
}

type Service struct {
base message.Base
client wsman.WSManClient
}

func NewSetupAndConfigurationService(wsmanMessageCreator *message.WSManMessageCreator) Service {
return Service{
base: message.NewBase(wsmanMessageCreator, AMT_SetupAndConfigurationService),
client: nil,
}
}

func NewSetupAndConfigurationServiceWithClient(wsmanMessageCreator *message.WSManMessageCreator, client wsman.WSManClient) Service {
return Service{
base: message.NewBaseWithClient(wsmanMessageCreator, AMT_SetupAndConfigurationService, client),
client: client,
}
}
func (s Service) Get() string {
return s.base.Get(nil)
func (s Service) Get() (response Response, err error) {

response = Response{
Message: &wsman.Message{
XMLInput: s.base.Get(nil),
},
}

// send the message to AMT
err = s.base.Execute(response.Message)
if err != nil {
return
}

// put the xml response into the go struct
err = xml.Unmarshal([]byte(response.XMLOutput), &response)
if err != nil {
return
}

return
}


// Enumerates the instances of this class
func (s Service) Enumerate() string {
return s.base.Enumerate()
func (s Service) Enumerate() (response Response, err error) {
response = Response{
Message: &wsman.Message{
XMLInput: s.base.Enumerate(),
},
}
// send the message to AMT
err = s.base.Execute(response.Message)
if err != nil {
return
}

// put the xml response into the go struct
err = xml.Unmarshal([]byte(response.XMLOutput), &response)
if err != nil {
return
}

return
}

// Pulls instances of this class, following an Enumerate operation
Expand Down
78 changes: 63 additions & 15 deletions pkg/amt/setupandconfiguration/service_test.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -6,49 +6,97 @@
package setupandconfiguration

import (
"encoding/xml"
"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/amt/actions"
"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"
)

func TestAMT_SetupAndConfigurationService(t *testing.T) {
messageID := 0
resourceUriBase := "http://intel.com/wbem/wscim/1/amt-schema/1/"
wsmanMessageCreator := message.NewWSManMessageCreator(resourceUriBase)
elementUnderTest := NewSetupAndConfigurationService(wsmanMessageCreator)
//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)

t.Run("amt_* Tests", func(t *testing.T) {
tests := []struct {
name string
method string
action string
body string
responseFunc func() string
responseFunc func() (Response, error)
expectedResponse interface{}
}{
//GETS
{"should create a valid AMT_SetupAndConfigurationService Get wsman message", "AMT_SetupAndConfigurationService", wsmantesting.GET, "", elementUnderTest.Get},
{
"should create a valid AMT_SetupAndConfigurationService Get wsman message",
"AMT_SetupAndConfigurationService",
wsmantesting.GET, "",
func() (Response, error) {
//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,
},
},
},
//ENUMERATES
{"should create a valid AMT_SetupAndConfigurationService Enumerate wsman message", "AMT_SetupAndConfigurationService", wsmantesting.ENUMERATE, wsmantesting.ENUMERATE_BODY, elementUnderTest.Enumerate},
// {
// "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",
// },
// },
// },
//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},
{"should create a valid AMT_SetupAndConfigurationService GetUuid wsman message", "AMT_SetupAndConfigurationService", string(actions.GetUuid), `<h:GetUuid_INPUT xmlns:h="http://intel.com/wbem/wscim/1/amt-schema/1/AMT_SetupAndConfigurationService"></h:GetUuid_INPUT>`, elementUnderTest.GetUuid},
{"should create a valid AMT_SetupAndConfigurationService SetMEBxPassword wsman message", "AMT_SetupAndConfigurationService", string(actions.SetMEBxPassword), `<h:SetMEBxPassword_INPUT xmlns:h="http://intel.com/wbem/wscim/1/amt-schema/1/AMT_SetupAndConfigurationService"><h:Password>P@ssw0rd</h:Password></h:SetMEBxPassword_INPUT>`, func() string { return elementUnderTest.SetMEBXPassword("P@ssw0rd") }},
{"should create a valid AMT_SetupAndConfigurationService Unprovision wsman message", "AMT_SetupAndConfigurationService", string(actions.Unprovision), `<h:Unprovision_INPUT xmlns:h="http://intel.com/wbem/wscim/1/amt-schema/1/AMT_SetupAndConfigurationService"><h:ProvisioningMode>1</h:ProvisioningMode></h:Unprovision_INPUT>`, func() string { return elementUnderTest.Unprovision(1) }},
// {"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},
// {"should create a valid AMT_SetupAndConfigurationService GetUuid wsman message", "AMT_SetupAndConfigurationService", string(actions.GetUuid), `<h:GetUuid_INPUT xmlns:h="http://intel.com/wbem/wscim/1/amt-schema/1/AMT_SetupAndConfigurationService"></h:GetUuid_INPUT>`, elementUnderTest.GetUuid},
// {"should create a valid AMT_SetupAndConfigurationService SetMEBxPassword wsman message", "AMT_SetupAndConfigurationService", string(actions.SetMEBxPassword), `<h:SetMEBxPassword_INPUT xmlns:h="http://intel.com/wbem/wscim/1/amt-schema/1/AMT_SetupAndConfigurationService"><h:Password>P@ssw0rd</h:Password></h:SetMEBxPassword_INPUT>`, func() string { return elementUnderTest.SetMEBXPassword("P@ssw0rd") }},
// {"should create a valid AMT_SetupAndConfigurationService Unprovision wsman message", "AMT_SetupAndConfigurationService", string(actions.Unprovision), `<h:Unprovision_INPUT xmlns:h="http://intel.com/wbem/wscim/1/amt-schema/1/AMT_SetupAndConfigurationService"><h:ProvisioningMode>1</h:ProvisioningMode></h:Unprovision_INPUT>`, func() string { return elementUnderTest.Unprovision(1) }},
}

for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
correctResponse := wsmantesting.ExpectedResponse(messageID, resourceUriBase, test.method, test.action, "", test.body)
expectedXMLInput := wsmantesting.ExpectedResponse(messageID, resourceUriBase, test.method, test.action, "", test.body)
messageID++
response := test.responseFunc()
if response != correctResponse {
assert.Equal(t, correctResponse, response)
}
response, err := test.responseFunc()
println(response.XMLOutput)
assert.NoError(t, err)
assert.Equal(t, expectedXMLInput, response.XMLInput)
assert.Equal(t, test.expectedResponse, response.Body)
})
}
})
Expand Down
34 changes: 34 additions & 0 deletions pkg/wsmantesting/responses/amt/setupandconfiguration/get.xml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,34 @@
<?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://intel.com/wbem/wscim/1/amt-schema/1/AMT_SetupAndConfigurationService"
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">http://schemas.xmlsoap.org/ws/2004/09/transfer/GetResponse</b:Action>
<b:MessageID>uuid:00000000-8086-8086-8086-000000000332</b:MessageID>
<c:ResourceURI>http://intel.com/wbem/wscim/1/amt-schema/1/AMT_SetupAndConfigurationService</c:ResourceURI>
</a:Header>
<a:Body>
<g:AMT_SetupAndConfigurationService>
<g:CreationClassName>AMT_SetupAndConfigurationService</g:CreationClassName>
<g:ElementName>Intel(r) AMT Setup and Configuration Service</g:ElementName>
<g:EnabledState>5</g:EnabledState>
<g:Name>Intel(r) AMT Setup and Configuration Service</g:Name>
<g:PasswordModel>1</g:PasswordModel>
<g:ProvisioningMode>1</g:ProvisioningMode>
<g:ProvisioningServerOTP>AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA=</g:ProvisioningServerOTP>
<g:ProvisioningState>2</g:ProvisioningState>
<g:RequestedState>12</g:RequestedState>
<g:SystemCreationClassName>CIM_ComputerSystem</g:SystemCreationClassName>
<g:SystemName>Intel(r) AMT</g:SystemName>
<g:ZeroTouchConfigurationEnabled>true</g:ZeroTouchConfigurationEnabled>
</g:AMT_SetupAndConfigurationService>
</a:Body>
</a:Envelope>

0 comments on commit 91bdaba

Please sign in to comment.