Issue #98 core acceptance tests

Acceptance_test.go

@@ -3,6 +3,7 @@ package main_test
 import (
 	"fmt"
 	"io/ioutil"
+	"math"
 	"net/http"
 	"regexp"
 
@@ -18,46 +19,203 @@ import (
 
 var _ = Describe("Acceptance", func() {
 
-	It("Halts execution if a payload input file is not found", func() {
-		list := []string{
-			fmt.Sprintf(`%s -X POST -d '{"name":"talula"}'`, URLForTestServer("/success")),
-			fmt.Sprintf(`%s -X POST -d @missing-file.json`, URLForTestServer("/success")),
-		}
+	BeforeEach(func() {
+		TestServer.Clear()
+		factory := rizo.HTTPResponseFactory(func(w http.ResponseWriter) {
+			w.Header().Add("done", "1")
+			w.WriteHeader(http.StatusOK)
+		})
 
-		output, err := test.ExecuteList("./corcel", list)
-		Expect(err).ToNot(BeNil())
+		TestServer.Use(factory).For(rizo.RequestWithPath("/people"))
+	})
 
-		Expect(string(output)).To(ContainSubstring("Request body file not found: missing-file.json"))
+	AfterEach(func() {
+		TestServer.Clear()
 	})
 
-	It("Outputs a summary to STDOUT", func() {
+	Describe("Core Command Line Usage", func() {
+
+		Describe("Iterations", func() {
+
+			It("Plan Usage", func() {
+
+				var plan = fmt.Sprintf(`---
+name: Some Plan
+iterations: 5
+random: false
+workers: 1
+waitTime: 0s
+duration: 0s
+jobs:
+    - name: Some Job
+      steps:
+      - name: Some Step
+        action:
+          type: HttpRequest
+          httpHeaders:
+            key: 1
+          method: GET
+          url: %s`, TestServer.CreateURL("/people"))
+
+				summary, err := test.ExecutePlanFromData(plan, "--summary")
+
+				Expect(err).To(BeNil())
+
+				Expect(summary.TotalRequests).To(Equal(float64(5)))
+			})
+
+			It("List Usage", func() {
+				list := []string{
+					fmt.Sprintf(`%s -X GET'`, TestServer.CreateURL("/people")),
+				}
+				summary, err := test.ExecuteList(list, "--summary", "--iterations", "5")
+				Expect(err).To(BeNil())
+
+				Expect(summary.TotalRequests).To(Equal(float64(5)))
+			})
+		})
+
+		Describe("Workers", func() {
+
+			It("Plan Usage", func() {
+
+				var plan = fmt.Sprintf(`---
+name: Some Plan
+iterations: 0
+random: false
+workers: 5
+waitTime: 0s
+duration: 0s
+jobs:
+    - name: Some Job
+      steps:
+      - name: Some Step
+        action:
+          type: HttpRequest
+          httpHeaders:
+            key: 1
+          method: GET
+          url: %s`, TestServer.CreateURL("/people"))
+
+				summary, err := test.ExecutePlanFromData(plan, "--summary")
+
+				Expect(err).To(BeNil())
+
+				Expect(summary.TotalRequests).To(Equal(float64(5)))
+			})
+
+			It("List Usage", func() {
+				list := []string{
+					fmt.Sprintf(`%s -X GET'`, TestServer.CreateURL("/people")),
+				}
+				summary, err := test.ExecuteList(list, "--summary", "--workers", "5")
+				Expect(err).To(BeNil())
+
+				Expect(summary.TotalRequests).To(Equal(float64(5)))
+			})
+		})
+
+		Describe("Wait Time", func() {
+
+			It("Plan Usage", func() {
+
+				var plan = fmt.Sprintf(`---
+name: Some Plan
+iterations: 5
+random: false
+workers: 1
+waitTime: 1s
+duration: 0s
+jobs:
+    - name: Some Job
+      steps:
+      - name: Some Step
+        action:
+          type: HttpRequest
+          httpHeaders:
+            key: 1
+          method: GET
+          url: %s`, TestServer.CreateURL("/people"))
+
+				summary, err := test.ExecutePlanFromData(plan, "--summary")
+
+				Expect(err).To(BeNil())
+
+				Expect(math.Floor(summary.RunningTime.Seconds())).To(Equal(float64(5)))
+			})
+
+			It("List Usage", func() {
+				list := []string{
+					fmt.Sprintf(`%s -X GET'`, TestServer.CreateURL("/people")),
+				}
+				summary, err := test.ExecuteList(list, "--summary", "--wait-time", "1s", "--iterations", "5")
+				Expect(err).To(BeNil())
+
+				Expect(math.Floor(summary.RunningTime.Seconds())).To(Equal(float64(5)))
+			})
+		})
+
+		Describe("Duration", func() {
+
+			It("Plan Usage", func() {
+
+				var plan = fmt.Sprintf(`---
+name: Some Plan
+iterations: 0
+random: false
+workers: 1
+waitTime: 0s
+duration: 5s
+jobs:
+    - name: Some Job
+      steps:
+      - name: Some Step
+        action:
+          type: HttpRequest
+          httpHeaders:
+            key: 1
+          method: GET
+          url: %s`, TestServer.CreateURL("/people"))
+
+				summary, err := test.ExecutePlanFromData(plan, "--summary")
+
+				Expect(err).To(BeNil())
+
+				Expect(math.Floor(summary.RunningTime.Seconds())).To(Equal(float64(5)))
+			})
+
+			It("List Usage", func() {
+				list := []string{
+					fmt.Sprintf(`%s -X GET'`, TestServer.CreateURL("/people")),
+				}
+				summary, err := test.ExecuteList(list, "--summary", "--duration", "5s")
+				Expect(err).To(BeNil())
+
+				Expect(math.Floor(summary.RunningTime.Seconds())).To(Equal(float64(5)))
+			})
+		})
+	})
+
+	It("Halts execution if a payload input file is not found", func() {
 		list := []string{
-			fmt.Sprintf(`%s -X POST `, URLForTestServer("/error")),
-			fmt.Sprintf(`%s -X POST `, URLForTestServer("/success")),
-			fmt.Sprintf(`%s -X POST `, URLForTestServer("/error")),
-			fmt.Sprintf(`%s -X POST `, URLForTestServer("/success")),
-			fmt.Sprintf(`%s -X POST `, URLForTestServer("/error")),
-			fmt.Sprintf(`%s -X POST `, URLForTestServer("/success")),
+			fmt.Sprintf(`%s -X POST -d '{"name":"talula"}'`, URLForTestServer("/success")),
+			fmt.Sprintf(`%s -X POST -d @missing-file.json`, URLForTestServer("/success")),
 		}
 
-		TestServer.Use(rizo.HTTPResponseFactory(func(w http.ResponseWriter) {
-			w.WriteHeader(500)
-		})).For(rizo.RequestWithPath("/error"))
-
-		output, err := test.ExecuteList("./corcel", list, "--summary")
-		Expect(err).To(BeNil())
+		_, err := test.ExecuteList(list, "--summary")
+		Expect(err).ToNot(BeNil())
 
-		Expect(string(output)).To(ContainSubstring("Summary"))
+		Expect(fmt.Sprintf("%v", err)).To(ContainSubstring("Request body file not found: missing-file.json"))
 	})
 
-	It("Error non-http url in the urls file causes a run time exception #21", func() {
+	It("Error non-http url in the urls file causes a run time exception", func() {
 		list := []string{
 			fmt.Sprintf(`-Something`),
 		}
 
-		output, err := test.ExecuteList("./corcel", list)
+		_, err := test.ExecuteList(list, "--summary")
 		Expect(err).ToNot(BeNil())
-		Expect(string(output)).To(ContainSubstring(errormanager.LogMessageVaidURLs))
+		Expect(fmt.Sprintf("%v", err)).To(ContainSubstring(errormanager.LogMessageVaidURLs))
 	})
 
 	It("Issue - Should write out panics to a log file and not std out", func() {
@@ -72,7 +230,7 @@ var _ = Describe("Acceptance", func() {
 		output, err := test.ExecutePlanBuilder(planBuilder)
 		Expect(err).ToNot(BeNil())
 
-		Expect(string(output)).To(ContainSubstring("An unexpected error has occurred.  The error has been logged to /tmp/"))
+		Expect(output).To(ContainSubstring("An unexpected error has occurred.  The error has been logged to /tmp/"))
 
 		//Ensure that the file which was generated contains the error which caused the panic
 		r, _ := regexp.Compile(`/tmp/[\w\d-]+`)
@@ -82,31 +240,4 @@ var _ = Describe("Acceptance", func() {
 		Expect(err).To(BeNil())
 		Expect(string(data)).To(ContainSubstring("IPanicAction has caused this panic"))
 	})
-
-	Describe("Shell", func() {
-
-		Describe("run", func() {
-
-			PIt("setting iterations", func() {
-
-			})
-
-			PIt("setting duration", func() {
-
-			})
-
-			PIt("setting workers", func() {
-
-			})
-
-			PIt("setting plan", func() {
-
-			})
-
-			PIt("setting summary", func() {
-
-			})
-		})
-
-	})
 })