github.com/criteo/command-launcher@v0.0.0-20230407142452-fb616f546e98/internal/gvault/file-vault_test.go (about)

     1  package vault
     2  
     3  import (
     4  	"fmt"
     5  	"testing"
     6  )
     7  
     8  func TestVault_Init(t *testing.T) {
     9  	t.Setenv("CDT_VAULT_SECRET", "very_secret")
    10  
    11  	_, err := CreateVault("unit-test")
    12  	if err != nil {
    13  		t.Error(err)
    14  	}
    15  }
    16  
    17  func TestVault_WriteRead(t *testing.T) {
    18  	t.Setenv("CDT_VAULT_SECRET", "very_secret")
    19  
    20  	fv, err := CreateVault("unit-test")
    21  	if err != nil {
    22  		t.Error(err)
    23  	}
    24  
    25  	err = fv.Write("mykey", "myvalue")
    26  	if err != nil {
    27  		t.Error(err)
    28  	}
    29  
    30  	val, err := fv.Read("mykey")
    31  	if err != nil {
    32  		t.Error(err)
    33  	}
    34  
    35  	if val != "myvalue" {
    36  		t.Errorf("wrong value")
    37  	}
    38  }
    39  
    40  func TestVault_MultiWriteRead(t *testing.T) {
    41  	t.Setenv("CDT_VAULT_SECRET", "very_secret")
    42  
    43  	fv, err := CreateVault("unit-test")
    44  	if err != nil {
    45  		t.Error(err)
    46  	}
    47  
    48  	for i := 0; i < 1000; i++ {
    49  		err = fv.Write(fmt.Sprintf("mykey-%d", i), fmt.Sprintf("myvalue-%d", i))
    50  		if err != nil {
    51  			t.Error(err)
    52  		}
    53  	}
    54  
    55  	for i := 0; i < 1000; i++ {
    56  		val, err := fv.Read(fmt.Sprintf("mykey-%d", i))
    57  		if err != nil {
    58  			t.Error(err)
    59  		}
    60  
    61  		if val != fmt.Sprintf("myvalue-%d", i) {
    62  			t.Errorf("wrong value")
    63  		}
    64  	}
    65  }