From 51c38a10c80e1fb9d6e6232b4007267234a550e9 Mon Sep 17 00:00:00 2001
From: Joshua MacVey <15506741+jmacvey@users.noreply.github.com>
Date: Sat, 27 Aug 2022 01:37:50 -0500
Subject: [PATCH] feat: add GetSchemaInfo to registry client (#184)

Co-authored-by: Joshua MacVey <joshua.macvey@traivefinance.com>
---
 registry/client.go      | 15 ++++++++++++
 registry/client_test.go | 54 +++++++++++++++++++++++++++++++++++++++++
 2 files changed, 69 insertions(+)

diff --git a/registry/client.go b/registry/client.go
index e5c1299c..d6c24bfb 100644
--- a/registry/client.go
+++ b/registry/client.go
@@ -45,6 +45,9 @@ type Registry interface {
 	// GetLatestSchema gets the latest schema for a subject.
 	GetLatestSchema(ctx context.Context, subject string) (avro.Schema, error)
 
+	// GetSchemaInfo gets the schema and schema metadata for a subject and version.
+	GetSchemaInfo(ctx context.Context, subject string, version int) (SchemaInfo, error)
+
 	// GetLatestSchemaInfo gets the latest schema and schema metadata for a subject.
 	GetLatestSchemaInfo(ctx context.Context, subject string) (SchemaInfo, error)
 
@@ -232,6 +235,18 @@ func (c *Client) GetLatestSchema(ctx context.Context, subject string) (avro.Sche
 	return avro.Parse(payload.Schema)
 }
 
+// GetSchemaInfo gets the schema and schema metadata for a subject and version.
+func (c *Client) GetSchemaInfo(ctx context.Context, subject string, version int) (SchemaInfo, error) {
+	var payload schemaInfoPayload
+	p := path.Join("subjects", subject, "versions", strconv.Itoa(version))
+	err := c.request(ctx, http.MethodGet, p, nil, &payload)
+	if err != nil {
+		return SchemaInfo{}, err
+	}
+
+	return payload.Parse()
+}
+
 // GetLatestSchemaInfo gets the latest schema and schema metadata for a subject.
 func (c *Client) GetLatestSchemaInfo(ctx context.Context, subject string) (SchemaInfo, error) {
 	var payload schemaInfoPayload
diff --git a/registry/client_test.go b/registry/client_test.go
index de91daba..593f5081 100644
--- a/registry/client_test.go
+++ b/registry/client_test.go
@@ -315,6 +315,60 @@ func TestClient_GetLatestSchemaSchemaError(t *testing.T) {
 	assert.Error(t, err)
 }
 
+func TestClient_GetSchemaInfo(t *testing.T) {
+	s := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		assert.Equal(t, "GET", r.Method)
+		assert.Equal(t, "/subjects/foobar/versions/1", r.URL.Path)
+
+		_, _ = w.Write([]byte(`{"subject": "foobar", "version": 1, "id": 2, "schema":"[\"null\",\"string\",\"int\"]"}`))
+	}))
+	defer s.Close()
+	client, _ := registry.NewClient(s.URL)
+
+	schemaInfo, err := client.GetSchemaInfo(context.Background(), "foobar", 1)
+
+	require.NoError(t, err)
+	assert.Equal(t, `["null","string","int"]`, schemaInfo.Schema.String())
+	assert.Equal(t, 2, schemaInfo.ID)
+	assert.Equal(t, 1, schemaInfo.Version)
+}
+
+func TestClient_GetSchemaInfoRequestError(t *testing.T) {
+	s := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.WriteHeader(500)
+	}))
+	defer s.Close()
+	client, _ := registry.NewClient(s.URL)
+
+	_, err := client.GetSchemaInfo(context.Background(), "foobar", 1)
+
+	assert.Error(t, err)
+}
+
+func TestClient_GetSchemaInfoJsonError(t *testing.T) {
+	s := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		_, _ = w.Write([]byte(`{"subject": "foobar", "version": 1, "id": 2, "schema":"[\"null\",\"string\",\"int\"]"`))
+	}))
+	defer s.Close()
+	client, _ := registry.NewClient(s.URL)
+
+	_, err := client.GetSchemaInfo(context.Background(), "foobar", 1)
+
+	assert.Error(t, err)
+}
+
+func TestClient_GetSchemaInfoSchemaError(t *testing.T) {
+	s := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		_, _ = w.Write([]byte(`{"schema":""}`))
+	}))
+	defer s.Close()
+	client, _ := registry.NewClient(s.URL)
+
+	_, err := client.GetSchemaInfo(context.Background(), "foobar", 1)
+
+	assert.Error(t, err)
+}
+
 func TestClient_GetLatestSchemaInfo(t *testing.T) {
 	s := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		assert.Equal(t, "GET", r.Method)