From a21e42ea4fed167b90e59d2aa8b9b748f30e0dc6 Mon Sep 17 00:00:00 2001 From: Leonard Lyubich Date: Fri, 14 Feb 2020 12:14:19 +0300 Subject: [PATCH] Implement Epoch and Version setters on ResponseMetaHeader --- service/meta.go | 6 ++++++ service/meta_test.go | 14 ++++++++++++++ 2 files changed, 20 insertions(+) diff --git a/service/meta.go b/service/meta.go index 95c1b13a..dc2cdd36 100644 --- a/service/meta.go +++ b/service/meta.go @@ -62,6 +62,12 @@ const ( ErrIncorrectTTL = internal.Error("incorrect ttl") ) +// SetVersion sets protocol version to ResponseMetaHeader. +func (m *ResponseMetaHeader) SetVersion(v uint32) { m.Version = v } + +// SetEpoch sets Epoch to ResponseMetaHeader. +func (m *ResponseMetaHeader) SetEpoch(v uint64) { m.Epoch = v } + // SetVersion sets protocol version to RequestMetaHeader. func (m *RequestMetaHeader) SetVersion(v uint32) { m.Version = v } diff --git a/service/meta_test.go b/service/meta_test.go index e208dfe1..083ccd66 100644 --- a/service/meta_test.go +++ b/service/meta_test.go @@ -88,3 +88,17 @@ func TestMetaRequest(t *testing.T) { }) } } + +func TestRequestMetaHeader_SetEpoch(t *testing.T) { + m := new(ResponseMetaHeader) + epoch := uint64(3) + m.SetEpoch(epoch) + require.Equal(t, epoch, m.GetEpoch()) +} + +func TestRequestMetaHeader_SetVersion(t *testing.T) { + m := new(ResponseMetaHeader) + version := uint32(3) + m.SetVersion(version) + require.Equal(t, version, m.GetVersion()) +}