@@ -83,10 +83,11 @@ func (suite *DiscoveryServiceSuite) TestReconcile() {
8383 MachineType : machine .TypeControlPlane ,
8484 Addresses : []netip.Addr {netip .MustParseAddr ("192.168.3.4" )},
8585 KubeSpan : cluster.KubeSpanAffiliateSpec {
86- PublicKey : "PLPNBddmTgHJhtw0vxltq1ZBdPP9RNOEUd5JjJZzBRY=" ,
87- Address : netip .MustParseAddr ("fd50:8d60:4238:6302:f857:23ff:fe21:d1e0" ),
88- AdditionalAddresses : []netip.Prefix {netip .MustParsePrefix ("10.244.3.1/24" )},
89- Endpoints : []netip.AddrPort {netip .MustParseAddrPort ("10.0.0.2:51820" ), netip .MustParseAddrPort ("192.168.3.4:51820" )},
86+ PublicKey : "PLPNBddmTgHJhtw0vxltq1ZBdPP9RNOEUd5JjJZzBRY=" ,
87+ Address : netip .MustParseAddr ("fd50:8d60:4238:6302:f857:23ff:fe21:d1e0" ),
88+ AdditionalAddresses : []netip.Prefix {netip .MustParsePrefix ("10.244.3.1/24" )},
89+ Endpoints : []netip.AddrPort {netip .MustParseAddrPort ("10.0.0.2:51820" ), netip .MustParseAddrPort ("192.168.3.4:51820" )},
90+ ExcludeAdvertisedNetworks : []netip.Prefix {netip .MustParsePrefix ("0.0.0.0/0" )},
9091 },
9192 ControlPlane : & cluster.ControlPlane {APIServerPort : 6443 },
9293 }
@@ -141,6 +142,12 @@ func (suite *DiscoveryServiceSuite) TestReconcile() {
141142 Bits : 24 ,
142143 },
143144 },
145+ ExcludeAdvertisedAddresses : []* pb.IPPrefix {
146+ {
147+ Ip : []byte ("\x00 \x00 \x00 \x00 " ),
148+ Bits : 0 ,
149+ },
150+ },
144151 },
145152 ControlPlane : & pb.ControlPlane {ApiServerPort : 6443 },
146153 }, affiliates [0 ].Affiliate ))
@@ -179,6 +186,12 @@ func (suite *DiscoveryServiceSuite) TestReconcile() {
179186 Bits : 24 ,
180187 },
181188 },
189+ ExcludeAdvertisedAddresses : []* pb.IPPrefix {
190+ {
191+ Ip : []byte ("\x01 \x01 \x01 \x01 " ),
192+ Bits : 32 ,
193+ },
194+ },
182195 },
183196 },
184197 Endpoints : []* pb.Endpoint {
@@ -204,6 +217,7 @@ func (suite *DiscoveryServiceSuite) TestReconcile() {
204217 suite .Assert ().Equal (netip .MustParseAddr ("fd50:8d60:4238:6302:f857:23ff:fe21:d1e1" ), spec .KubeSpan .Address )
205218 suite .Assert ().Equal ("1CXkdhWBm58c36kTpchR8iGlXHG1ruHa5W8gsFqD8Qs=" , spec .KubeSpan .PublicKey )
206219 suite .Assert ().Equal ([]netip.Prefix {netip .MustParsePrefix ("10.244.4.1/24" )}, spec .KubeSpan .AdditionalAddresses )
220+ suite .Assert ().Equal ([]netip.Prefix {netip .MustParsePrefix ("1.1.1.1/32" )}, spec .KubeSpan .ExcludeAdvertisedNetworks )
207221 suite .Assert ().Equal ([]netip.AddrPort {netip .MustParseAddrPort ("192.168.3.5:51820" )}, spec .KubeSpan .Endpoints )
208222 suite .Assert ().Zero (spec .ControlPlane )
209223 },
0 commit comments