|
1 | 1 | package configs
|
2 | 2 |
|
3 | 3 | import (
|
4 |
| - "reflect" |
5 | 4 | "testing"
|
6 | 5 |
|
7 | 6 | "github.com/google/go-cmp/cmp"
|
@@ -62,6 +61,157 @@ func TestTransportServerExString(t *testing.T) {
|
62 | 61 | }
|
63 | 62 | }
|
64 | 63 |
|
| 64 | +func TestGenerateTransportServerConfigForTCPSnippets(t *testing.T) { |
| 65 | + transportServerEx := TransportServerEx{ |
| 66 | + TransportServer: &conf_v1alpha1.TransportServer{ |
| 67 | + ObjectMeta: meta_v1.ObjectMeta{ |
| 68 | + Name: "tcp-server", |
| 69 | + Namespace: "default", |
| 70 | + }, |
| 71 | + Spec: conf_v1alpha1.TransportServerSpec{ |
| 72 | + Listener: conf_v1alpha1.TransportServerListener{ |
| 73 | + Name: "tcp-listener", |
| 74 | + Protocol: "TCP", |
| 75 | + }, |
| 76 | + Upstreams: []conf_v1alpha1.Upstream{ |
| 77 | + { |
| 78 | + Name: "tcp-app", |
| 79 | + Service: "tcp-app-svc", |
| 80 | + Port: 5001, |
| 81 | + }, |
| 82 | + }, |
| 83 | + Action: &conf_v1alpha1.Action{ |
| 84 | + Pass: "tcp-app", |
| 85 | + }, |
| 86 | + Snippets: "deny 192.168.1.1;\nallow 192.168.1.0/24;", |
| 87 | + }, |
| 88 | + }, |
| 89 | + Endpoints: map[string][]string{ |
| 90 | + "default/tcp-app-svc:5001": { |
| 91 | + "10.0.0.20:5001", |
| 92 | + }, |
| 93 | + }, |
| 94 | + } |
| 95 | + |
| 96 | + listenerPort := 2020 |
| 97 | + |
| 98 | + expected := version2.TransportServerConfig{ |
| 99 | + Upstreams: []version2.StreamUpstream{ |
| 100 | + { |
| 101 | + Name: "ts_default_tcp-server_tcp-app", |
| 102 | + Servers: []version2.StreamUpstreamServer{ |
| 103 | + { |
| 104 | + Address: "10.0.0.20:5001", |
| 105 | + MaxFails: 1, |
| 106 | + FailTimeout: "10s", |
| 107 | + }, |
| 108 | + }, |
| 109 | + UpstreamLabels: version2.UpstreamLabels{ |
| 110 | + ResourceName: "tcp-server", |
| 111 | + ResourceType: "transportserver", |
| 112 | + ResourceNamespace: "default", |
| 113 | + Service: "tcp-app-svc", |
| 114 | + }, |
| 115 | + }, |
| 116 | + }, |
| 117 | + Server: version2.StreamServer{ |
| 118 | + Port: listenerPort, |
| 119 | + UDP: false, |
| 120 | + StatusZone: "tcp-listener", |
| 121 | + ProxyPass: "ts_default_tcp-server_tcp-app", |
| 122 | + Name: "tcp-server", |
| 123 | + Namespace: "default", |
| 124 | + ProxyConnectTimeout: "60s", |
| 125 | + ProxyNextUpstream: false, |
| 126 | + ProxyNextUpstreamTries: 0, |
| 127 | + ProxyNextUpstreamTimeout: "0s", |
| 128 | + ProxyTimeout: "10m", |
| 129 | + HealthCheck: nil, |
| 130 | + Snippets: []string{"deny 192.168.1.1;", "allow 192.168.1.0/24;"}, |
| 131 | + }, |
| 132 | + } |
| 133 | + |
| 134 | + result, _ := generateTransportServerConfig(&transportServerEx, listenerPort, true, true) |
| 135 | + if diff := cmp.Diff(expected, result); diff != "" { |
| 136 | + t.Errorf("generateTransportServerConfig() mismatch (-want +got):\n%s", diff) |
| 137 | + } |
| 138 | +} |
| 139 | + |
| 140 | +func TestGenerateTransportServerConfigForTCPSnippetsNotEnabled(t *testing.T) { |
| 141 | + transportServerEx := TransportServerEx{ |
| 142 | + TransportServer: &conf_v1alpha1.TransportServer{ |
| 143 | + ObjectMeta: meta_v1.ObjectMeta{ |
| 144 | + Name: "tcp-server", |
| 145 | + Namespace: "default", |
| 146 | + }, |
| 147 | + Spec: conf_v1alpha1.TransportServerSpec{ |
| 148 | + Listener: conf_v1alpha1.TransportServerListener{ |
| 149 | + Name: "tcp-listener", |
| 150 | + Protocol: "TCP", |
| 151 | + }, |
| 152 | + Upstreams: []conf_v1alpha1.Upstream{ |
| 153 | + { |
| 154 | + Name: "tcp-app", |
| 155 | + Service: "tcp-app-svc", |
| 156 | + Port: 5001, |
| 157 | + }, |
| 158 | + }, |
| 159 | + Action: &conf_v1alpha1.Action{ |
| 160 | + Pass: "tcp-app", |
| 161 | + }, |
| 162 | + Snippets: "deny 192.168.1.1;\nallow 192.168.1.0/24;", |
| 163 | + }, |
| 164 | + }, |
| 165 | + Endpoints: map[string][]string{ |
| 166 | + "default/tcp-app-svc:5001": { |
| 167 | + "10.0.0.20:5001", |
| 168 | + }, |
| 169 | + }, |
| 170 | + } |
| 171 | + |
| 172 | + listenerPort := 2020 |
| 173 | + |
| 174 | + expected := version2.TransportServerConfig{ |
| 175 | + Upstreams: []version2.StreamUpstream{ |
| 176 | + { |
| 177 | + Name: "ts_default_tcp-server_tcp-app", |
| 178 | + Servers: []version2.StreamUpstreamServer{ |
| 179 | + { |
| 180 | + Address: "10.0.0.20:5001", |
| 181 | + MaxFails: 1, |
| 182 | + FailTimeout: "10s", |
| 183 | + }, |
| 184 | + }, |
| 185 | + UpstreamLabels: version2.UpstreamLabels{ |
| 186 | + ResourceName: "tcp-server", |
| 187 | + ResourceType: "transportserver", |
| 188 | + ResourceNamespace: "default", |
| 189 | + Service: "tcp-app-svc", |
| 190 | + }, |
| 191 | + }, |
| 192 | + }, |
| 193 | + Server: version2.StreamServer{ |
| 194 | + Port: 2020, |
| 195 | + UDP: false, |
| 196 | + StatusZone: "tcp-listener", |
| 197 | + ProxyPass: "ts_default_tcp-server_tcp-app", |
| 198 | + Name: "tcp-server", |
| 199 | + Namespace: "default", |
| 200 | + ProxyConnectTimeout: "60s", |
| 201 | + ProxyNextUpstream: false, |
| 202 | + ProxyNextUpstreamTries: 0, |
| 203 | + ProxyNextUpstreamTimeout: "0s", |
| 204 | + ProxyTimeout: "10m", |
| 205 | + Snippets: []string{}, |
| 206 | + }, |
| 207 | + } |
| 208 | + |
| 209 | + result, _ := generateTransportServerConfig(&transportServerEx, listenerPort, true, false) |
| 210 | + if diff := cmp.Diff(expected, result); diff != "" { |
| 211 | + t.Errorf("generateTransportServerConfig() mismatch (-want +got):\n%s", diff) |
| 212 | + } |
| 213 | +} |
| 214 | + |
65 | 215 | func TestGenerateTransportServerConfigForTCP(t *testing.T) {
|
66 | 216 | transportServerEx := TransportServerEx{
|
67 | 217 | TransportServer: &conf_v1alpha1.TransportServer{
|
@@ -136,14 +286,15 @@ func TestGenerateTransportServerConfigForTCP(t *testing.T) {
|
136 | 286 | ProxyNextUpstreamTimeout: "0s",
|
137 | 287 | ProxyTimeout: "50s",
|
138 | 288 | HealthCheck: nil,
|
| 289 | + Snippets: []string{}, |
139 | 290 | },
|
140 | 291 | }
|
141 | 292 |
|
142 |
| - isPlus := true |
143 |
| - result := generateTransportServerConfig(&transportServerEx, listenerPort, isPlus) |
144 |
| - if !reflect.DeepEqual(result, expected) { |
145 |
| - t.Errorf("generateTransportServerConfig() returned \n%+v but expected \n%+v", result, expected) |
| 293 | + result, _ := generateTransportServerConfig(&transportServerEx, listenerPort, true, false) |
| 294 | + if diff := cmp.Diff(expected, result); diff != "" { |
| 295 | + t.Errorf("generateTransportServerConfig() mismatch (-want +got):\n%s", diff) |
146 | 296 | }
|
| 297 | + |
147 | 298 | }
|
148 | 299 |
|
149 | 300 | func TestGenerateTransportServerConfigForTLSPasstrhough(t *testing.T) {
|
@@ -220,13 +371,13 @@ func TestGenerateTransportServerConfigForTLSPasstrhough(t *testing.T) {
|
220 | 371 | ProxyNextUpstreamTries: 0,
|
221 | 372 | ProxyTimeout: "10m",
|
222 | 373 | HealthCheck: nil,
|
| 374 | + Snippets: []string{}, |
223 | 375 | },
|
224 | 376 | }
|
225 | 377 |
|
226 |
| - isPlus := true |
227 |
| - result := generateTransportServerConfig(&transportServerEx, listenerPort, isPlus) |
228 |
| - if !reflect.DeepEqual(result, expected) { |
229 |
| - t.Errorf("generateTransportServerConfig() returned \n%+v but expected \n%+v", result, expected) |
| 378 | + result, _ := generateTransportServerConfig(&transportServerEx, listenerPort, true, false) |
| 379 | + if diff := cmp.Diff(expected, result); diff != "" { |
| 380 | + t.Errorf("generateTransportServerConfig() mismatch (-want +got):\n%s", diff) |
230 | 381 | }
|
231 | 382 | }
|
232 | 383 |
|
@@ -309,13 +460,13 @@ func TestGenerateTransportServerConfigForUDP(t *testing.T) {
|
309 | 460 | ProxyNextUpstreamTries: 0,
|
310 | 461 | ProxyTimeout: "10m",
|
311 | 462 | HealthCheck: nil,
|
| 463 | + Snippets: []string{}, |
312 | 464 | },
|
313 | 465 | }
|
314 | 466 |
|
315 |
| - isPlus := true |
316 |
| - result := generateTransportServerConfig(&transportServerEx, listenerPort, isPlus) |
317 |
| - if !reflect.DeepEqual(result, expected) { |
318 |
| - t.Errorf("generateTransportServerConfig() returned \n%+v but expected \n%+v", result, expected) |
| 467 | + result, _ := generateTransportServerConfig(&transportServerEx, listenerPort, true, false) |
| 468 | + if diff := cmp.Diff(expected, result); diff != "" { |
| 469 | + t.Errorf("generateTransportServerConfig() mismatch (-want +got):\n%s", diff) |
319 | 470 | }
|
320 | 471 | }
|
321 | 472 |
|
|
0 commit comments