add option to enable/disable remount on resume

rockset · Oct 2, 2023 · 9df63c1 · 9df63c1
1 parent 0d5f678
commit 9df63c1
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/option/virtual_instance.go b/option/virtual_instance.go
@@ -38,6 +38,13 @@ func WithNoMountRefresh() VirtualInstanceOption {
 	}
 }
 
+// WithRemountOnResume sets remount on Virtual Instance resume.
+func WithRemountOnResume(remount bool) VirtualInstanceOption {
+	return func(o *VirtualInstanceOptions) {
+		o.EnableRemountOnResume = &remount
+	}
+}
+
 // WithAutoSuspend duration without queries after which the virtual instance is suspended. Minimum time is 15 minutes.
 func WithAutoSuspend(d time.Duration) VirtualInstanceOption {
 	return func(o *VirtualInstanceOptions) {

diff --git a/virtual_instances_test.go b/virtual_instances_test.go
@@ -126,6 +126,7 @@ func (s *VirtualInstanceIntegrationSuite) TestVirtualInstance_0_Create() {
 	vi, err := rc.CreateVirtualInstance(ctx, s.name,
 		option.WithVirtualInstanceSize(option.SizeSmall),
 		option.WithNoMountRefresh(),
+		option.WithRemountOnResume(true),
 		option.WithAutoSuspend(15*time.Minute),
 	)
 	s.Require().NoError(err)