From 2ec147ccfa2e67ee5dcf4d776ba1c2a770d97320 Mon Sep 17 00:00:00 2001 From: Marten Seemann Date: Thu, 1 Feb 2018 12:59:24 +0800 Subject: [PATCH] add a string representation for the perspective --- internal/protocol/perspective.go | 11 +++++++++++ internal/protocol/perspective_test.go | 14 ++++++++++++++ 2 files changed, 25 insertions(+) create mode 100644 internal/protocol/perspective_test.go diff --git a/internal/protocol/perspective.go b/internal/protocol/perspective.go index 6aa3b70c3..948e371ae 100644 --- a/internal/protocol/perspective.go +++ b/internal/protocol/perspective.go @@ -8,3 +8,14 @@ const ( PerspectiveServer Perspective = 1 PerspectiveClient Perspective = 2 ) + +func (p Perspective) String() string { + switch p { + case PerspectiveServer: + return "Server" + case PerspectiveClient: + return "Client" + default: + return "invalid perspective" + } +} diff --git a/internal/protocol/perspective_test.go b/internal/protocol/perspective_test.go new file mode 100644 index 000000000..55e477069 --- /dev/null +++ b/internal/protocol/perspective_test.go @@ -0,0 +1,14 @@ +package protocol + +import ( + . "github.com/onsi/ginkgo" + . "github.com/onsi/gomega" +) + +var _ = Describe("Perspective", func() { + It("has a string representation", func() { + Expect(PerspectiveClient.String()).To(Equal("Client")) + Expect(PerspectiveServer.String()).To(Equal("Server")) + Expect(Perspective(0).String()).To(Equal("invalid perspective")) + }) +})