odo/tests/helper/helper_dev.go

package helper

import (
	"os"
	"regexp"
	"time"

	"github.com/ActiveState/termtest/expect"
	"github.com/onsi/gomega"
	. "github.com/onsi/gomega"
	"github.com/onsi/gomega/gexec"
)

// DevSession represents a session running `odo dev`
/*
	It can be used in different ways:

	# Starting a session for a series of tests and stopping the session after the tests:

	This format can be used when you want to run several independent tests
	when the `odo dev` command is running in the background

	```
	When("running dev session", func() {
		var devSession DevSession
		var outContents []byte
		var errContents []byte
		BeforeEach(func() {
			devSession, outContents, errContents = helper.StartDevMode(nil)
		})
		AfterEach(func() {
			devSession.Stop()
		})

		It("...", func() {
			// Test with `dev odo` running in the background
			// outContents and errContents are contents of std/err output when dev mode is started
		})
		It("...", func() {
			// Test with `dev odo` running in the background
		})
	})

	# Starting a session and stopping it cleanly

	This format can be used to test the behaviour of `odo dev` when it is stopped cleanly

	When("running dev session and stopping it with cleanup", func() {
		var devSession DevSession
		var outContents []byte
		var errContents []byte
		BeforeEach(func() {
			devSession, outContents, errContents = helper.StartDevMode(nil)
			defer devSession.Stop()
			[...]
		})

		It("...", func() {
			// Test after `odo dev` has been stopped cleanly
			// outContents and errContents are contents of std/err output when dev mode is started
		})
		It("...", func() {
			// Test after `odo dev` has been stopped cleanly
		})
	})

	# Starting a session and stopping it immediately without cleanup

	This format can be used to test the behaviour of `odo dev` when it is stopped with a KILL signal

	When("running dev session and stopping it without cleanup", func() {
		var devSession DevSession
		var outContents []byte
		var errContents []byte
		BeforeEach(func() {
			devSession, outContents, errContents = helper.StartDevMode(nil)
			defer devSession.Kill()
			[...]
		})

		It("...", func() {
			// Test after `odo dev` has been killed
			// outContents and errContents are contents of std/err output when dev mode is started
		})
		It("...", func() {
			// Test after `odo dev` has been killed
		})
	})


	# Running a dev session and executing some tests inside this session

	This format can be used to run a series of related tests in dev mode
	All tests will be ran in the same session (ideal for e2e tests)
	To run independent tests, previous formats should be used instead.

	It("should do ... in dev mode", func() {
		helper.RunDevMode(func(session *gexec.Session, outContents []byte, errContents []byte, ports map[string]string) {
			// test on dev mode
			// outContents and errContents are contents of std/err output when dev mode is started
			// ports contains a map where keys are container ports and associated values are local IP:port redirecting to these local ports
		})
	})

	# Waiting for file synchronisation to finish

	The method session.WaitSync() can be used to wait for the synchronization of files to finish.
	The method returns the contents of std/err output since the end of the dev mode started or previous sync, and until the end of the synchronization.
*/

type DevSession struct {
	session *gexec.Session
	stopped bool
	console *expect.Console
}

// StartDevMode starts a dev session with `odo dev`
// It returns a session structure, the contents of the standard and error outputs
// and the redirections endpoints to access ports opened by component
// when the dev mode is completely started
func StartDevMode(envvars []string, opts ...string) (DevSession, []byte, []byte, map[string]string, error) {

	c, err := expect.NewConsole(expect.WithStdout(os.Stdout))
	if err != nil {
		return DevSession{}, nil, nil, nil, err
	}

	args := []string{"dev", "--random-ports"}
	args = append(args, opts...)
	cmd := Cmd("odo", args...)
	cmd.Cmd.Stdin = c.Tty()
	cmd.Cmd.Stdout = c.Tty()
	cmd.Cmd.Stderr = c.Tty()

	session := cmd.AddEnv(envvars...).Runner().session
	WaitForOutputToContain("[Ctrl+c] - Exit", 360, 10, session)
	result := DevSession{
		session: session,
		console: c,
	}
	outContents := session.Out.Contents()
	errContents := session.Err.Contents()
	err = session.Out.Clear()
	if err != nil {
		return DevSession{}, nil, nil, nil, err
	}
	err = session.Err.Clear()
	if err != nil {
		return DevSession{}, nil, nil, nil, err
	}
	return result, outContents, errContents, getPorts(string(outContents)), nil

}

// Kill a Dev session abruptly, without handling any cleanup
func (o DevSession) Kill() {
	if o.console != nil {
		err := o.console.Close()
		gomega.Expect(err).NotTo(gomega.HaveOccurred())
	}
	o.session.Kill()
}

