|
4 | 4 | package redis |
5 | 5 |
|
6 | 6 | import ( |
| 7 | + "crypto/tls" |
7 | 8 | "errors" |
8 | 9 | "testing" |
9 | 10 | "time" |
10 | 11 | ) |
11 | 12 |
|
12 | 13 | func TestParseURL(t *testing.T) { |
13 | 14 | cases := []struct { |
14 | | - u string |
15 | | - addr string |
16 | | - db int |
17 | | - tls bool |
18 | | - err error |
19 | | - user string |
20 | | - pass string |
| 15 | + url string |
| 16 | + o *Options // expected value |
| 17 | + err error |
21 | 18 | }{ |
22 | 19 | { |
23 | | - "redis://localhost:123/1", |
24 | | - "localhost:123", |
25 | | - 1, false, nil, |
26 | | - "", "", |
27 | | - }, |
28 | | - { |
29 | | - "redis://localhost:123", |
30 | | - "localhost:123", |
31 | | - 0, false, nil, |
32 | | - "", "", |
33 | | - }, |
34 | | - { |
35 | | - "redis://localhost/1", |
36 | | - "localhost:6379", |
37 | | - 1, false, nil, |
38 | | - "", "", |
39 | | - }, |
40 | | - { |
41 | | - "redis://12345", |
42 | | - "12345:6379", |
43 | | - 0, false, nil, |
44 | | - "", "", |
45 | | - }, |
46 | | - { |
47 | | - "rediss://localhost:123", |
48 | | - "localhost:123", |
49 | | - 0, true, nil, |
50 | | - "", "", |
51 | | - }, |
52 | | - { |
53 | | - "redis://:bar@localhost:123", |
54 | | - "localhost:123", |
55 | | - 0, false, nil, |
56 | | - "", "bar", |
57 | | - }, |
58 | | - { |
59 | | - "redis://foo@localhost:123", |
60 | | - "localhost:123", |
61 | | - 0, false, nil, |
62 | | - "foo", "", |
63 | | - }, |
64 | | - { |
65 | | - "redis://foo:bar@localhost:123", |
66 | | - "localhost:123", |
67 | | - 0, false, nil, |
68 | | - "foo", "bar", |
69 | | - }, |
70 | | - { |
71 | | - "unix:///tmp/redis.sock", |
72 | | - "/tmp/redis.sock", |
73 | | - 0, false, nil, |
74 | | - "", "", |
75 | | - }, |
76 | | - { |
77 | | - "unix://foo:bar@/tmp/redis.sock", |
78 | | - "/tmp/redis.sock", |
79 | | - 0, false, nil, |
80 | | - "foo", "bar", |
81 | | - }, |
82 | | - { |
83 | | - "unix://foo:bar@/tmp/redis.sock?db=3", |
84 | | - "/tmp/redis.sock", |
85 | | - 3, false, nil, |
86 | | - "foo", "bar", |
87 | | - }, |
88 | | - { |
89 | | - "unix://foo:bar@/tmp/redis.sock?db=test", |
90 | | - "/tmp/redis.sock", |
91 | | - 0, false, errors.New("redis: invalid database number: strconv.Atoi: parsing \"test\": invalid syntax"), |
92 | | - "", "", |
93 | | - }, |
94 | | - { |
95 | | - "redis://localhost/?abc=123", |
96 | | - "", |
97 | | - 0, false, errors.New("redis: no options supported"), |
98 | | - "", "", |
99 | | - }, |
100 | | - { |
101 | | - "http://google.com", |
102 | | - "", |
103 | | - 0, false, errors.New("redis: invalid URL scheme: http"), |
104 | | - "", "", |
105 | | - }, |
106 | | - { |
107 | | - "redis://localhost/1/2/3/4", |
108 | | - "", |
109 | | - 0, false, errors.New("redis: invalid URL path: /1/2/3/4"), |
110 | | - "", "", |
111 | | - }, |
112 | | - { |
113 | | - "12345", |
114 | | - "", |
115 | | - 0, false, errors.New("redis: invalid URL scheme: "), |
116 | | - "", "", |
117 | | - }, |
118 | | - { |
119 | | - "redis://localhost/iamadatabase", |
120 | | - "", |
121 | | - 0, false, errors.New(`redis: invalid database number: "iamadatabase"`), |
122 | | - "", "", |
| 20 | + url: "redis://localhost:123/1", |
| 21 | + o: &Options{Addr: "localhost:123", DB: 1}, |
| 22 | + }, { |
| 23 | + url: "redis://localhost:123", |
| 24 | + o: &Options{Addr: "localhost:123"}, |
| 25 | + }, { |
| 26 | + url: "redis://localhost/1", |
| 27 | + o: &Options{Addr: "localhost:6379", DB: 1}, |
| 28 | + }, { |
| 29 | + url: "redis://12345", |
| 30 | + o: &Options{Addr: "12345:6379"}, |
| 31 | + }, { |
| 32 | + url: "rediss://localhost:123", |
| 33 | + o: &Options{Addr: "localhost:123", TLSConfig: &tls.Config{ /* no deep comparison */ }}, |
| 34 | + }, { |
| 35 | + url: "redis://:bar@localhost:123", |
| 36 | + o: &Options{Addr: "localhost:123", Password: "bar"}, |
| 37 | + }, { |
| 38 | + url: "redis://foo@localhost:123", |
| 39 | + o: &Options{Addr: "localhost:123", Username: "foo"}, |
| 40 | + }, { |
| 41 | + url: "redis://foo:bar@localhost:123", |
| 42 | + o: &Options{Addr: "localhost:123", Username: "foo", Password: "bar"}, |
| 43 | + }, { |
| 44 | + url: "unix:///tmp/redis.sock", |
| 45 | + o: &Options{Addr: "/tmp/redis.sock"}, |
| 46 | + }, { |
| 47 | + url: "unix://foo:bar@/tmp/redis.sock", |
| 48 | + o: &Options{Addr: "/tmp/redis.sock", Username: "foo", Password: "bar"}, |
| 49 | + }, { |
| 50 | + url: "unix://foo:bar@/tmp/redis.sock?db=3", |
| 51 | + o: &Options{Addr: "/tmp/redis.sock", Username: "foo", Password: "bar", DB: 3}, |
| 52 | + }, { |
| 53 | + url: "unix://foo:bar@/tmp/redis.sock?db=test", |
| 54 | + err: errors.New(`redis: invalid database number: strconv.Atoi: parsing "test": invalid syntax`), |
| 55 | + }, { |
| 56 | + url: "redis://localhost/?abc=123", |
| 57 | + err: errors.New("redis: no options supported"), |
| 58 | + }, { |
| 59 | + url: "http://google.com", |
| 60 | + err: errors.New("redis: invalid URL scheme: http"), |
| 61 | + }, { |
| 62 | + url: "redis://localhost/1/2/3/4", |
| 63 | + err: errors.New("redis: invalid URL path: /1/2/3/4"), |
| 64 | + }, { |
| 65 | + url: "12345", |
| 66 | + err: errors.New("redis: invalid URL scheme: "), |
| 67 | + }, { |
| 68 | + url: "redis://localhost/iamadatabase", |
| 69 | + err: errors.New(`redis: invalid database number: "iamadatabase"`), |
123 | 70 | }, |
124 | 71 | } |
125 | 72 |
|
126 | | - for _, c := range cases { |
127 | | - t.Run(c.u, func(t *testing.T) { |
128 | | - o, err := ParseURL(c.u) |
129 | | - if c.err == nil && err != nil { |
| 73 | + for i := range cases { |
| 74 | + tc := cases[i] |
| 75 | + t.Run(tc.url, func(t *testing.T) { |
| 76 | + t.Parallel() |
| 77 | + |
| 78 | + actual, err := ParseURL(tc.url) |
| 79 | + if tc.err == nil && err != nil { |
130 | 80 | t.Fatalf("unexpected error: %q", err) |
131 | 81 | return |
132 | 82 | } |
133 | | - if c.err != nil && err != nil { |
134 | | - if c.err.Error() != err.Error() { |
135 | | - t.Fatalf("got %q, expected %q", err, c.err) |
| 83 | + if tc.err != nil && err != nil { |
| 84 | + if tc.err.Error() != err.Error() { |
| 85 | + t.Fatalf("got %q, expected %q", err, tc.err) |
136 | 86 | } |
137 | 87 | return |
138 | 88 | } |
139 | | - if o.Addr != c.addr { |
140 | | - t.Errorf("got %q, want %q", o.Addr, c.addr) |
141 | | - } |
142 | | - if o.DB != c.db { |
143 | | - t.Errorf("got %q, expected %q", o.DB, c.db) |
144 | | - } |
145 | | - if c.tls && o.TLSConfig == nil { |
146 | | - t.Errorf("got nil TLSConfig, expected a TLSConfig") |
147 | | - } |
148 | | - if o.Username != c.user { |
149 | | - t.Errorf("got %q, expected %q", o.Username, c.user) |
150 | | - } |
151 | | - if o.Password != c.pass { |
152 | | - t.Errorf("got %q, expected %q", o.Password, c.pass) |
153 | | - } |
| 89 | + comprareOptions(t, actual, tc.o) |
154 | 90 | }) |
155 | 91 | } |
156 | 92 | } |
157 | 93 |
|
| 94 | +func comprareOptions(t *testing.T, actual, expected *Options) { |
| 95 | + t.Helper() |
| 96 | + |
| 97 | + if actual.Addr != expected.Addr { |
| 98 | + t.Errorf("got %q, want %q", actual.Addr, expected.Addr) |
| 99 | + } |
| 100 | + if actual.DB != expected.DB { |
| 101 | + t.Errorf("got %q, expected %q", actual.DB, expected.DB) |
| 102 | + } |
| 103 | + if actual.TLSConfig == nil && expected.TLSConfig != nil { |
| 104 | + t.Errorf("got nil TLSConfig, expected a TLSConfig") |
| 105 | + } |
| 106 | + if actual.TLSConfig != nil && expected.TLSConfig == nil { |
| 107 | + t.Errorf("got TLSConfig, expected no TLSConfig") |
| 108 | + } |
| 109 | + if actual.Username != expected.Username { |
| 110 | + t.Errorf("got %q, expected %q", actual.Username, expected.Username) |
| 111 | + } |
| 112 | + if actual.Password != expected.Password { |
| 113 | + t.Errorf("got %q, expected %q", actual.Password, expected.Password) |
| 114 | + } |
| 115 | +} |
| 116 | + |
158 | 117 | // Test ReadTimeout option initialization, including special values -1 and 0. |
159 | 118 | // And also test behaviour of WriteTimeout option, when it is not explicitly set and use |
160 | 119 | // ReadTimeout value. |
|
0 commit comments