feat(driver): add env param to Launch interface + all implementations

internal/driver/chrome/driver.go

@@ -77,7 +77,7 @@ func New() *Driver {
 	return d
 }
 
-func (d *Driver) Launch(ctx context.Context, bundleID string, clearState bool) error {
+func (d *Driver) Launch(ctx context.Context, bundleID string, clearState bool, _ map[string]string) error {
 	if clearState {
 		if err := chromedp.Run(d.tabCtx, network.ClearBrowserCookies()); err != nil {
 			return fmt.Errorf("clear cookies: %w", err)

internal/driver/driver.go

@@ -10,7 +10,7 @@ import (
 // lives under driver/sidecar; the web implementation under driver/chrome;
 // tests use driver/mock.
 type DeviceDriver interface {
-	Launch(ctx context.Context, bundleID string, clearState bool) error
+	Launch(ctx context.Context, bundleID string, clearState bool, env map[string]string) error
 	Terminate(ctx context.Context) error
 
 	Tap(ctx context.Context, x, y int) error

internal/driver/mock/mock.go

@@ -88,7 +88,7 @@ func (d *Driver) failure(kind ActionKind) error {
 	return d.Failures[kind]
 }
 
-func (d *Driver) Launch(_ context.Context, bundleID string, clearState bool) error {
+func (d *Driver) Launch(_ context.Context, bundleID string, clearState bool, _ map[string]string) error {
 	if err := d.failure(ActionLaunch); err != nil {
 		return err
 	}

internal/driver/mock/mock_test.go

@@ -29,7 +29,7 @@ func TestRecordsAllActionsInOrder(t *testing.T) {
 	mock := New()
 	ctx := context.Background()
 
-	if err := mock.Launch(ctx, "com.example", true); err != nil {
+	if err := mock.Launch(ctx, "com.example", true, nil); err != nil {
 		t.Fatal(err)
 	}
 	if err := mock.Tap(ctx, 100, 200); err != nil {

internal/driver/sidecar/client.go

@@ -48,10 +48,11 @@ func (c *Client) WaitForHealth(ctx context.Context, pollInterval time.Duration)
 	}
 }
 
-func (c *Client) Launch(ctx context.Context, bundleID string, clearState bool) error {
+func (c *Client) Launch(ctx context.Context, bundleID string, clearState bool, env map[string]string) error {
 	_, err := c.stub.Launch(ctx, &driverpb.LaunchRequest{
 		BundleId:   bundleID,
 		ClearState: clearState,
+		Env:        env,
 	})
 	return err
 }

internal/driver/sidecar/client_test.go

@@ -191,7 +191,7 @@ func TestClient_LaunchAndTerminate(t *testing.T) {
 	client, _ := Dial(state.address)
 	defer client.Close()
 
-	if err := client.Launch(context.Background(), "com.example", true); err != nil {
+	if err := client.Launch(context.Background(), "com.example", true, nil); err != nil {
 		t.Fatal(err)
 	}
 	if state.fake.launchedBundleID != "com.example" || !state.fake.clearState {