Merge branch 'main' into ebracci-protocol-original-string

hamba · Oct 20, 2023 · 3c6fbb7 · 3c6fbb7
2 parents c192f00 + 50a7897
commit 3c6fbb7
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/protocol.go b/protocol.go
@@ -106,6 +106,11 @@ func (p *Protocol) OriginalString() string {
 	return p.originalString
 }
 
+// Types returns the types of the protocol.
+func (p *Protocol) Types() []NamedSchema {
+	return p.types
+}
+
 // String returns the canonical form of the protocol.
 func (p *Protocol) String() string {
 	types := ""

diff --git a/protocol_test.go b/protocol_test.go
@@ -199,3 +199,17 @@ func TestParseProtocolFile_InvalidPath(t *testing.T) {
 
 	assert.Error(t, err)
 }
+
+func TestParseProtocol_Types(t *testing.T) {
+	protocol, err := avro.ParseProtocolFile("testdata/echo.avpr")
+
+	wantPing := `{"name":"org.hamba.avro.Ping","type":"record","fields":[{"name":"timestamp","type":"long"},{"name":"text","type":"string"}]}`
+	wantPong := `{"name":"org.hamba.avro.Pong","type":"record","fields":[{"name":"timestamp","type":"long"},{"name":"ping","type":"org.hamba.avro.Ping"}]}`
+	wantPongError := `{"name":"org.hamba.avro.PongError","type":"error","fields":[{"name":"timestamp","type":"long"},{"name":"reason","type":"string"}]}`
+	wantLen := 3
+	require.NoError(t, err)
+	assert.Equal(t, wantLen, len(protocol.Types()))
+	assert.Equal(t, wantPing, protocol.Types()[0].String())
+	assert.Equal(t, wantPong, protocol.Types()[1].String())
+	assert.Equal(t, wantPongError, protocol.Types()[2].String())
+}