storage: add config options for UseFileRegistry and EncryptionOptions

This patch adds the ability to configure encryption-at-rest for
in-memory engines, which are used in tests.

Release note: None

pkg/server/config.go

@@ -521,6 +521,8 @@ func (cfg *Config) CreateEngines(ctx context.Context) (Engines, error) {
 					storage.Attributes(spec.Attributes),
 					storage.CacheSize(cfg.CacheSize),
 					storage.MaxSize(sizeInBytes),
+					storage.UseFileRegistry(spec.UseFileRegistry),
+					storage.EncryptionOptions(spec.EncryptionOptions),
 					storage.Settings(cfg.Settings))
 				if err != nil {
 					return Engines{}, err

pkg/server/sticky_engine.go

@@ -112,6 +112,8 @@ func (registry *stickyInMemEnginesRegistryImpl) GetOrCreateStickyInMemEngine(
 		storage.Attributes(spec.Attributes),
 		storage.CacheSize(cfg.CacheSize),
 		storage.MaxSize(spec.Size.InBytes),
+		storage.UseFileRegistry(spec.UseFileRegistry),
+		storage.EncryptionOptions(spec.EncryptionOptions),
 		storage.ForTesting)
 
 	engineEntry := &stickyInMemEngine{

pkg/storage/open.go

@@ -106,6 +106,27 @@ func CacheSize(size int64) ConfigOption {
 	}
 }
 
+// UseFileRegistry configures an engine to use the encryption-at-rest file
+// registry. It is used to configure encryption-at-rest for in-memory engines,
+// which are used in tests.
+func UseFileRegistry(useRegistry bool) ConfigOption {
+	return func(cfg *engineConfig) error {
+		cfg.UseFileRegistry = useRegistry
+		return nil
+	}
+}
+
+// EncryptionOptions configures the encryption-at-rest options for an engine.
+// It is used to configure encryption-at-rest for in-memory engines, which are
+// used in tests.
+func EncryptionOptions(b []byte) ConfigOption {
+	return func(cfg *engineConfig) error {
+		cfg.EncryptionOptions = make([]byte, len(b))
+		copy(cfg.EncryptionOptions, b)
+		return nil
+	}
+}
+
 // Hook configures a hook to initialize additional storage options. It's used
 // to initialize encryption-at-rest details in CCL builds.
 func Hook(hookFunc func(*base.StorageConfig) error) ConfigOption {