aboutsummaryrefslogtreecommitdiff
path: root/models/mirror_test.go
blob: b4af58ba2f6f6bbee6ff1fc4e3ab826b593b5735 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
// Copyright 2017 The Gogs Authors. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.

package models

import (
	"testing"

	. "github.com/smartystreets/goconvey/convey"
)

func Test_findPasswordInMirrorAddress(t *testing.T) {
	Convey("Find password portion in mirror address", t, func() {
		testCases := []struct {
			addr       string
			start, end int
			found      bool
			password   string
		}{
			{"http://localhost:3000/user/repo.git", -1, -1, false, ""},
			{"http://user@localhost:3000/user/repo.git", -1, -1, false, ""},
			{"http://user:@localhost:3000/user/repo.git", -1, -1, false, ""},
			{"http://user:password@localhost:3000/user/repo.git", 12, 20, true, "password"},
			{"http://username:my%3Asecure%3Bpassword@localhost:3000/user/repo.git", 16, 38, true, "my%3Asecure%3Bpassword"},
			{"http://username:my%40secure%23password@localhost:3000/user/repo.git", 16, 38, true, "my%40secure%23password"},
			{"http://username:@@localhost:3000/user/repo.git", 16, 17, true, "@"},
		}

		for _, tc := range testCases {
			start, end, found := findPasswordInMirrorAddress(tc.addr)
			So(start, ShouldEqual, tc.start)
			So(end, ShouldEqual, tc.end)
			So(found, ShouldEqual, tc.found)
			if found {
				So(tc.addr[start:end], ShouldEqual, tc.password)
			}
		}
	})
}

func Test_unescapeMirrorCredentials(t *testing.T) {
	Convey("Escape credentials in mirror address", t, func() {
		testCases := []string{
			"http://localhost:3000/user/repo.git", "http://localhost:3000/user/repo.git",
			"http://user@localhost:3000/user/repo.git", "http://user@localhost:3000/user/repo.git",
			"http://user:@localhost:3000/user/repo.git", "http://user:@localhost:3000/user/repo.git",
			"http://user:password@localhost:3000/user/repo.git", "http://user:password@localhost:3000/user/repo.git",
			"http://user:my%3Asecure%3Bpassword@localhost:3000/user/repo.git", "http://user:my:secure;password@localhost:3000/user/repo.git",
			"http://user:my%40secure%23password@localhost:3000/user/repo.git", "http://user:my@secure#password@localhost:3000/user/repo.git",
		}

		for i := 0; i < len(testCases); i += 2 {
			So(unescapeMirrorCredentials(testCases[i]), ShouldEqual, testCases[i+1])
		}
	})
}

func Test_escapeMirrorCredentials(t *testing.T) {
	Convey("Escape credentials in mirror address", t, func() {
		testCases := []string{
			"http://localhost:3000/user/repo.git", "http://localhost:3000/user/repo.git",
			"http://user@localhost:3000/user/repo.git", "http://user@localhost:3000/user/repo.git",
			"http://user:@localhost:3000/user/repo.git", "http://user:@localhost:3000/user/repo.git",
			"http://user:password@localhost:3000/user/repo.git", "http://user:password@localhost:3000/user/repo.git",
			"http://user:my:secure;password@localhost:3000/user/repo.git", "http://user:my%3Asecure%3Bpassword@localhost:3000/user/repo.git",
			"http://user:my@secure#password@localhost:3000/user/repo.git", "http://user:my%40secure%23password@localhost:3000/user/repo.git",
		}

		for i := 0; i < len(testCases); i += 2 {
			So(escapeMirrorCredentials(testCases[i]), ShouldEqual, testCases[i+1])
		}
	})
}