Capture VCR record

gemmahou · Apr 2, 2024 · 94aac82 · 94aac82
1 parent dce5573
commit 94aac82
Show file tree

Hide file tree

Showing 17 changed files with 49,100 additions and 61 deletions.
diff --git a/config/tests/samples/create/harness.go b/config/tests/samples/create/harness.go
@@ -16,6 +16,8 @@ package create
 
 import (
 	"context"
+	"github.com/GoogleCloudPlatform/k8s-config-connector/pkg/gcp"
+	"golang.org/x/oauth2/google"
 	"net/http"
 	"os"
 	"path/filepath"
@@ -27,7 +29,6 @@ import (
 	"github.com/go-logr/logr"
 	transport_tpg "github.com/hashicorp/terraform-provider-google-beta/google-beta/transport"
 	"golang.org/x/oauth2"
-	"golang.org/x/oauth2/google"
 	cloudresourcemanagerv1 "google.golang.org/api/cloudresourcemanager/v1"
 	"google.golang.org/api/option"
 	"gopkg.in/dnaeon/go-vcr.v3/recorder"
@@ -52,7 +53,6 @@ import (
 	"github.com/GoogleCloudPlatform/k8s-config-connector/pkg/controller/kccmanager/nocache"
 	"github.com/GoogleCloudPlatform/k8s-config-connector/pkg/controller/registration"
 	"github.com/GoogleCloudPlatform/k8s-config-connector/pkg/crd/crdloader"
-	"github.com/GoogleCloudPlatform/k8s-config-connector/pkg/gcp"
 	"github.com/GoogleCloudPlatform/k8s-config-connector/pkg/k8s"
 	"github.com/GoogleCloudPlatform/k8s-config-connector/pkg/logging"
 	"github.com/GoogleCloudPlatform/k8s-config-connector/pkg/test"
@@ -297,6 +297,22 @@ func NewHarness(ctx context.Context, t *testing.T) *Harness {
 		}
 		testgcp.TestKCCAttachedClusterProject.Set("mock-project")
 		h.Project = project
+	} else if os.Getenv("E2E_GCP_TARGET") == "vcr" && os.Getenv("VCR_MODE") == "replay" {
+		h.Project = testgcp.GCPProject{
+			ProjectID:     "example-project",
+			ProjectNumber: 12345678,
+		}
+		mockCloud := mockgcp.NewMockRoundTripper(t, h.client, storage.NewInMemoryStorage())
+
+		roundTripper := http.RoundTripper(mockCloud)
+
+		kccConfig.HTTPClient = &http.Client{Transport: roundTripper}
+
+		// Also hook the oauth2 library
+		ctx = context.WithValue(ctx, oauth2.HTTPClient, kccConfig.HTTPClient)
+		h.Ctx = ctx
+		h.gcpAccessToken = "dummytoken"
+		kccConfig.GCPAccessToken = h.gcpAccessToken
 	} else {
 		h.Project = testgcp.GetDefaultProject(t)
 	}
@@ -346,7 +362,7 @@ func NewHarness(ctx context.Context, t *testing.T) *Harness {
 			if t := ctx.Value(httpRoundTripperKey); t != nil {
 				ret = &http.Client{Transport: t.(http.RoundTripper)}
 			}
-			dir := "pkg/test/resourcefixture/testdata/vcr/cassette/"
+			dir := "./testdata/vcr-cassette/"
 			testName := strings.ReplaceAll(t.Name(), "/", "_")
 			opts := &recorder.Options{
 				CassetteName:  filepath.Join(dir, testName),
@@ -527,6 +543,28 @@ func MaybeSkip(t *testing.T, name string, resources []*unstructured.Unstructured
 			}
 		}
 	}
+	if os.Getenv("E2E_GCP_TARGET") == "vcr" {
+		switch name {
+		case "basicalloydbbackup":
+		//case "fullalloydbbackup":
+		case "basicalloydbcluster":
+		case "basicalloydbsecondarycluster":
+		case "fullalloydbcluster":
+		case "restorebackupalloydbcluster":
+		case "basicalloydbinstance":
+		//case "basicalloydbsecondaryinstance":
+		case "fullalloydbinstance":
+		case "readalloydbinstance":
+		case "zonalalloydbinstance":
+		case "databasealloydbuser":
+		case "iamalloydbuser":
+
+		case "computenodegroup":
+		case "computenodetemplate":
+		default:
+			t.Skipf("test %v not suppported by vcr; skipping", name)
+		}
+	}
 }
 
 func (h *Harness) waitForCRDReady(obj client.Object) {

diff --git a/scripts/github-actions/ga-vcr-test.sh b/scripts/github-actions/ga-vcr-test.sh
@@ -25,5 +25,5 @@ source ${REPO_ROOT}/scripts/fetch_ext_bins.sh && \
 cd ${REPO_ROOT}/
 echo "Running e2e vcr tests..."
 E2E_KUBE_TARGET=envtest \
-	RUN_E2E=1 E2E_GCP_TARGET=vcr VCR_MODE=record \
-	go test -test.count=1 -timeout 3600s -v ./tests/e2e -run TestAllInSeries/fixtures/computenodetemplate 2>&1
+	RUN_E2E=1 E2E_GCP_TARGET=vcr VCR_MODE=replay \
+	go test -test.count=1 -timeout 3600s -v ./tests/e2e -run TestAllInSeries/fixtures 2>&1