github.com/aquasecurity/trivy-iac@v0.8.1-0.20240127024015-3d8e412cf0ab/pkg/scanners/cloudformation/parser/intrinsics_test.go (about)

     1  package parser
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/stretchr/testify/assert"
     7  	"gopkg.in/yaml.v3"
     8  )
     9  
    10  func Test_is_intrinsic_returns_expected(t *testing.T) {
    11  
    12  	testCases := []struct {
    13  		nodeTag        string
    14  		expectedResult bool
    15  	}{
    16  		{
    17  			nodeTag:        "!Ref",
    18  			expectedResult: true,
    19  		},
    20  		{
    21  			nodeTag:        "!Join",
    22  			expectedResult: true,
    23  		},
    24  		{
    25  			nodeTag:        "!Sub",
    26  			expectedResult: true,
    27  		},
    28  		{
    29  			nodeTag:        "!Equals",
    30  			expectedResult: true,
    31  		},
    32  		{
    33  			nodeTag:        "!Equal",
    34  			expectedResult: false,
    35  		},
    36  	}
    37  
    38  	for _, tt := range testCases {
    39  		n := &yaml.Node{
    40  			Tag: tt.nodeTag,
    41  		}
    42  		assert.Equal(t, tt.expectedResult, IsIntrinsicFunc(n))
    43  	}
    44  
    45  }