mirror of
https://github.com/opentffoundation/opentf.git
synced 2026-03-13 10:01:08 -04:00
6fe2703665
* Remove `make updatedeps` from Travis build. We'll follow up with more specific plans around dependency updating in subsequent PRs. * Update all `make` targets to set `GO15VENDOREXPERIMENT=1` and to filter out `/vendor/` from `./...` where appropriate. * Temporarily remove `vet` from the `make test` target until we can figure out how to get it to not vet `vendor/`. (Initial experimentation failed to yield the proper incantation.) Everything is pinned to current master, with the exception of: * Azure/azure-sdk-for-go which is pinned before the breaking change today * aws/aws-sdk-go which is pinned to the most recent tag The documentation still needs to be updated, which we can do in a follow up PR. The goal here is to unblock release.
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
|
|
}
|