drive_option_test.go

package onedrive

import (
	"fmt"
	"testing"

	"gotest.tools/v3/assert"
)

func TestDriveOption_SearchPath(t *testing.T) {
	cases := []struct {
		name string
		in   DriveOption
		out  string
	}{
		{
			name: "me scope returns onedrive search path",
			in:   DriveOption{Scope: MeScope{}, Query: "test"},
			out:  "/me/drive/search(q='test')",
		},
		{
			name: "site scope returns sharepoint site search path",
			in:   DriveOption{Scope: SharepointSiteScope{SiteID: "sp_123"}, Query: "test"},
			out:  "/sites/sp_123/drive/search(q='test')",
		},
	}

	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			assert.Equal(t, tc.in.SearchPath(), tc.out)
		})
	}

	t.Run("sanitizes dangerous characters from query", func(t *testing.T) {
		cases := []struct {
			in  string
			out string
		}{
			{"test*", "/search(q='test')"},
			{"test<tom", "/search(q='testtom')"},
			{">", "/search(q='')"},
			{"test's & skill's", "/search(q='tests%20&%20skills')"},
			{"some_path\\testfile", "/search(q='some_pathtestfile')"},
			{"some_path/testfile", "/search(q='some_pathtestfile')"},
			{"test?", "/search(q='test')"},
		}

		for _, tc := range cases {
			t.Run(fmt.Sprintf("scoped to me with %s", tc.in), func(t *testing.T) {
				ds := DriveOption{Query: tc.in, Scope: MeScope{}}
				assert.Equal(t, ds.SearchPath(), "/me/drive"+tc.out)
			})

			t.Run(fmt.Sprintf("scoped to site with %s", tc.in), func(t *testing.T) {
				ds := DriveOption{Query: tc.in, Scope: SharepointSiteScope{SiteID: "sp_234"}}
				assert.Equal(t, ds.SearchPath(), "/sites/sp_234/drive"+tc.out)
			})
		}
	})
}