blob: d7a642112cb6aa8fc1065105b0e692945ac539c2 [file] [log] [blame]
package main
import (
"bytes"
"io/ioutil"
"os"
"strings"
"testing"
"veyron.io/veyron/veyron2"
"veyron.io/veyron/veyron2/ipc"
"veyron.io/veyron/veyron2/naming"
"veyron.io/veyron/veyron2/rt"
"veyron.io/veyron/veyron2/security"
"veyron.io/veyron/veyron2/services/mgmt/application"
"veyron.io/veyron/veyron2/vlog"
"veyron.io/veyron/veyron/profiles"
"veyron.io/veyron/veyron/services/mgmt/repository"
)
var (
envelope = application.Envelope{
Title: "fifa world cup",
Args: []string{"arg1", "arg2", "arg3"},
Binary: "/path/to/binary",
Env: []string{"env1", "env2", "env3"},
Packages: map[string]string{
"pkg1": "/path/to/package1",
},
}
jsonEnv = `{
"Title": "fifa world cup",
"Args": [
"arg1",
"arg2",
"arg3"
],
"Binary": "/path/to/binary",
"Env": [
"env1",
"env2",
"env3"
],
"Packages": {
"pkg1": "/path/to/package1"
}
}`
)
type server struct {
suffix string
}
func (s *server) Match(_ ipc.ServerContext, profiles []string) (application.Envelope, error) {
vlog.VI(2).Infof("%v.Match(%v) was called", s.suffix, profiles)
return envelope, nil
}
func (s *server) Put(_ ipc.ServerContext, profiles []string, env application.Envelope) error {
vlog.VI(2).Infof("%v.Put(%v, %v) was called", s.suffix, profiles, env)
return nil
}
func (s *server) Remove(_ ipc.ServerContext, profile string) error {
vlog.VI(2).Infof("%v.Remove(%v) was called", s.suffix, profile)
return nil
}
type dispatcher struct {
}
func NewDispatcher() *dispatcher {
return &dispatcher{}
}
func (d *dispatcher) Lookup(suffix string) (interface{}, security.Authorizer, error) {
return repository.ApplicationServer(&server{suffix: suffix}), nil, nil
}
func startServer(t *testing.T, r veyron2.Runtime) (ipc.Server, naming.Endpoint, error) {
dispatcher := NewDispatcher()
server, err := r.NewServer()
if err != nil {
t.Errorf("NewServer failed: %v", err)
return nil, nil, err
}
endpoint, err := server.Listen(profiles.LocalListenSpec)
if err != nil {
t.Errorf("Listen failed: %v", err)
return nil, nil, err
}
if err := server.ServeDispatcher("", dispatcher); err != nil {
t.Errorf("Serve failed: %v", err)
return nil, nil, err
}
return server, endpoint, nil
}
func stopServer(t *testing.T, server ipc.Server) {
if err := server.Stop(); err != nil {
t.Errorf("server.Stop failed: %v", err)
}
}
func TestApplicationClient(t *testing.T) {
var err error
runtime, err = rt.New()
if err != nil {
t.Fatalf("Unexpected error initializing runtime: %s", err)
}
defer runtime.Cleanup()
server, endpoint, err := startServer(t, runtime)
if err != nil {
return
}
defer stopServer(t, server)
// Setup the command-line.
cmd := root()
var stdout, stderr bytes.Buffer
cmd.Init(nil, &stdout, &stderr)
appName := naming.JoinAddressName(endpoint.String(), "myapp/1")
profile := "myprofile"
// Test the 'Match' command.
if err := cmd.Execute([]string{"match", appName, profile}); err != nil {
t.Fatalf("%v", err)
}
if expected, got := jsonEnv, strings.TrimSpace(stdout.String()); got != expected {
t.Errorf("Unexpected output from match. Got %q, expected %q", got, expected)
}
stdout.Reset()
// Test the 'put' command.
f, err := ioutil.TempFile("", "test")
if err != nil {
t.Fatalf("%v", err)
}
fileName := f.Name()
defer os.Remove(fileName)
if _, err = f.Write([]byte(jsonEnv)); err != nil {
t.Fatalf("%v", err)
}
if err = f.Close(); err != nil {
t.Fatalf("%v", err)
}
if err := cmd.Execute([]string{"put", appName, profile, fileName}); err != nil {
t.Fatalf("%v", err)
}
if expected, got := "Application envelope added successfully.", strings.TrimSpace(stdout.String()); got != expected {
t.Errorf("Unexpected output from put. Got %q, expected %q", got, expected)
}
stdout.Reset()
// Test the 'remove' command.
if err := cmd.Execute([]string{"remove", appName, profile}); err != nil {
t.Fatalf("%v", err)
}
if expected, got := "Application envelope removed successfully.", strings.TrimSpace(stdout.String()); got != expected {
t.Errorf("Unexpected output from remove. Got %q, expected %q", got, expected)
}
stdout.Reset()
// Test the 'edit' command. (nothing changed)
os.Setenv("EDITOR", "true")
if err := cmd.Execute([]string{"edit", appName, profile}); err != nil {
t.Fatalf("%v", err)
}
if expected, got := "Nothing changed", strings.TrimSpace(stdout.String()); got != expected {
t.Errorf("Unexpected output from edit. Got %q, expected %q", got, expected)
}
stdout.Reset()
// Test the 'edit' command.
os.Setenv("EDITOR", "perl -pi -e 's/arg1/arg111/'")
if err := cmd.Execute([]string{"edit", appName, profile}); err != nil {
t.Fatalf("%v", err)
}
if expected, got := "Application envelope updated successfully.", strings.TrimSpace(stdout.String()); got != expected {
t.Errorf("Unexpected output from edit. Got %q, expected %q", got, expected)
}
stdout.Reset()
}