gitlab.com/jfprevost/gitlab-runner-notlscheck@v11.11.4+incompatible/helpers/trace/buffer_test.go (about)

     1  package trace
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/stretchr/testify/assert"
     7  	"github.com/stretchr/testify/require"
     8  )
     9  
    10  func TestVariablesMasking(t *testing.T) {
    11  	traceMessage := "This is the secret message containing secret duplicateValues"
    12  	maskedValues := []string{
    13  		"is",
    14  		"duplicateValue",
    15  		"duplicateValue",
    16  		"secret",
    17  		"containing",
    18  	}
    19  
    20  	buffer := New()
    21  	buffer.SetMasked(maskedValues)
    22  
    23  	_, err := buffer.Write([]byte(traceMessage))
    24  	require.NoError(t, err)
    25  
    26  	err = buffer.Close()
    27  	require.NoError(t, err)
    28  
    29  	assert.Equal(t, "Th[MASKED] [MASKED] the [MASKED] message [MASKED] [MASKED] [MASKED]s", buffer.String())
    30  }
    31  
    32  func TestTraceLimit(t *testing.T) {
    33  	traceMessage := "This is the long message"
    34  
    35  	buffer := New()
    36  	buffer.SetLimit(10)
    37  
    38  	_, err := buffer.Write([]byte(traceMessage))
    39  	require.NoError(t, err)
    40  
    41  	err = buffer.Close()
    42  	require.NoError(t, err)
    43  
    44  	assert.Contains(t, buffer.String(), "Job's log exceeded limit of")
    45  }