go/src/v.io/x/mojo/tests/cmd/runtest.go

// Copyright 2015 The Vanadium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package main

import (
	"bufio"
	"bytes"
	"flag"
	"fmt"
	"io"
	"os"
	"strings"

	"v.io/x/mojo/tests/expected"
	"v.io/x/mojo/tests/util"
)

var runBench *bool = flag.Bool("bench", false, "run benchmarks instead of tests")
var clientType *string = flag.String("client", "go", "run test with a different client")
var serverType *string = flag.String("server", "go", "run test with a different server")

var (
	// Maps the client type to client mojo files.
	clientMap = map[string]string{
		"go":   "test_client.mojo",
		"dart": "dart-tests/end_to_end_test/lib/client.dart",
	}

	// Maps the server type to server mojo files.
	serverMap = map[string]string{
		"go":   "test_server.mojo",
		"dart": "dart-tests/end_to_end_test/lib/server.dart",
	}
)

func main() {
	flag.Parse()
	wd, err := os.Getwd()
	if err != nil {
		panic(err)
	}
	v23proxy, err := util.StartV23ServerProxy(wd)
	if err != nil {
		panic(err)
	}
	endpoint, err := v23proxy.Endpoint()
	if err != nil {
		panic(err)
	}
	endpointFlag := fmt.Sprintf("-endpoint=%s/https://mojo.v.io/%s/mojo::v23proxy::tests::V23ProxyTest", endpoint, serverMap[*serverType])
	args := []string{endpointFlag, "--v23.tcp.address=127.0.0.1:0"}
	if *runBench {
		args = append(args, "-test.run=XXXX", "-test.bench=.")
	}
	if err := runTestClient(wd, args...); err != nil {
		fmt.Fprintf(os.Stderr, "%s", err)
		os.Exit(2)
	}
	if err := v23proxy.Stop(); err != nil {
		panic(err)
	}
}

func runTestClient(v23ProxyRoot string, args ...string) error {
	cmd := util.RunMojoShellForV23ProxyTests(clientMap[*clientType], v23ProxyRoot, args)
	stdout, err := cmd.StdoutPipe()
	if err != nil {
		return err
	}
	var stdoutBuf bytes.Buffer
	go func() {
		io.Copy(os.Stdout, io.TeeReader(stdout, &stdoutBuf))
	}()
	stderr, err := cmd.StderrPipe()
	if err != nil {
		return err
	}
	go func() {
		io.Copy(os.Stderr, stderr)
	}()

	if err := cmd.Run(); err != nil {
		return err
	}

	scanner := bufio.NewScanner(bytes.NewReader(stdoutBuf.Bytes()))
	for scanner.Scan() {
		if strings.HasSuffix(scanner.Text(), expected.SuccessMessage) {
			return nil
		}
	}
	if err := scanner.Err(); err != nil {
		return err
	}
	return fmt.Errorf("TESTS FAILED")
}