host-device: cleanup + completed tests;

This commit is contained in:
Tino Rusch 2017-06-29 11:44:59 +02:00
parent f2faf549b4
commit ca3f28fa9e
No known key found for this signature in database
GPG Key ID: E931E57509A81CEA

View File

@ -15,62 +15,102 @@
package main package main
import ( import (
"fmt"
"github.com/containernetworking/cni/pkg/ns" "github.com/containernetworking/cni/pkg/ns"
"github.com/containernetworking/cni/pkg/skel" "github.com/containernetworking/cni/pkg/skel"
"github.com/containernetworking/cni/pkg/testutils" "github.com/containernetworking/cni/pkg/testutils"
. "github.com/onsi/ginkgo" . "github.com/onsi/ginkgo"
. "github.com/onsi/gomega" . "github.com/onsi/gomega"
"github.com/vishvananda/netlink"
) )
var ifname = "dummy0"
var _ = Describe("base functionality", func() { var _ = Describe("base functionality", func() {
var targetNs ns.NetNS var originalNS ns.NetNS
BeforeEach(func() { BeforeEach(func() {
var err error var err error
targetNs, err = ns.NewNS() originalNS, err = ns.NewNS()
Expect(err).NotTo(HaveOccurred()) Expect(err).NotTo(HaveOccurred())
}) })
AfterEach(func() { AfterEach(func() {
targetNs.Close() originalNS.Close()
}) })
It("Works with a valid config", func() { It("Works with a valid config", func() {
ifname := "eth0"
// prepare ifname in original namespace
err := originalNS.Do(func(ns.NetNS) error {
defer GinkgoRecover()
err := netlink.LinkAdd(&netlink.Dummy{
LinkAttrs: netlink.LinkAttrs{
Name: ifname,
},
})
Expect(err).NotTo(HaveOccurred())
link, err := netlink.LinkByName(ifname)
Expect(err).NotTo(HaveOccurred())
err = netlink.LinkSetUp(link)
Expect(err).NotTo(HaveOccurred())
return nil
})
Expect(err).NotTo(HaveOccurred())
// call CmdAdd
targetNS, err := ns.NewNS()
Expect(err).NotTo(HaveOccurred())
conf := `{ conf := `{
"name": "cni-plugin-host-device-test", "cniVersion": "0.3.0",
"type": "host-device", "name": "cni-plugin-host-device-test",
"device": "eth0" "type": "host-device",
}` "device": ifname
conf = fmt.Sprintf(conf, ifname, targetNs.Path()) }`
args := &skel.CmdArgs{ args := &skel.CmdArgs{
ContainerID: "dummy", ContainerID: "dummy",
Netns: targetNs.Path(), Netns: targetNS.Path(),
IfName: ifname, IfName: ifname,
StdinData: []byte(conf), StdinData: []byte(conf),
} }
_, _, err := testutils.CmdAddWithResult(targetNs.Path(), "eth0", []byte(conf), func() error { return cmdAdd(args) }) _, _, err = testutils.CmdAddWithResult(targetNS.Path(), ifname, []byte(conf), func() error { return cmdAdd(args) })
Expect(err).NotTo(HaveOccurred()) Expect(err).NotTo(HaveOccurred())
// assert that dummy0 is now in the target namespace
err = targetNS.Do(func(ns.NetNS) error {
defer GinkgoRecover()
link, err := netlink.LinkByName(ifname)
Expect(err).NotTo(HaveOccurred())
Expect(link.Attrs().Name).To(Equal(ifname))
return nil
})
Expect(err).NotTo(HaveOccurred())
// assert that dummy0 is now NOT in the original namespace anymore
err = originalNS.Do(func(ns.NetNS) error {
defer GinkgoRecover()
_, err := netlink.LinkByName(ifname)
Expect(err).To(HaveOccurred())
return nil
})
Expect(err).NotTo(HaveOccurred())
}) })
It("fails an invalid config", func() { It("fails an invalid config", func() {
conf := `{ conf := `{
"cniVersion": "0.3.0", "cniVersion": "0.3.0",
"name": "cni-plugin-sample-test", "name": "cni-plugin-host-device-test",
"type": "host-device" "type": "host-device"
}` }`
args := &skel.CmdArgs{ args := &skel.CmdArgs{
ContainerID: "dummy", ContainerID: "dummy",
Netns: targetNs.Path(), Netns: originalNS.Path(),
IfName: "eth0", IfName: ifname,
StdinData: []byte(conf), StdinData: []byte(conf),
} }
_, _, err := testutils.CmdAddWithResult(targetNs.Path(), "eth0", []byte(conf), func() error { return cmdAdd(args) }) _, _, err := testutils.CmdAddWithResult(originalNS.Path(), ifname, []byte(conf), func() error { return cmdAdd(args) })
Expect(err).To(MatchError("anotherAwesomeArg must be specified")) Expect(err).To(MatchError(`specify either "device", "hwaddr" or "kernelpath"`))
}) })