Skip to content

Commit

Permalink
Use containernetworking/cni/pkg/types/100
Browse files Browse the repository at this point in the history
Signed-off-by: Manuel Buil <[email protected]>
  • Loading branch information
manuelbuil committed Feb 23, 2024
1 parent 5dd2fb5 commit 6eba564
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 30 deletions.
13 changes: 1 addition & 12 deletions cmd/whereabouts.go
Original file line number Diff line number Diff line change
Expand Up @@ -8,10 +8,9 @@ import (

"github.com/containernetworking/cni/pkg/skel"
cnitypes "github.com/containernetworking/cni/pkg/types"
current "github.com/containernetworking/cni/pkg/types/040"
current "github.com/containernetworking/cni/pkg/types/100"
cniversion "github.com/containernetworking/cni/pkg/version"
"github.com/k8snetworkplumbingwg/whereabouts/pkg/config"
"github.com/k8snetworkplumbingwg/whereabouts/pkg/iphelpers"
"github.com/k8snetworkplumbingwg/whereabouts/pkg/logging"
"github.com/k8snetworkplumbingwg/whereabouts/pkg/storage/kubernetes"
"github.com/k8snetworkplumbingwg/whereabouts/pkg/types"
Expand Down Expand Up @@ -83,25 +82,15 @@ func cmdAdd(args *skel.CmdArgs, client *kubernetes.KubernetesIPAM, cniVersion st
return fmt.Errorf("error at storage engine: %w", err)
}

var useVersion string
for _, newip := range newips {
// Determine if v4 or v6.
if iphelpers.IsIPv4(newip.IP) {
useVersion = "4"
} else {
useVersion = "6"
}

result.IPs = append(result.IPs, &current.IPConfig{
Version: useVersion,
Address: newip,
Gateway: client.Config.Gateway})
}

// Assign all the static IP elements.
for _, v := range client.Config.Addresses {
result.IPs = append(result.IPs, &current.IPConfig{
Version: v.Version,
Address: v.Address,
Gateway: v.Gateway})
}
Expand Down
19 changes: 1 addition & 18 deletions cmd/whereabouts_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ import (

"github.com/containernetworking/cni/pkg/skel"
"github.com/containernetworking/cni/pkg/types"
current "github.com/containernetworking/cni/pkg/types/040"
current "github.com/containernetworking/cni/pkg/types/100"
"github.com/containernetworking/plugins/pkg/testutils"

metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
Expand Down Expand Up @@ -84,7 +84,6 @@ func AllocateAndReleaseAddressesTest(ipVersion string, ipamConf *whereaboutstype
// Gomega is cranky about slices with different caps
ExpectWithOffset(1, *result.IPs[0]).To(Equal(
current.IPConfig{
Version: ipVersion,
Address: mustCIDR(expectedAddresses[i]),
Gateway: ipamConf.Gateway,
}))
Expand All @@ -107,7 +106,6 @@ func AllocateAndReleaseAddressesTest(ipVersion string, ipamConf *whereaboutstype

Expect(*result.IPs[0]).To(Equal(
current.IPConfig{
Version: ipVersion,
Address: mustCIDR(expectedAddresses[i]),
Gateway: ipamConf.Gateway,
}))
Expand Down Expand Up @@ -333,7 +331,6 @@ var _ = Describe("Whereabouts operations", func() {
// Gomega is cranky about slices with different caps
Expect(*result.IPs[0]).To(Equal(
current.IPConfig{
Version: "4",
Address: mustCIDR("192.168.1.32/24"),
Gateway: net.ParseIP("192.168.10.1"),
}))
Expand Down Expand Up @@ -407,7 +404,6 @@ var _ = Describe("Whereabouts operations", func() {
// Gomega is cranky about slices with different caps
Expect(*result.IPs[0]).To(Equal(
current.IPConfig{
Version: "6",
Address: mustCIDR("2001::3/116"),
Gateway: net.ParseIP("2001::f:1"),
}))
Expand Down Expand Up @@ -478,7 +474,6 @@ var _ = Describe("Whereabouts operations", func() {
// Gomega is cranky about slices with different caps
Expect(*result.IPs[0]).To(Equal(
current.IPConfig{
Version: "6",
Address: mustCIDR("caa5::8000/112"),
Gateway: net.ParseIP("2001::f:1"),
}))
Expand Down Expand Up @@ -561,21 +556,18 @@ var _ = Describe("Whereabouts operations", func() {

Expect(*result.IPs[0]).To(Equal(
current.IPConfig{
Version: "4",
Address: mustCIDR("192.168.1.33/28"),
Gateway: net.ParseIP("192.168.1.1"),
}))

Expect(*result.IPs[1]).To(Equal(
current.IPConfig{
Version: "4",
Address: mustCIDR("10.10.0.1/24"),
Gateway: net.ParseIP("10.10.0.254"),
}))

Expect(*result.IPs[2]).To(Equal(
current.IPConfig{
Version: "6",
Address: mustCIDR("3ffe:ffff:0:01ff::1/64"),
Gateway: net.ParseIP("3ffe:ffff:0::1"),
},
Expand Down Expand Up @@ -850,7 +842,6 @@ var _ = Describe("Whereabouts operations", func() {

Expect(*result.IPs[0]).To(Equal(
current.IPConfig{
Version: "4",
Address: mustCIDR("192.168.1.5/24"),
Gateway: net.ParseIP("192.168.10.1"),
}))
Expand Down Expand Up @@ -918,7 +909,6 @@ var _ = Describe("Whereabouts operations", func() {

Expect(*result.IPs[0]).To(Equal(
current.IPConfig{
Version: "4",
Address: mustCIDR("192.168.1.5/24"),
Gateway: net.ParseIP("192.168.10.1"),
}))
Expand Down Expand Up @@ -987,7 +977,6 @@ var _ = Describe("Whereabouts operations", func() {

Expect(*result.IPs[0]).To(Equal(
current.IPConfig{
Version: "4",
Address: mustCIDR(fmt.Sprintf("192.168.1.%d/24", 5+i)),
Gateway: net.ParseIP("192.168.10.1"),
}))
Expand Down Expand Up @@ -1076,7 +1065,6 @@ var _ = Describe("Whereabouts operations", func() {
// Gomega is cranky about slices with different caps
ExpectWithOffset(1, *result.IPs[0]).To(Equal(
current.IPConfig{
Version: "4",
Address: mustCIDR("192.168.22.1/24"),
}))

Expand Down Expand Up @@ -1124,7 +1112,6 @@ var _ = Describe("Whereabouts operations", func() {
// Gomega is cranky about slices with different caps
ExpectWithOffset(1, *result.IPs[0]).To(Equal(
current.IPConfig{
Version: "4",
Address: mustCIDR("192.168.22.2/28"),
}))

Expand Down Expand Up @@ -1198,7 +1185,6 @@ var _ = Describe("Whereabouts operations", func() {
// Gomega is cranky about slices with different caps
ExpectWithOffset(1, *result.IPs[0]).To(Equal(
current.IPConfig{
Version: "6",
Address: mustCIDR("2001::2:3:1/124"),
}))

Expand Down Expand Up @@ -1246,7 +1232,6 @@ var _ = Describe("Whereabouts operations", func() {
// Gomega is cranky about slices with different caps
ExpectWithOffset(1, *result.IPs[0]).To(Equal(
current.IPConfig{
Version: "6",
Address: mustCIDR("2001::2:3:2/126"),
}))

Expand Down Expand Up @@ -1321,7 +1306,6 @@ var _ = Describe("Whereabouts operations", func() {
// Gomega is cranky about slices with different caps
ExpectWithOffset(1, *result.IPs[0]).To(Equal(
current.IPConfig{
Version: "4",
Address: mustCIDR("192.168.33.1/24"),
}))

Expand Down Expand Up @@ -1369,7 +1353,6 @@ var _ = Describe("Whereabouts operations", func() {
// Gomega is cranky about slices with different caps
ExpectWithOffset(1, *result.IPs[0]).To(Equal(
current.IPConfig{
Version: "4",
Address: mustCIDR("192.168.33.1/28"),
}))

Expand Down

0 comments on commit 6eba564

Please sign in to comment.