move subfolders in pkg/

pkg/auth/package_test.go

@@ -0,0 +1,76 @@
+package auth
+
+import (
+	"net/url"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+
+	"github.com/aler9/gortsplib/pkg/base"
+	"github.com/aler9/gortsplib/pkg/headers"
+)
+
+var casesAuth = []struct {
+	name    string
+	methods []headers.AuthMethod
+}{
+	{
+		"basic",
+		[]headers.AuthMethod{headers.AuthBasic},
+	},
+	{
+		"digest",
+		[]headers.AuthMethod{headers.AuthDigest},
+	},
+	{
+		"both",
+		[]headers.AuthMethod{headers.AuthBasic, headers.AuthDigest},
+	},
+}
+
+func TestAuthMethods(t *testing.T) {
+	for _, c := range casesAuth {
+		t.Run(c.name, func(t *testing.T) {
+			authServer := NewServer("testuser", "testpass", c.methods)
+			wwwAuthenticate := authServer.GenerateHeader()
+
+			ac, err := NewClient(wwwAuthenticate, url.UserPassword("testuser", "testpass"))
+			require.NoError(t, err)
+			authorization := ac.GenerateHeader(base.ANNOUNCE,
+				base.MustParseURL("rtsp://myhost/mypath"))
+
+			err = authServer.ValidateHeader(authorization, base.ANNOUNCE,
+				base.MustParseURL("rtsp://myhost/mypath"))
+			require.NoError(t, err)
+		})
+	}
+}
+
+func TestAuthVLC(t *testing.T) {
+	for _, ca := range []struct {
+		clientURL string
+		serverURL string
+	}{
+		{
+			"rtsp://myhost/mypath/",
+			"rtsp://myhost/mypath/trackId=0",
+		},
+		{
+			"rtsp://myhost/mypath/test?testing/",
+			"rtsp://myhost/mypath/test?testing/trackId=0",
+		},
+	} {
+		authServer := NewServer("testuser", "testpass",
+			[]headers.AuthMethod{headers.AuthBasic, headers.AuthDigest})
+		wwwAuthenticate := authServer.GenerateHeader()
+
+		ac, err := NewClient(wwwAuthenticate, url.UserPassword("testuser", "testpass"))
+		require.NoError(t, err)
+		authorization := ac.GenerateHeader(base.ANNOUNCE,
+			base.MustParseURL(ca.clientURL))
+
+		err = authServer.ValidateHeader(authorization, base.ANNOUNCE,
+			base.MustParseURL(ca.serverURL))
+		require.NoError(t, err)
+	}
+}