@@ -81,9 +81,10 @@ var _ = Describe("public ip resolvers", func() {
81
81
It ("should return the IP" , func () {
82
82
backendConfig [publicIPConfig ] = lbPublicIP
83
83
client := fake .NewClientset (serviceWithIngress (v1.LoadBalancerIngress {Hostname : "" , IP : testIP }))
84
- ip , err := getPublicIP (submSpec , client , backendConfig , false )
84
+ ip , resolver , err := getPublicIP (submSpec , client , backendConfig , false )
85
85
Expect (err ).ToNot (HaveOccurred ())
86
86
Expect (ip ).To (Equal (testIP ))
87
+ Expect (resolver ).To (Equal (lbPublicIP ))
87
88
})
88
89
})
89
90
@@ -94,9 +95,10 @@ var _ = Describe("public ip resolvers", func() {
94
95
Hostname : dnsHost ,
95
96
IP : "" ,
96
97
}))
97
- ip , err := getPublicIP (submSpec , client , backendConfig , false )
98
+ ip , resolver , err := getPublicIP (submSpec , client , backendConfig , false )
98
99
Expect (err ).ToNot (HaveOccurred ())
99
100
Expect (ip ).To (Equal (testIPDNS ))
101
+ Expect (resolver ).To (Equal (lbPublicIP ))
100
102
})
101
103
})
102
104
@@ -107,7 +109,7 @@ var _ = Describe("public ip resolvers", func() {
107
109
loadBalancerRetryConfig .Steps = 1
108
110
backendConfig [publicIPConfig ] = lbPublicIP
109
111
client := fake .NewClientset (serviceWithIngress ())
110
- _ , err := getPublicIP (submSpec , client , backendConfig , false )
112
+ _ , _ , err := getPublicIP (submSpec , client , backendConfig , false )
111
113
Expect (err ).To (HaveOccurred ())
112
114
})
113
115
})
@@ -116,59 +118,65 @@ var _ = Describe("public ip resolvers", func() {
116
118
It ("should return the IP" , func () {
117
119
backendConfig [publicIPConfig ] = ipv4PublicIP
118
120
client := fake .NewClientset ()
119
- ip , err := getPublicIP (submSpec , client , backendConfig , false )
121
+ ip , resolver , err := getPublicIP (submSpec , client , backendConfig , false )
120
122
Expect (err ).ToNot (HaveOccurred ())
121
123
Expect (ip ).To (Equal (testIP ))
124
+ Expect (resolver ).To (Equal (ipv4PublicIP ))
122
125
})
123
126
})
124
127
125
128
When ("an IPv4 entry specified in air-gapped deployment" , func () {
126
129
It ("should return the IP and not an empty value" , func () {
127
130
backendConfig [publicIPConfig ] = ipv4PublicIP
128
131
client := fake .NewClientset ()
129
- ip , err := getPublicIP (submSpec , client , backendConfig , true )
132
+ ip , resolver , err := getPublicIP (submSpec , client , backendConfig , true )
130
133
Expect (err ).ToNot (HaveOccurred ())
131
134
Expect (ip ).To (Equal (testIP ))
135
+ Expect (resolver ).To (Equal (ipv4PublicIP ))
132
136
})
133
137
})
134
138
135
139
When ("a DNS entry specified" , func () {
136
140
It ("should return the IP" , func () {
137
141
backendConfig [publicIPConfig ] = "dns:" + dnsHost
138
142
client := fake .NewClientset ()
139
- ip , err := getPublicIP (submSpec , client , backendConfig , false )
143
+ ip , resolver , err := getPublicIP (submSpec , client , backendConfig , false )
140
144
Expect (err ).ToNot (HaveOccurred ())
141
145
Expect (ip ).To (Equal (testIPDNS ))
146
+ Expect (resolver ).To (Equal (backendConfig [publicIPConfig ]))
142
147
})
143
148
})
144
149
145
150
When ("an API entry specified" , func () {
146
151
It ("should return some IP" , func () {
147
152
backendConfig [publicIPConfig ] = "api:4.icanhazip.com/"
148
153
client := fake .NewClientset ()
149
- ip , err := getPublicIP (submSpec , client , backendConfig , false )
154
+ ip , resolver , err := getPublicIP (submSpec , client , backendConfig , false )
150
155
Expect (err ).ToNot (HaveOccurred ())
151
156
Expect (net .ParseIP (ip )).NotTo (BeNil ())
157
+ Expect (resolver ).To (Equal (backendConfig [publicIPConfig ]))
152
158
})
153
159
})
154
160
155
161
When ("multiple entries are specified" , func () {
156
162
It ("should return the first working one" , func () {
157
163
backendConfig [publicIPConfig ] = ipv4PublicIP + ",dns:" + dnsHost
158
164
client := fake .NewClientset ()
159
- ip , err := getPublicIP (submSpec , client , backendConfig , false )
165
+ ip , resolver , err := getPublicIP (submSpec , client , backendConfig , false )
160
166
Expect (err ).ToNot (HaveOccurred ())
161
167
Expect (ip ).To (Equal (testIP ))
168
+ Expect (resolver ).To (Equal (ipv4PublicIP ))
162
169
})
163
170
})
164
171
165
172
When ("multiple entries are specified and the first one doesn't succeed" , func () {
166
173
It ("should return the first working one" , func () {
167
- backendConfig [publicIPConfig ] = "dns:thisdomaindoesntexistforsure.badbadbad,ipv4: " + testIP
174
+ backendConfig [publicIPConfig ] = "dns:thisdomaindoesntexistforsure.badbadbad," + ipv4PublicIP
168
175
client := fake .NewClientset ()
169
- ip , err := getPublicIP (submSpec , client , backendConfig , false )
176
+ ip , resolver , err := getPublicIP (submSpec , client , backendConfig , false )
170
177
Expect (err ).ToNot (HaveOccurred ())
171
178
Expect (ip ).To (Equal (testIP ))
179
+ Expect (resolver ).To (Equal (ipv4PublicIP ))
172
180
})
173
181
})
174
182
})
0 commit comments