From 8ea999357771723465a01607f3d88a6edafb582e Mon Sep 17 00:00:00 2001
From: Angira Kekteeva <kira@nspcc.ru>
Date: Wed, 16 Jun 2021 22:04:14 +0300
Subject: [PATCH] [#309] pkg/session: Add marshal-unmarshal

Added Marshal, Unmarshal, MarshalJSON, UnmarshalJSON methods
to ContainerContext

Signed-off-by: Angira Kekteeva <kira@nspcc.ru>
---
 pkg/session/container.go      | 24 ++++++++++++++++++++++++
 pkg/session/container_test.go | 25 +++++++++++++++++++++++++
 2 files changed, 49 insertions(+)

diff --git a/pkg/session/container.go b/pkg/session/container.go
index bcf5c3d..a3d2d91 100644
--- a/pkg/session/container.go
+++ b/pkg/session/container.go
@@ -113,3 +113,27 @@ func (x *ContainerContext) ForSetEACL() {
 func (x *ContainerContext) IsForSetEACL() bool {
 	return x.isForVerb(session.ContainerVerbSetEACL)
 }
+
+// Marshal marshals ContainerContext into a protobuf binary form.
+func (x *ContainerContext) Marshal(bs ...[]byte) ([]byte, error) {
+	var buf []byte
+	if len(bs) > 0 {
+		buf = bs[0]
+	}
+	return x.ToV2().StableMarshal(buf)
+}
+
+// Unmarshal unmarshals protobuf binary representation of ContainerContext.
+func (x *ContainerContext) Unmarshal(data []byte) error {
+	return x.ToV2().Unmarshal(data)
+}
+
+// MarshalJSON encodes ContainerContext to protobuf JSON format.
+func (x *ContainerContext) MarshalJSON() ([]byte, error) {
+	return x.ToV2().MarshalJSON()
+}
+
+// UnmarshalJSON decodes ContainerContext from protobuf JSON format.
+func (x *ContainerContext) UnmarshalJSON(data []byte) error {
+	return x.ToV2().UnmarshalJSON(data)
+}
diff --git a/pkg/session/container_test.go b/pkg/session/container_test.go
index 6dd2c1f..38c97ee 100644
--- a/pkg/session/container_test.go
+++ b/pkg/session/container_test.go
@@ -5,6 +5,7 @@ import (
 
 	cidtest "github.com/nspcc-dev/neofs-api-go/pkg/container/id/test"
 	"github.com/nspcc-dev/neofs-api-go/pkg/session"
+	sessiontest "github.com/nspcc-dev/neofs-api-go/pkg/session/test"
 	v2session "github.com/nspcc-dev/neofs-api-go/v2/session"
 	"github.com/stretchr/testify/require"
 )
@@ -85,3 +86,27 @@ func TestFilter_ToV2(t *testing.T) {
 		require.Nil(t, cV2.ContainerID())
 	})
 }
+
+func TestContainerContextEncoding(t *testing.T) {
+	c := sessiontest.ContainerContext()
+
+	t.Run("binary", func(t *testing.T) {
+		data, err := c.Marshal()
+		require.NoError(t, err)
+
+		c2 := session.NewContainerContext()
+		require.NoError(t, c2.Unmarshal(data))
+
+		require.Equal(t, c, c2)
+	})
+
+	t.Run("json", func(t *testing.T) {
+		data, err := c.MarshalJSON()
+		require.NoError(t, err)
+
+		c2 := session.NewContainerContext()
+		require.NoError(t, c2.UnmarshalJSON(data))
+
+		require.Equal(t, c, c2)
+	})
+}