1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798 |
- package csstring_test
- import (
- "fmt"
- "testing"
- "github.com/stretchr/testify/assert"
- "github.com/crowdsecurity/crowdsec/pkg/csstring"
- )
- func TestStrictExpand(t *testing.T) {
- t.Parallel()
- testenv := func(key string) (string, bool) {
- switch key {
- case "USER":
- return "testuser", true
- case "HOME":
- return "/home/testuser", true
- case "empty":
- return "", true
- default:
- return "", false
- }
- }
- home, _ := testenv("HOME")
- user, _ := testenv("USER")
- tests := []struct {
- input string
- expected string
- }{
- {
- input: "$HOME",
- expected: home,
- },
- {
- input: "${USER}",
- expected: user,
- },
- {
- input: "Hello, $USER!",
- expected: fmt.Sprintf("Hello, %s!", user),
- },
- {
- input: "My home directory is ${HOME}",
- expected: fmt.Sprintf("My home directory is %s", home),
- },
- {
- input: "This is a $SINGLE_VAR string with ${HOME}",
- expected: fmt.Sprintf("This is a $SINGLE_VAR string with %s", home),
- },
- {
- input: "This is a $SINGLE_VAR string with $HOME",
- expected: fmt.Sprintf("This is a $SINGLE_VAR string with %s", home),
- },
- {
- input: "This variable does not exist: $NON_EXISTENT_VAR",
- expected: "This variable does not exist: $NON_EXISTENT_VAR",
- },
- {
- input: "This is a $MULTI_VAR string with ${HOME} and ${USER}",
- expected: fmt.Sprintf("This is a $MULTI_VAR string with %s and %s", home, user),
- },
- {
- input: "This is a ${MULTI_VAR} string with $HOME and $USER",
- expected: fmt.Sprintf("This is a ${MULTI_VAR} string with %s and %s", home, user),
- },
- {
- input: "This is a plain string with no variables",
- expected: "This is a plain string with no variables",
- },
- {
- input: "$empty",
- expected: "",
- },
- {
- input: "",
- expected: "",
- },
- {
- input: "$USER:$empty:$HOME",
- expected: fmt.Sprintf("%s::%s", user, home),
- },
- }
- for _, tc := range tests {
- tc := tc
- t.Run(tc.input, func(t *testing.T) {
- t.Parallel()
- output := csstring.StrictExpand(tc.input, testenv)
- assert.Equal(t, tc.expected, output)
- })
- }
- }
|