-
Notifications
You must be signed in to change notification settings - Fork 250
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #386 from heschlie/install-cni-tests
Adding tests for install-cni.sh
- Loading branch information
Showing
8 changed files
with
273 additions
and
9 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -10,3 +10,5 @@ delstats | |
| *.pyc | ||
| *.created | ||
| tmp-cni | ||
| k8s-install/scripts/tmp/ | ||
| install_cni.test | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,20 @@ | ||
| { | ||
| "name": "k8s-pod-network", | ||
| "type": "calico", | ||
| "etcd_endpoints": "", | ||
| "etcd_key_file": "", | ||
| "etcd_cert_file": "", | ||
| "etcd_ca_cert_file": "", | ||
| "log_level": "warn", | ||
| "ipam": { | ||
| "type": "calico-ipam" | ||
| }, | ||
| "policy": { | ||
| "type": "k8s", | ||
| "k8s_api_root": "https://127.0.0.1:8080", | ||
| "k8s_auth_token": "my-secret-key" | ||
| }, | ||
| "kubernetes": { | ||
| "kubeconfig": "/etc/cni/net.d/calico-kubeconfig" | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,204 @@ | ||
| package scripts_test | ||
|
|
||
| import ( | ||
| . "github.com/onsi/ginkgo" | ||
| . "github.com/onsi/gomega" | ||
| "github.com/projectcalico/felix/fv/containers" | ||
|
|
||
| "fmt" | ||
| "io/ioutil" | ||
| "os" | ||
| "time" | ||
| ) | ||
|
|
||
| var secretFile = "tmp/serviceaccount/token" | ||
|
|
||
| // runCniContainer will run install-cni.sh. | ||
| // TODO: We should be returning an error if the command fails, there currently is | ||
| // not a way to get that from the container package. | ||
| func runCniContainer(extraArgs ...string) error { | ||
| // Get the CWD for mounting directories into container. | ||
| cwd, err := os.Getwd() | ||
| if err != nil { | ||
| Fail("Could not get CWD") | ||
| } | ||
|
|
||
| // Assemble our arguments. | ||
| args := []string{ | ||
| "-e", "SLEEP=false", | ||
| "-e", "KUBERNETES_SERVICE_HOST=127.0.0.1", | ||
| "-e", "KUBERNETES_SERVICE_PORT=8080", | ||
| "-v", cwd + "/tmp/bin:/host/opt/cni/bin", | ||
| "-v", cwd + "/tmp/net.d:/host/etc/cni/net.d", | ||
| "-v", cwd + "/tmp/serviceaccount:/var/run/secrets/kubernetes.io/serviceaccount", | ||
| } | ||
| args = append(args, extraArgs...) | ||
| args = append(args, "calico/cni:latest", "/install-cni.sh") | ||
|
|
||
| c := containers.Run("cni", args...) | ||
| Eventually(func() bool { return c.Stopped() }, 5*time.Second, 100*time.Millisecond).Should(BeTrue()) | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| // cleanup uses the calico/cni container to cleanup after itself as it creates | ||
| // things as root. | ||
| func cleanup() { | ||
| cwd, err := os.Getwd() | ||
| if err != nil { | ||
| Fail("Could not get CWD") | ||
| } | ||
| containers.Run("cni", | ||
| "-e", "SLEEP=false", | ||
| "-e", "KUBERNETES_SERVICE_HOST=127.0.0.1", | ||
| "-e", "KUBERNETES_SERVICE_PORT=8080", | ||
| "-v", cwd+"/tmp/bin:/host/opt/cni/bin", | ||
| "-v", cwd+"/tmp/net.d:/host/etc/cni/net.d", | ||
| "-v", cwd+"/tmp/serviceaccount:/var/run/secrets/kubernetes.io/serviceaccount", | ||
| "calico/cni:latest", | ||
| "sh", "-c", "rm -f /host/opt/cni/bin/* /host/etc/cni/net.d/*") | ||
| } | ||
|
|
||
| var _ = BeforeSuite(func() { | ||
| // Make the directories we'll need for storing files. | ||
| err := os.MkdirAll("tmp/bin", 0755) | ||
| if err != nil { | ||
| Fail("Failed to create directory tmp/bin") | ||
| } | ||
| err = os.MkdirAll("tmp/net.d", 0755) | ||
| if err != nil { | ||
| Fail("Failed to create directory tmp/net.d") | ||
| } | ||
| err = os.MkdirAll("tmp/serviceaccount", 0755) | ||
| if err != nil { | ||
| Fail("Failed to create directory tmp/serviceaccount") | ||
| } | ||
| cleanup() | ||
|
|
||
| // Create a secrets file for parsing. | ||
| k8sSecret := []byte("my-secret-key") | ||
| err = ioutil.WriteFile(secretFile, k8sSecret, 0755) | ||
| if err != nil { | ||
| Fail(fmt.Sprintf("Failed to write k8s secret file: %v", err)) | ||
| } | ||
| }) | ||
|
|
||
| var _ = AfterSuite(func() { | ||
| err := os.RemoveAll("tmp") | ||
| if err != nil { | ||
| fmt.Println("Failed to clean up directories") | ||
| } | ||
| }) | ||
|
|
||
| var _ = Describe("install-cni.sh tests", func() { | ||
| AfterEach(func() { | ||
| cleanup() | ||
| }) | ||
|
|
||
| Describe("Run install-cni", func() { | ||
| Context("With default values", func() { | ||
| It("Should install bins and config", func() { | ||
| err := runCniContainer() | ||
| Expect(err).NotTo(HaveOccurred()) | ||
|
|
||
| // Get a list of files in the defualt CNI bin location. | ||
| files, err := ioutil.ReadDir("tmp/bin") | ||
| if err != nil { | ||
| Fail("Could not list the files in tmp/bin") | ||
| } | ||
| names := []string{} | ||
| for _, file := range files { | ||
| names = append(names, file.Name()) | ||
| } | ||
|
|
||
| // Get a list of files in the default location for CNI config. | ||
| files, err = ioutil.ReadDir("tmp/net.d") | ||
| if err != nil { | ||
| Fail("Could not list the files in tmp/net.d") | ||
| } | ||
| for _, file := range files { | ||
| names = append(names, file.Name()) | ||
| } | ||
|
|
||
| Expect(names).To(ContainElement("calico")) | ||
| Expect(names).To(ContainElement("calico-ipam")) | ||
| Expect(names).To(ContainElement("10-calico.conf")) | ||
| }) | ||
| It("Should parse and output a templated config", func() { | ||
| err := runCniContainer() | ||
| Expect(err).NotTo(HaveOccurred()) | ||
|
|
||
| expectedFile := "expected_10-calico.conf" | ||
|
|
||
| var expected = []byte{} | ||
| if file, err := os.Open(expectedFile); err != nil { | ||
| Fail("Could not open " + expectedFile + " for reading") | ||
| } else { | ||
| _, err := file.Read(expected) | ||
| if err != nil { | ||
| Fail("Could not read from expected_10-calico.conf") | ||
| } | ||
| err = file.Close() | ||
| if err != nil { | ||
| fmt.Println("Failed to close expected_10-calico.conf") | ||
| } | ||
| } | ||
|
|
||
| var received = []byte{} | ||
| if file, err := os.Open("tmp/net.d/10-calico.conf"); err != nil { | ||
| Fail("Could not open 10-calico.conf for reading") | ||
| } else { | ||
| _, err := file.Read(received) | ||
| if err != nil { | ||
| Fail("Could not read from 10-calico.conf") | ||
| } | ||
| err = file.Close() | ||
| if err != nil { | ||
| fmt.Println("Failed to close 10-calico.conf") | ||
| } | ||
| } | ||
|
|
||
| Expect(expected).To(Equal(received)) | ||
| }) | ||
| }) | ||
|
|
||
| Context("With modified env vars", func() { | ||
| It("Should rename '10-calico.conf' to '10-calico.conflist'", func() { | ||
| err := runCniContainer("-e", "CNI_CONF_NAME=10-calico.conflist") | ||
| Expect(err).NotTo(HaveOccurred()) | ||
|
|
||
| expectedFile := "expected_10-calico.conf" | ||
|
|
||
| var expected = []byte{} | ||
| if file, err := os.Open(expectedFile); err != nil { | ||
| Fail("Could not open " + expectedFile + " for reading") | ||
| } else { | ||
| _, err := file.Read(expected) | ||
| if err != nil { | ||
| Fail("Could not read from expected_10-calico.conf") | ||
| } | ||
| err = file.Close() | ||
| if err != nil { | ||
| fmt.Println("Failed to close expected_10-calico.conf") | ||
| } | ||
| } | ||
|
|
||
| var received = []byte{} | ||
| if file, err := os.Open("tmp/net.d/10-calico.conflist"); err != nil { | ||
| Fail("Could not open 10-calico.conflist for reading") | ||
| } else { | ||
| _, err := file.Read(received) | ||
| if err != nil { | ||
| Fail("Could not read from 10-calico.conflist") | ||
| } | ||
| err = file.Close() | ||
| if err != nil { | ||
| fmt.Println("Failed to close 10-calico.conflist") | ||
| } | ||
| } | ||
|
|
||
| Expect(expected).To(Equal(received)) | ||
| }) | ||
| }) | ||
| }) | ||
| }) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,13 @@ | ||
| package scripts_test | ||
|
|
||
| import ( | ||
| . "github.com/onsi/ginkgo" | ||
| . "github.com/onsi/gomega" | ||
|
|
||
| "testing" | ||
| ) | ||
|
|
||
| func TestScripts(t *testing.T) { | ||
| RegisterFailHandler(Fail) | ||
| RunSpecs(t, "Scripts Suite") | ||
| } |