Fix unit tests

bcongdon · Apr 28, 2018 · dc9bbad · dc9bbad
1 parent 421633a
commit dc9bbad
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/driver.go b/driver.go
@@ -131,7 +131,7 @@ func (d *Driver) runMapPhase(job *Job, jobNumber int, inputs []string) {
 	inputSplits := job.inputSplits(inputs, d.config.SplitSize)
 	if len(inputSplits) == 0 {
 		log.Warnf("No input splits")
-		os.Exit(0)
+		return
 	}
 	log.Debugf("Number of job input splits: %d", len(inputSplits))
 
@@ -188,7 +188,7 @@ func (d *Driver) run() {
 
 	if len(d.config.Inputs) == 0 {
 		log.Error("No inputs!")
-		os.Exit(1)
+		return
 	}
 
 	inputs := d.config.Inputs

diff --git a/lambda_test.go b/lambda_test.go
@@ -40,7 +40,9 @@ func TestHandleRequest(t *testing.T) {
 		WorkingLocation:  ".",
 	}
 
-	job := &Job{}
+	job := &Job{
+		config: &config{},
+	}
 
 	// These values should be reset to 0 by Lambda handler function
 	job.bytesRead = 10