// Stop a Dev session cleanly (equivalent as hitting Ctrl-c)
func (o *DevSession) Stop() {
	if o.console != nil {
		err := o.console.Close()
		gomega.Expect(err).NotTo(gomega.HaveOccurred())
	}
	if o.stopped {
		return
	}
	err := terminateProc(o.session)
	gomega.Expect(err).NotTo(gomega.HaveOccurred())
	o.stopped = true
}

func (o *DevSession) PressKey(p byte) {
	if o.console == nil {
		return
	}
	_, err := o.console.Write([]byte{p})
	Expect(err).ToNot(HaveOccurred())
}

func (o DevSession) WaitEnd() {
	o.session.Wait(3 * time.Minute)
}

// WaitSync waits for the synchronization of files to be finished
// It returns the contents of the standard and error outputs
// and the list of forwarded ports
// since the end of the dev mode or the last time WaitSync/GetInfo has been called
func (o DevSession) WaitSync() ([]byte, []byte, map[string]string, error) {
	WaitForOutputToContainOne([]string{"Pushing files...", "Updating Component..."}, 180, 10, o.session)
	WaitForOutputToContain("Watching for changes in the current directory", 240, 10, o.session)
	return o.GetInfo()
}

func (o DevSession) WaitRestartPortforward() ([]byte, []byte, map[string]string, error) {
	WaitForOutputToContain("Forwarding from", 30, 5, o.session)
	return o.GetInfo()
}

// GetInfo returns the contents of the standard and error outputs
// and the list of forwarded ports
// since the end of the dev mode or the last time WaitSync/GetInfo has been called
func (o DevSession) GetInfo() ([]byte, []byte, map[string]string, error) {
	outContents := o.session.Out.Contents()
	errContents := o.session.Err.Contents()
	err := o.session.Out.Clear()
	if err != nil {
		return nil, nil, nil, err
	}
	err = o.session.Err.Clear()
	if err != nil {
		return nil, nil, nil, err
	}
	return outContents, errContents, getPorts(string(outContents)), nil
}

func (o DevSession) CheckNotSynced(timeout time.Duration) {
	Consistently(func() string {
		return string(o.session.Out.Contents())
	}, timeout).ShouldNot(ContainSubstring("Pushing files..."))
}

// RunDevMode runs a dev session and executes the `inside` code when the dev mode is completely started
// The inside handler is passed the internal session pointer, the contents of the standard and error outputs,
// and a slice of strings - ports - giving the redirections in the form localhost:<port_number> to access ports opened by component
func RunDevMode(additionalOpts []string, envvars []string, inside func(session *gexec.Session, outContents []byte, errContents []byte, ports map[string]string)) error {
	session, outContents, errContents, urls, err := StartDevMode(envvars, additionalOpts...)
	if err != nil {
		return err
	}
	defer func() {
		session.Stop()
		session.WaitEnd()
	}()
	inside(session.session, outContents, errContents, urls)
	return nil
}

// DevModeShouldFail runs `odo dev` with an intention to fail, and checks for a given substring
// `odo dev` runs in an infinite reconciliation loop, and hence running it with Cmd will not work for a lot of failing cases,
// this function is helpful in such cases.
// TODO(pvala): Modify StartDevMode to take substring arg into account, and replace this method with it.
func DevModeShouldFail(envvars []string, substring string, opts ...string) (DevSession, []byte, []byte, error) {
	args := []string{"dev", "--random-ports"}
	args = append(args, opts...)
	session := Cmd("odo", args...).AddEnv(envvars...).Runner().session
	WaitForOutputToContain(substring, 360, 10, session)
	result := DevSession{
		session: session,
	}
	defer func() {
		result.Stop()
		result.WaitEnd()
	}()
	outContents := session.Out.Contents()
	errContents := session.Err.Contents()
	err := session.Out.Clear()
	if err != nil {
		return DevSession{}, nil, nil, err
	}
	err = session.Err.Clear()
	if err != nil {
		return DevSession{}, nil, nil, err
	}
	return result, outContents, errContents, nil
}

// getPorts returns a map of ports redirected depending on the information in s
//
//	`- Forwarding from 127.0.0.1:40001 -> 3000` will return { "3000": "127.0.0.1:40001" }
func getPorts(s string) map[string]string {
	result := map[string]string{}
	re := regexp.MustCompile("(127.0.0.1:[0-9]+) -> ([0-9]+)")
	matches := re.FindAllStringSubmatch(s, -1)
	for _, match := range matches {
		result[match[2]] = match[1]
	}
	return result
}