[Internal] Variable storage

sdk/internal/CHANGELOG.md

@@ -1,5 +1,9 @@
 # Release History
 
+## 0.8.4 (Unreleased)
+### Features Added
+* Added variables storage to the `Stop` function. Pass in a `map[string]interface{}` to the `Stop` method options and the values can be retrieved with the `GetVariables(t *testing.T)` function
+
 ## 0.8.3 (2021-11-30)
 
 ### Features Added

sdk/internal/recording/recording.go

@@ -10,6 +10,7 @@ import (
 	"bytes"
 	"crypto/tls"
 	"crypto/x509"
+	"encoding/json"
 	"errors"
 	"fmt"
 	"io/ioutil"
@@ -492,6 +493,7 @@ const (
 type recordedTest struct {
 	recordingId string
 	liveOnly    bool
+	variables   map[string]interface{}
 }
 
 var testSuite = map[string]recordedTest{}
@@ -505,6 +507,7 @@ var client = http.Client{
 type RecordingOptions struct {
 	UseHTTPS        bool
 	GroupForReplace string
+	Variables       map[string]interface{}
 }
 
 func defaultOptions() *RecordingOptions {
@@ -585,11 +588,30 @@ func Start(t *testing.T, pathToRecordings string, options *RecordingOptions) err
 		}
 		return fmt.Errorf("Recording ID was not returned by the response. Response body: %s", b)
 	}
+
+	// Unmarshal any variables returned by the proxy
+	var m map[string]interface{}
+	body, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		return err
+	}
+	if len(body) > 0 {
+		err = json.Unmarshal(body, &m)
+		if err != nil {
+			return err
+		}
+	}
+
 	if val, ok := testSuite[t.Name()]; ok {
 		val.recordingId = recId
+		val.variables = m
 		testSuite[t.Name()] = val
 	} else {
-		testSuite[t.Name()] = recordedTest{recordingId: recId, liveOnly: false}
+		testSuite[t.Name()] = recordedTest{
+			recordingId: recId,
+			liveOnly:    false,
+			variables:   m,
+		}
 	}
 	return nil
 }
@@ -614,13 +636,30 @@ func Stop(t *testing.T, options *RecordingOptions) error {
 	if err != nil {
 		return err
 	}
+	if options.Variables != nil && len(options.Variables) > 0 {
+		req.Header.Set("Content-Type", "application/json")
+		marshalled, err := json.Marshal(options.Variables)
+		if err != nil {
+			return err
+		}
+		req.Body = ioutil.NopCloser(bytes.NewReader(marshalled))
+		req.ContentLength = int64(len(marshalled))
+	}
+
 	var recTest recordedTest
 	var ok bool
 	if recTest, ok = testSuite[t.Name()]; !ok {
 		return errors.New("Recording ID was never set. Did you call StartRecording?")
 	}
 	req.Header.Set("x-recording-id", recTest.recordingId)
-	_, err = client.Do(req)
+	resp, err := client.Do(req)
+	if resp.StatusCode != 200 {
+		b, err := ioutil.ReadAll(resp.Body)
+		if err == nil {
+			return fmt.Errorf("proxy did not stop the recording properly: %s", string(b))
+		}
+		return errors.New("proxy did not stop the recording properly")
+	}
 	return err
 }
 
@@ -721,3 +760,10 @@ func IsLiveOnly(t *testing.T) bool {
 	}
 	return false
 }
+
+func GetVariables(t *testing.T) map[string]interface{} {
+	if s, ok := testSuite[t.Name()]; ok {
+		return s.variables
+	}
+	return nil
+}

sdk/internal/recording/recording_test.go

@@ -569,22 +569,47 @@ func TestFindProxyCertLocation(t *testing.T) {
 	require.Contains(t, location, filepath.Join("eng", "common", "testproxy", "dotnet-devcert.crt"))
 }
 
-func TestModeNotSet(t *testing.T) {
-	proxyMode, _ := os.LookupEnv("AZURE_RECORD_MODE")
-	defer os.Setenv("AZURE_RECORD_MODE", proxyMode)
+func TestVariables(t *testing.T) {
+	temp := recordMode
+	recordMode = RecordingMode
+	defer func() { recordMode = temp }()
 
-	os.Unsetenv("AZURE_RECORD_MODE")
-	require.Equal(t, PlaybackMode, GetRecordMode())
-}
+	err := Start(t, packagePath, nil)
+	require.NoError(t, err)
+
+	client, err := NewRecordingHTTPClient(t, nil)
+	require.NoError(t, err)
 
-func TestModeNotSetStartStop(t *testing.T) {
-	proxyMode, _ := os.LookupEnv("AZURE_RECORD_MODE")
-	defer os.Setenv("AZURE_RECORD_MODE", proxyMode)
+	req, err := http.NewRequest("POST", "https://azsdkengsys.azurecr.io/acr/v1/some_registry/_tags", nil)
+	require.NoError(t, err)
 
-	os.Unsetenv("AZURE_RECORD_MODE")
-	err := Start(t, packagePath, nil)
+	resp, err := client.Do(req)
+	require.NoError(t, err)
+	require.NotNil(t, resp)
+
+	require.NotNil(t, GetRecordingId(t))
+
+	err = Stop(t, &RecordingOptions{Variables: map[string]interface{}{"key1": "value1", "key2": "1"}})
+	require.NoError(t, err)
+
+	recordMode = PlaybackMode
+	err = Start(t, packagePath, nil)
 	require.NoError(t, err)
-	require.Equal(t, PlaybackMode, GetRecordMode())
+
+	variables := GetVariables(t)
+	require.Equal(t, variables["key1"], "value1")
+	require.Equal(t, variables["key2"], "1")
+
 	err = Stop(t, nil)
 	require.NoError(t, err)
+
+	// Make sure the file is there
+	jsonFile, err := os.Open(fmt.Sprintf("./testdata/recordings/%s.json", t.Name()))
+	require.NoError(t, err)
+	defer func() {
+		err = jsonFile.Close()
+		require.NoError(t, err)
+		err = os.Remove(jsonFile.Name())
+		require.NoError(t, err)
+	}()
 }