added test for spark driver too

crawshaw · Jul 30, 2015 · e1d76d5 · e1d76d5
1 parent f64d89f
commit e1d76d5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/examples/examples_test.go b/examples/examples_test.go
@@ -317,6 +317,7 @@ func TestExampleObjectSchemas(t *testing.T) {
 			"spark-master-service":    &api.Service{},
 			"spark-master":            &api.Pod{},
 			"spark-worker-controller": &api.ReplicationController{},
+			"spark-driver":            &api.Pod{},
 		},
 		"../examples/storm": {
 			"storm-nimbus-service":    &api.Service{},

diff --git a/test/e2e/examples.go b/test/e2e/examples.go
@@ -159,22 +159,26 @@ var _ = Describe("Examples e2e", func() {
 	})
 
 	Describe("[Skipped][Example]Spark", func() {
-		It("should start spark master and workers", func() {
+		It("should start spark master, driver and workers", func() {
 			mkpath := func(file string) string {
 				return filepath.Join(testContext.RepoRoot, "examples", "spark", file)
 			}
 			serviceJson := mkpath("spark-master-service.json")
 			masterJson := mkpath("spark-master.json")
+			driverJson := mkpath("spark-driver.json")
 			workerControllerJson := mkpath("spark-worker-controller.json")
 			nsFlag := fmt.Sprintf("--namespace=%v", ns)
 
 			By("starting master")
 			runKubectl("create", "-f", serviceJson, nsFlag)
 			runKubectl("create", "-f", masterJson, nsFlag)
+			runKubectl("create", "-f", driverJson, nsFlag)
 			err := waitForPodRunningInNamespace(c, "spark-master", ns)
 			Expect(err).NotTo(HaveOccurred())
 			_, err = lookForStringInLog(ns, "spark-master", "spark-master", "Starting Spark master at", serverStartTimeout)
 			Expect(err).NotTo(HaveOccurred())
+			_, err = lookForStringInLog(ns, "spark-driver", "spark-driver", "Starting Spark driver at", serverStartTimeout)
+			Expect(err).NotTo(HaveOccurred())
 
 			By("starting workers")
 			runKubectl("create", "-f", workerControllerJson, nsFlag)