From 2952af0d5974fb37e3f308f03f6a7fc9cecd6af9 Mon Sep 17 00:00:00 2001
From: deads2k <deads@redhat.com>
Date: Wed, 12 Aug 2015 15:38:51 -0400
Subject: [PATCH] expose e2e methods for downstream use

---
 test/e2e/core.go     | 2 +-
 test/e2e/e2e_test.go | 2 +-
 test/e2e/util.go     | 4 ++++
 3 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/test/e2e/core.go b/test/e2e/core.go
index 8c67669f424..941f1a3aaaa 100644
--- a/test/e2e/core.go
+++ b/test/e2e/core.go
@@ -29,7 +29,7 @@ type command struct {
 	component string
 }
 
-func coreDump(dir string) {
+func CoreDump(dir string) {
 	c, err := loadClient()
 	if err != nil {
 		fmt.Printf("Error creating client: %v", err)
diff --git a/test/e2e/e2e_test.go b/test/e2e/e2e_test.go
index 9b9e6ee55e5..4e504843e28 100644
--- a/test/e2e/e2e_test.go
+++ b/test/e2e/e2e_test.go
@@ -91,7 +91,7 @@ func TestE2E(t *testing.T) {
 		if err := os.MkdirAll(*reportDir, 0755); err != nil {
 			glog.Errorf("Failed creating report directory: %v", err)
 		}
-		defer coreDump(*reportDir)
+		defer CoreDump(*reportDir)
 	}
 
 	if testContext.Provider == "" {
diff --git a/test/e2e/util.go b/test/e2e/util.go
index 066a8f8fc7b..48867f1ebe4 100644
--- a/test/e2e/util.go
+++ b/test/e2e/util.go
@@ -125,6 +125,10 @@ type TestContextType struct {
 
 var testContext TestContextType
 
+func SetTestContext(t TestContextType) {
+	testContext = t
+}
+
 type ContainerFailures struct {
 	status   *api.ContainerStateTerminated
 	restarts int