golang-github-dylanmei-winrmtest-dev 0.0~git20151226.0256178-1 / usr / share / gocode / src / github.com / dylanmei / winrmtest / remote.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
}