bridge: add random datadir to all testcases

This commit is contained in:
Dan Williams 2018-07-25 12:19:17 -05:00 committed by Mauricio Vasquez B
parent 696b1f9ab1
commit 4b296ba330

View File

@ -16,7 +16,9 @@ package main
import ( import (
"fmt" "fmt"
"io/ioutil"
"net" "net"
"os"
"strings" "strings"
"github.com/containernetworking/cni/pkg/skel" "github.com/containernetworking/cni/pkg/skel"
@ -83,6 +85,9 @@ const (
"ipam": { "ipam": {
"type": "host-local"` "type": "host-local"`
ipamDataDirStr = `,
"dataDir": "%s"`
// Single subnet configuration (legacy) // Single subnet configuration (legacy)
subnetConfStr = `, subnetConfStr = `,
"subnet": "%s"` "subnet": "%s"`
@ -110,10 +115,13 @@ const (
// netConfJSON() generates a JSON network configuration string // netConfJSON() generates a JSON network configuration string
// for a test case. // for a test case.
func (tc testCase) netConfJSON() string { func (tc testCase) netConfJSON(dataDir string) string {
conf := fmt.Sprintf(netConfStr, tc.cniVersion, BRNAME) conf := fmt.Sprintf(netConfStr, tc.cniVersion, BRNAME)
if tc.subnet != "" || tc.ranges != nil { if tc.subnet != "" || tc.ranges != nil {
conf += ipamStartStr conf += ipamStartStr
if dataDir != "" {
conf += fmt.Sprintf(ipamDataDirStr, dataDir)
}
if tc.subnet != "" { if tc.subnet != "" {
conf += tc.subnetConfig() conf += tc.subnetConfig()
} }
@ -152,8 +160,8 @@ var counter uint
// createCmdArgs generates network configuration and creates command // createCmdArgs generates network configuration and creates command
// arguments for a test case. // arguments for a test case.
func (tc testCase) createCmdArgs(targetNS ns.NetNS) *skel.CmdArgs { func (tc testCase) createCmdArgs(targetNS ns.NetNS, dataDir string) *skel.CmdArgs {
conf := tc.netConfJSON() conf := tc.netConfJSON(dataDir)
defer func() { counter += 1 }() defer func() { counter += 1 }()
return &skel.CmdArgs{ return &skel.CmdArgs{
ContainerID: fmt.Sprintf("dummy-%d", counter), ContainerID: fmt.Sprintf("dummy-%d", counter),
@ -215,7 +223,7 @@ func ipVersion(ip net.IP) string {
type cmdAddDelTester interface { type cmdAddDelTester interface {
setNS(testNS ns.NetNS, targetNS ns.NetNS) setNS(testNS ns.NetNS, targetNS ns.NetNS)
cmdAddTest(tc testCase) cmdAddTest(tc testCase, dataDir string)
cmdDelTest(tc testCase) cmdDelTest(tc testCase)
} }
@ -240,9 +248,9 @@ func (tester *testerV03x) setNS(testNS ns.NetNS, targetNS ns.NetNS) {
tester.targetNS = targetNS tester.targetNS = targetNS
} }
func (tester *testerV03x) cmdAddTest(tc testCase) { func (tester *testerV03x) cmdAddTest(tc testCase, dataDir string) {
// Generate network config and command arguments // Generate network config and command arguments
tester.args = tc.createCmdArgs(tester.targetNS) tester.args = tc.createCmdArgs(tester.targetNS, dataDir)
// Execute cmdADD on the plugin // Execute cmdADD on the plugin
var result *current.Result var result *current.Result
@ -419,9 +427,9 @@ func (tester *testerV01xOr02x) setNS(testNS ns.NetNS, targetNS ns.NetNS) {
tester.targetNS = targetNS tester.targetNS = targetNS
} }
func (tester *testerV01xOr02x) cmdAddTest(tc testCase) { func (tester *testerV01xOr02x) cmdAddTest(tc testCase, dataDir string) {
// Generate network config and calculate gateway addresses // Generate network config and calculate gateway addresses
tester.args = tc.createCmdArgs(tester.targetNS) tester.args = tc.createCmdArgs(tester.targetNS, dataDir)
// Execute cmdADD on the plugin // Execute cmdADD on the plugin
err := tester.testNS.Do(func(ns.NetNS) error { err := tester.testNS.Do(func(ns.NetNS) error {
@ -537,7 +545,7 @@ func (tester *testerV01xOr02x) cmdDelTest(tc testCase) {
Expect(err).NotTo(HaveOccurred()) Expect(err).NotTo(HaveOccurred())
} }
func cmdAddDelTest(testNS ns.NetNS, tc testCase) { func cmdAddDelTest(testNS ns.NetNS, tc testCase, dataDir string) {
// Get a Add/Del tester based on test case version // Get a Add/Del tester based on test case version
tester := testerByVersion(tc.cniVersion) tester := testerByVersion(tc.cniVersion)
@ -547,7 +555,7 @@ func cmdAddDelTest(testNS ns.NetNS, tc testCase) {
tester.setNS(testNS, targetNS) tester.setNS(testNS, targetNS)
// Test IP allocation // Test IP allocation
tester.cmdAddTest(tc) tester.cmdAddTest(tc, dataDir)
// Test IP Release // Test IP Release
tester.cmdDelTest(tc) tester.cmdDelTest(tc)
@ -558,15 +566,20 @@ func cmdAddDelTest(testNS ns.NetNS, tc testCase) {
var _ = Describe("bridge Operations", func() { var _ = Describe("bridge Operations", func() {
var originalNS ns.NetNS var originalNS ns.NetNS
var dataDir string
BeforeEach(func() { BeforeEach(func() {
// Create a new NetNS so we don't modify the host // Create a new NetNS so we don't modify the host
var err error var err error
originalNS, err = testutils.NewNS() originalNS, err = testutils.NewNS()
Expect(err).NotTo(HaveOccurred()) Expect(err).NotTo(HaveOccurred())
dataDir, err = ioutil.TempDir("", "bridge_test")
Expect(err).NotTo(HaveOccurred())
}) })
AfterEach(func() { AfterEach(func() {
Expect(os.RemoveAll(dataDir)).To(Succeed())
Expect(originalNS.Close()).To(Succeed()) Expect(originalNS.Close()).To(Succeed())
}) })
@ -661,7 +674,7 @@ var _ = Describe("bridge Operations", func() {
} }
for _, tc := range testCases { for _, tc := range testCases {
tc.cniVersion = "0.3.0" tc.cniVersion = "0.3.0"
cmdAddDelTest(originalNS, tc) cmdAddDelTest(originalNS, tc, dataDir)
} }
}) })
@ -691,7 +704,7 @@ var _ = Describe("bridge Operations", func() {
} }
for _, tc := range testCases { for _, tc := range testCases {
tc.cniVersion = "0.3.1" tc.cniVersion = "0.3.1"
cmdAddDelTest(originalNS, tc) cmdAddDelTest(originalNS, tc, dataDir)
} }
}) })
@ -707,7 +720,7 @@ var _ = Describe("bridge Operations", func() {
Expect(err).NotTo(HaveOccurred()) Expect(err).NotTo(HaveOccurred())
defer targetNS.Close() defer targetNS.Close()
tester.setNS(originalNS, targetNS) tester.setNS(originalNS, targetNS)
tester.args = tc.createCmdArgs(targetNS) tester.args = tc.createCmdArgs(targetNS, dataDir)
// Execute cmdDEL on the plugin, expect no errors // Execute cmdDEL on the plugin, expect no errors
tester.cmdDelTest(tc) tester.cmdDelTest(tc)
@ -739,7 +752,7 @@ var _ = Describe("bridge Operations", func() {
} }
for _, tc := range testCases { for _, tc := range testCases {
tc.cniVersion = "0.1.0" tc.cniVersion = "0.1.0"
cmdAddDelTest(originalNS, tc) cmdAddDelTest(originalNS, tc, dataDir)
} }
}) })
@ -909,7 +922,7 @@ var _ = Describe("bridge Operations", func() {
Expect(err).NotTo(HaveOccurred()) Expect(err).NotTo(HaveOccurred())
origMac := link.Attrs().HardwareAddr origMac := link.Attrs().HardwareAddr
cmdAddDelTest(originalNS, tc) cmdAddDelTest(originalNS, tc, dataDir)
link, err = netlink.LinkByName(BRNAME) link, err = netlink.LinkByName(BRNAME)
Expect(err).NotTo(HaveOccurred()) Expect(err).NotTo(HaveOccurred())