From bf23b548ab20b51a53acaed2aa53861c8f5c0175 Mon Sep 17 00:00:00 2001
From: Alex Vanin <alexey@nspcc.ru>
Date: Wed, 2 Mar 2022 15:49:03 +0300
Subject: [PATCH] [#158] client: Support object context in session token setter

Signed-off-by: Alex Vanin <alexey@nspcc.ru>
---
 session/session.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/session/session.go b/session/session.go
index 5d08e63..0c905a9 100644
--- a/session/session.go
+++ b/session/session.go
@@ -200,7 +200,8 @@ func (t *Token) Signature() *signature.Signature {
 // SetContext sets context of the Token.
 //
 // Supported contexts:
-//  - *ContainerContext.
+//  - *ContainerContext,
+//  - *ObjectContext.
 //
 // Resets context if it is not supported.
 func (t *Token) SetContext(v interface{}) {
@@ -209,6 +210,8 @@ func (t *Token) SetContext(v interface{}) {
 	switch c := v.(type) {
 	case *ContainerContext:
 		cV2 = c.ToV2()
+	case *ObjectContext:
+		cV2 = c.ToV2()
 	}
 
 	t.setBodyField(func(body *session.TokenBody) {