state wasm using go-plugin features

wasm-state-go-plugin/test-store/test_store.go

@@ -41,7 +41,7 @@ func (store TestStore) Init(context.Context, state.InitRequest) (state.InitRespo
 }
 
 func (store TestStore) Features(context.Context, state.FeaturesRequest) (state.FeaturesResponse, error) {
-	return state.FeaturesResponse{Features: []string{}}, nil
+	return state.FeaturesResponse{Features: []string{"TRANSACTIONAL"}}, nil
 }
 
 func (store TestStore) Set(_ context.Context, req state.SetRequest) (state.SetResponse, error) {

wasm-state-go-plugin/wasm.go

@@ -51,7 +51,29 @@ func (wasmStore *wasmStore) Close() error {
 }
 
 func (wasmStore *wasmStore) Features() []state.Feature {
-	return []state.Feature{state.FeatureETag, state.FeatureTransactional}
+	features, err := wasmStore.store.Features(context.Background(), stateWasm.FeaturesRequest{})
+
+	if err != nil {
+		return []state.Feature{}
+	}
+
+	var featuresArray []state.Feature
+
+	for _, feature := range features.Features {
+		switch feature {
+		case string(state.FeatureETag):
+			featuresArray = append(featuresArray, state.FeatureETag)
+			break
+		case string(state.FeatureTransactional):
+			featuresArray = append(featuresArray, state.FeatureTransactional)
+			break
+		case string(state.FeatureQueryAPI):
+			featuresArray = append(featuresArray, state.FeatureQueryAPI)
+			break
+		}
+	}
+
+	return featuresArray
 }
 
 func (wasmStore *wasmStore) Delete(req *state.DeleteRequest) error {

wasm-state-go-plugin/wasm_test.go

@@ -65,3 +65,17 @@ func TestReadAndWrite(t *testing.T) {
 		assert.Nil(t, err)
 	})
 }
+
+func TestFeatures(t *testing.T) {
+	store, err := NewStateWasmStore(logger.NewLogger("test"), "./test-store/test_store.wasm")
+	assert.Nil(t, err)
+
+	store.Init(state.Metadata{})
+
+	t.Run("get features", func(t *testing.T) {
+		features := store.Features()
+		assert.NotNil(t, features)
+		assert.Equal(t, 1, len(features))
+		assert.Equal(t, state.FeatureTransactional, features[0])
+	})
+}