github.com/ari-anchor/sei-tendermint@v0.0.0-20230519144642-dc826b7b56bb/test/e2e/runner/cleanup.go (about)

     1  package main
     2  
     3  import (
     4  	"errors"
     5  	"fmt"
     6  	"os"
     7  	"path/filepath"
     8  
     9  	"github.com/ari-anchor/sei-tendermint/libs/log"
    10  	e2e "github.com/ari-anchor/sei-tendermint/test/e2e/pkg"
    11  )
    12  
    13  // Cleanup removes the Docker Compose containers and testnet directory.
    14  func Cleanup(logger log.Logger, testnet *e2e.Testnet) error {
    15  	err := cleanupDocker(logger)
    16  	if err != nil {
    17  		return err
    18  	}
    19  	return cleanupDir(logger, testnet.Dir)
    20  }
    21  
    22  // cleanupDocker removes all E2E resources (with label e2e=True), regardless
    23  // of testnet.
    24  func cleanupDocker(logger log.Logger) error {
    25  	logger.Info("Removing Docker containers and networks")
    26  
    27  	// GNU xargs requires the -r flag to not run when input is empty, macOS
    28  	// does this by default. Ugly, but works.
    29  	xargsR := `$(if [[ $OSTYPE == "linux-gnu"* ]]; then echo -n "-r"; fi)`
    30  
    31  	err := exec("bash", "-c", fmt.Sprintf(
    32  		"docker container ls -qa --filter label=e2e | xargs %v docker container rm -f", xargsR))
    33  	if err != nil {
    34  		return err
    35  	}
    36  
    37  	return exec("bash", "-c", fmt.Sprintf(
    38  		"docker network ls -q --filter label=e2e | xargs %v docker network rm", xargsR))
    39  }
    40  
    41  // cleanupDir cleans up a testnet directory
    42  func cleanupDir(logger log.Logger, dir string) error {
    43  	if dir == "" {
    44  		return errors.New("no directory set")
    45  	}
    46  
    47  	_, err := os.Stat(dir)
    48  	if os.IsNotExist(err) {
    49  		return nil
    50  	} else if err != nil {
    51  		return err
    52  	}
    53  
    54  	logger.Info(fmt.Sprintf("Removing testnet directory %q", dir))
    55  
    56  	// On Linux, some local files in the volume will be owned by root since Tendermint
    57  	// runs as root inside the container, so we need to clean them up from within a
    58  	// container running as root too.
    59  	absDir, err := filepath.Abs(dir)
    60  	if err != nil {
    61  		return err
    62  	}
    63  	err = execDocker("run", "--rm", "--entrypoint", "", "-v", fmt.Sprintf("%v:/network", absDir),
    64  		"tendermint/e2e-node", "sh", "-c", "rm -rf /network/*/")
    65  	if err != nil {
    66  		return err
    67  	}
    68  
    69  	return os.RemoveAll(dir)
    70  }