fix config unit tests

Signed-off-by: Rowan Jacobs <rojacobs@pivotal.io>

config/load_state_test.go

@@ -1,7 +1,6 @@
 package config_test
 
 import (
-	"bytes"
 	"errors"
 	"fmt"
 	"io/ioutil"
@@ -180,10 +179,6 @@ var _ = Describe("LoadState", func() {
 			})
 
 			Context("when an external bbl-state is specified", func() {
-				BeforeEach(func() {
-					fakeDownloader.DownloadCall.Returns.State = ioutil.NopCloser(bytes.NewReader([]byte("some-remote-state")))
-				})
-
 				It("downloads the bbl state", func() {
 					_, err := c.Bootstrap(bootstrapArgs([]string{
 						"bbl", "print-env",

fakes/downloader.go

@@ -18,8 +18,8 @@ type Downloader struct {
 }
 
 func (d *Downloader) DownloadAndPrepareState(flags config.GlobalFlags) error {
-	d.DownloadAndPrepareStateCall.CallCount++
-	d.DownloadAndPrepareStateCall.Receives.GlobalFlags = flags
+	d.DownloadCall.CallCount++
+	d.DownloadCall.Receives.GlobalFlags = flags
 
-	return d.DownloadAndPrepareStateCall.Returns.Error
+	return d.DownloadCall.Returns.Error
 }