From 20d644758be63ae4cc18c31daf686a34706313ac Mon Sep 17 00:00:00 2001 From: Leonard Lyubich Date: Thu, 10 Sep 2020 14:49:17 +0300 Subject: [PATCH] [#140] sdk/object: Define Attribute type Signed-off-by: Leonard Lyubich --- pkg/object/attribute.go | 45 ++++++++++++++++++++++++++++++++++++ pkg/object/attribute_test.go | 23 ++++++++++++++++++ 2 files changed, 68 insertions(+) create mode 100644 pkg/object/attribute.go create mode 100644 pkg/object/attribute_test.go diff --git a/pkg/object/attribute.go b/pkg/object/attribute.go new file mode 100644 index 0000000..0b43bcc --- /dev/null +++ b/pkg/object/attribute.go @@ -0,0 +1,45 @@ +package object + +import ( + "github.com/nspcc-dev/neofs-api-go/v2/object" +) + +// Attribute represents v2-compatible object attribute. +type Attribute object.Attribute + +// NewAttributeFromV2 wraps v2 Attribute message to Attribute. +func NewAttributeFromV2(aV2 *object.Attribute) *Attribute { + return (*Attribute)(aV2) +} + +// NewAttribute creates and initializes blank Attribute. +// +// Works similar as NewAttributeFromV2(new(Attribute)). +func NewAttribute() *Attribute { + return NewAttributeFromV2(new(object.Attribute)) +} + +// GetKey returns key to the object attribute. +func (a *Attribute) GetKey() string { + return (*object.Attribute)(a).GetKey() +} + +// SetKey sets key to the object attribute. +func (a *Attribute) SetKey(v string) { + (*object.Attribute)(a).SetKey(v) +} + +// GetValue return value of the object attribute. +func (a *Attribute) GetValue() string { + return (*object.Attribute)(a).GetValue() +} + +// SetValue sets value of the object attribute. +func (a *Attribute) SetValue(v string) { + (*object.Attribute)(a).SetValue(v) +} + +// ToV2 converts Attribute to v2 Attribute message. +func (a *Attribute) ToV2() *object.Attribute { + return (*object.Attribute)(a) +} diff --git a/pkg/object/attribute_test.go b/pkg/object/attribute_test.go new file mode 100644 index 0000000..cc503e4 --- /dev/null +++ b/pkg/object/attribute_test.go @@ -0,0 +1,23 @@ +package object + +import ( + "testing" + + "github.com/stretchr/testify/require" +) + +func TestAttribute(t *testing.T) { + key, val := "some key", "some value" + + a := NewAttribute() + a.SetKey(key) + a.SetValue(val) + + require.Equal(t, key, a.GetKey()) + require.Equal(t, val, a.GetValue()) + + aV2 := a.ToV2() + + require.Equal(t, key, aV2.GetKey()) + require.Equal(t, val, aV2.GetValue()) +}