80 lines
1.6 KiB
Go
80 lines
1.6 KiB
Go
|
package winrmtest
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"net/url"
|
||
|
|
"regexp"
|
||
|
|
"strconv"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Remote respresents a WinRM server
|
||
|
|
type Remote struct {
|
||
|
|
Host string
|
||
|
|
Port int
|
||
|
|
server *httptest.Server
|
||
|
|
service *wsman
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewRemote returns a new initialized Remote
|
||
|
|
func NewRemote() *Remote {
|
||
|
|
mux := http.NewServeMux()
|
||
|
|
srv := httptest.NewServer(mux)
|
||
|
|
|
||
|
|
host, port, _ := splitAddr(srv.URL)
|
||
|
|
remote := Remote{
|
||
|
|
Host: host,
|
||
|
|
Port: port,
|
||
|
|
server: srv,
|
||
|
|
service: &wsman{},
|
||
|
|
}
|
||
|
|
|
||
|
|
mux.Handle("/wsman", remote.service)
|
||
|
|
return &remote
|
||
|
|
}
|
||
|
|
|
||
|
|
// Close closes the WinRM server
|
||
|
|
func (r *Remote) Close() {
|
||
|
|
r.server.Close()
|
||
|
|
}
|
||
|
|
|
||
|
|
// MatcherFunc respresents a function used to match WinRM commands
|
||
|
|
type MatcherFunc func(candidate string) bool
|
||
|
|
|
||
|
|
// MatchText return a new MatcherFunc based on text matching
|
||
|
|
func MatchText(text string) MatcherFunc {
|
||
|
|
return func(candidate string) bool {
|
||
|
|
return text == candidate
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// MatchPattern return a new MatcherFunc based on pattern matching
|
||
|
|
func MatchPattern(pattern string) MatcherFunc {
|
||
|
|
r := regexp.MustCompile(pattern)
|
||
|
|
return func(candidate string) bool {
|
||
|
|
return r.MatchString(candidate)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// CommandFunc respresents a function used to mock WinRM commands
|
||
|
|
type CommandFunc func(out, err io.Writer) (exitCode int)
|
||
|
|
|
||
|
|
// CommandFunc adds a WinRM command mock function to the WinRM server
|
||
|
|
func (r *Remote) CommandFunc(m MatcherFunc, f CommandFunc) {
|
||
|
|
r.service.HandleCommand(m, f)
|
||
|
|
}
|
||
|
|
|
||
|
|
func splitAddr(addr string) (host string, port int, err error) {
|
||
|
|
u, err := url.Parse(addr)
|
||
|
|
if err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
split := strings.Split(u.Host, ":")
|
||
|
|
host = split[0]
|
||
|
|
port, err = strconv.Atoi(split[1])
|
||
|
|
return
|
||
|
|
}
|