schnutibox/integration_test.go

// +build integration

package main_test

import (
	"fmt"
	"io"
	"io/ioutil"
	"os"
	"testing"
	"time"

	sshclient "github.com/helloyi/go-sshclient"
	"github.com/ory/dockertest/v3"
	"github.com/rs/zerolog"
	"github.com/rs/zerolog/log"
	"github.com/tmc/scp"
	"golang.org/x/crypto/ssh"
)

const (
	sdcard   = "/home/marv/tmp/2021-03-04-raspios-buster-armhf-lite.img"
	sshUser  = "pi"
	sshPass  = "raspberry"
	sshHost  = "localhost"
	bootWait = 3 * time.Minute
)

// raspbianWorkCopy creates a temp image file.
func raspbianWorkCopy() (string, error) {
	f, err := ioutil.TempFile("", "schnutibox")
	if err != nil {
		return "", fmt.Errorf("could not create tempfile: %w", err)
	}

	o, err := os.Open(sdcard)
	if err != nil {
		return "", fmt.Errorf("could not open sdcard image: %w", err)
	}

	defer o.Close()

	if _, err := io.Copy(f, o); err != nil {
		return "", fmt.Errorf("could not copy image: %w", err)
	}

	return f.Name(), nil
}

func copySchnutibox(user, pass, host string) error {
	// Local file.
	pwd, err := os.Getwd()
	if err != nil {
		return fmt.Errorf("could not get pwd: %w", err)
	}

	binPath := pwd + "/dist/schnutibox_linux_arm_6/schnutibox"

	config := &ssh.ClientConfig{
		User: user,
		Auth: []ssh.AuthMethod{
			ssh.Password(pass),
		},
		HostKeyCallback: ssh.InsecureIgnoreHostKey(),
	}

	client, err := ssh.Dial("tcp", host, config)
	if err != nil {
		return fmt.Errorf("could not create ssh client: %w", err)
	}

	session, err := client.NewSession()
	if err != nil {
		return fmt.Errorf("could not create session: %w", err)
	}

	log.Debug().Str("schnutibox path", binPath).Msg("copy binary")

	if err := scp.CopyPath(binPath, "/tmp/schnutibox", session); err != nil {
		return fmt.Errorf("could not scp binary: %w", err)
	}

	return nil
}

//nolint:funlen
func TestIntegration(t *testing.T) {
	t.Parallel()

	log.Logger = log.Output(zerolog.ConsoleWriter{Out: os.Stderr})

	// Create tmp image.
	img, err := raspbianWorkCopy()
	if err != nil {
		log.Error().Err(err).Msg("could not create temp work image")
	}

	log.Info().Str("img", img).Msg("created temp image file")

	pool, err := dockertest.NewPool("")
	if err != nil {
		log.Error().Err(err).Msg("could not connect to docker")
		t.Fatal()
	}

	if err != nil {
		log.Error().Err(err).Msg("could not get pwd")
		t.Fatal()
	}

	log.Info().Msg("starting container")

	resource, err := pool.RunWithOptions(
		&dockertest.RunOptions{
			Repository:   "lukechilds/dockerpi",
			Tag:          "vm",
			Mounts:       []string{img + ":/sdcard/filesystem.img"},
			ExposedPorts: []string{"5022/tcp"},
		})
	if err != nil {
		log.Error().Err(err).Msg("could not start resource")
		t.Fatal()
	}

	// Register cleanup.
	t.Cleanup(func() {
		log.Info().Msg("getting rid of container")
		if err := pool.Purge(resource); err != nil {
			log.Error().Err(err).Msg("could not cleanup")
			t.Fatal()
		}

		log.Info().Str("img", img).Msg("deleting temp image")
		if err := os.Remove(img); err != nil {
			log.Error().Err(err).Msg("could not delete temp image")
			t.Fatal()
		}
	})

	// Starting container.
	log.Info().Msg("waiting to be ready")

	sshConn := sshHost + ":" + resource.GetPort("5022/tcp")

	if err := pool.Retry(func() error {
		time.Sleep(bootWait)
		client, err := sshclient.DialWithPasswd(sshConn, sshUser, sshPass)
		if err != nil {
			return err
		}

		if err := client.Close(); err != nil {
			return err
		}

		return nil
	}); err != nil {
		log.Error().Err(err).Msg("could not connect to container via ssh")
		t.Fatal()
	}

	// Connect via SSH.
	log.Info().Msg("connect via ssh")

	client, err := sshclient.DialWithPasswd(sshConn, sshUser, sshPass)
	if err != nil {
		log.Error().Err(err).Msg("could not connect via ssh")
		t.Fatal()
	}

	// Copy schnutibox binary to container.
	log.Info().
		Str("user", sshUser).
		Str("pass", sshPass).
		Str("conn", sshConn).
		Msg("copy schnutibox")

	if err := copySchnutibox(sshUser, sshPass, sshConn); err != nil {
		log.Error().Err(err).Msg("could not copy schnutibox")
		t.Fatal()
	}

	// Move schnutibox to /usr/local/bin
	log.Info().Msg("move binary to /usr/local/bin")

	if err := client.
		Cmd("sudo mv /tmp/schnutibox /usr/local/bin/schnutibox").
		SetStdio(os.Stdout, os.Stderr).
		Run(); err != nil {
		log.Error().Err(err).Msg("could not create /usr/local/bin on container")
		t.Fatal()
	}

	// Doing the testing.
	log.Info().Msg("doing to testing")

	if err := client.
		Cmd("sudo schnutibox prepare --read-only").
		SetStdio(os.Stdout, os.Stderr).
		Run(); err != nil {
		log.Error().Err(err).Msg("could not run command")
		t.Fatal()
	}
}