github.com/khulnasoft-lab/defsec@v1.0.5-0.20230827010352-5e9f46893d95/pkg/scanners/azure/functions/data_uri_test.go (about)

     1  package functions
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/stretchr/testify/require"
     7  )
     8  
     9  func Test_data_uri_from_string(t *testing.T) {
    10  	tests := []struct {
    11  		name     string
    12  		args     []interface{}
    13  		expected string
    14  	}{
    15  		{
    16  			name: "data uri from string",
    17  			args: []interface{}{
    18  				"Hello",
    19  			},
    20  			expected: "data:text/plain;charset=utf8;base64,SGVsbG8=",
    21  		},
    22  	}
    23  
    24  	for _, tt := range tests {
    25  		t.Run(tt.name, func(t *testing.T) {
    26  			dataUri := DataUri(tt.args...)
    27  			require.Equal(t, tt.expected, dataUri)
    28  		})
    29  	}
    30  }
    31  
    32  func Test_string_from_data_uri(t *testing.T) {
    33  	tests := []struct {
    34  		name     string
    35  		args     []interface{}
    36  		expected string
    37  	}{
    38  		{
    39  			name: "data uri to string",
    40  			args: []interface{}{
    41  				"data:;base64,SGVsbG8sIFdvcmxkIQ==",
    42  			},
    43  			expected: "Hello, World!",
    44  		},
    45  	}
    46  
    47  	for _, tt := range tests {
    48  		t.Run(tt.name, func(t *testing.T) {
    49  			dataUri := DataUriToString(tt.args...)
    50  			require.Equal(t, tt.expected, dataUri)
    51  		})
    52  	}
    53  }