github.com/devtron-labs/ci-runner@v0.0.0-20240518055909-b2672f3349d7/helper/DockerHelper_test.go (about)

     1  package helper
     2  
     3  import (
     4  	"fmt"
     5  	"os/exec"
     6  	"testing"
     7  )
     8  
     9  func TestCreateBuildXK8sDriver(t *testing.T) {
    10  	buildxOpts := make([]map[string]string, 0)
    11  	buildxOpts = append(buildxOpts, map[string]string{"node": "builder-amd64", "driverOptions": "namespace=devtron-ci,nodeselector=kubernetes.io/arch:amd64"})
    12  	buildxOpts = append(buildxOpts, map[string]string{"node": "builder-amd64-test", "driverOptions": "namespace=devtron-ci,nodeselector=kubernetes.io/arch:amd64"})
    13  	err := createBuildxBuilderWithK8sDriver(buildxOpts, 1, 1)
    14  	t.Cleanup(func() {
    15  		buildxDelete := fmt.Sprintf("docker buildx rm %s", BUILDX_K8S_DRIVER_NAME)
    16  		builderRemoveCmd := exec.Command("/bin/sh", "-c", buildxDelete)
    17  		builderRemoveCmd.Run()
    18  	})
    19  	if err != nil {
    20  		fmt.Println(err.Error())
    21  		t.Fail()
    22  	}
    23  }
    24  
    25  func TestCleanBuildxK8sDriver(t *testing.T) {
    26  	buildxOpts := make([]map[string]string, 0)
    27  	buildxOpts = append(buildxOpts, map[string]string{"node": "", "driverOptions": "namespace=devtron-ci,nodeselector=kubernetes.io/arch:amd64"})
    28  	buildxOpts = append(buildxOpts, map[string]string{"node": "builder-amd64-test", "driverOptions": "namespace=devtron-ci,nodeselector=kubernetes.io/arch:amd64"})
    29  	err := createBuildxBuilderWithK8sDriver(buildxOpts, 1, 1)
    30  	if err != nil {
    31  		fmt.Println(err.Error())
    32  		t.Fail()
    33  	}
    34  
    35  	err = CleanBuildxK8sDriver(buildxOpts)
    36  	if err != nil {
    37  		fmt.Println(err.Error())
    38  		t.Fail()
    39  	}
    40  
    41  }