Merge pull request #207 from RedisLabs/feat/get-tls-certificate-for-database

Get TLS certificate for database

Author: burythehammer

CHANGELOG.md

@@ -2,6 +2,12 @@
 All notable changes to this project will be documented in this file.
 See updating [Changelog example here](https://keepachangelog.com/en/1.0.0/).
 
+## 0.28.0
+
+### Added
+
+* Adding an endpoint to get a TLS certificate for a database.
+
 ## 0.27.0
 
 ### Updated

database_test.go

@@ -543,3 +543,19 @@ func TestDatabase_Import(t *testing.T) {
 	})
 	require.NoError(t, err)
 }
+
+func TestDatabase_Certificate(t *testing.T) {
+	s := httptest.NewServer(testServer("key", "secret", getRequest(t, "/subscriptions/42/databases/18/certificate",
+		`{ "publicCertificatePEMString": "public-cert" }`)))
+
+	subject, err := clientFromTestServer(s, "key", "secret")
+	require.NoError(t, err)
+
+	certificate, err := subject.Database.GetCertificate(context.TODO(), 42, 18)
+	require.NoError(t, err)
+
+	assert.Equal(t, &databases.DatabaseCertificate{
+		PublicCertificatePEMString: "public-cert",
+	}, certificate)
+
+}

service/databases/model.go

@@ -200,6 +200,15 @@ func (o Import) String() string {
 	return internal.ToString(o)
 }
 
+// DatabaseCertificate represents the TLS certificate information for a database
+type DatabaseCertificate struct {
+	PublicCertificatePEMString string `json:"publicCertificatePEMString"`
+}
+
+func (o DatabaseCertificate) String() string {
+	return internal.ToString(o)
+}
+
 type listDatabaseResponse struct {
 	Subscription []*listDbSubscription `json:"subscription,omitempty"`
 }

service/databases/service.go

@@ -56,7 +56,7 @@ func (a *API) Create(ctx context.Context, subscription int, db CreateDatabase) (
 	return id, nil
 }
 
-// List will return a ListDatabase that is capable of paging through all of the databases associated with a
+// List will return a ListDatabase that is capable of paging through all the databases associated with a
 // subscription.
 func (a *API) List(ctx context.Context, subscription int) *ListDatabase {
 	return newListDatabase(ctx, a.client, subscription, 100)
@@ -125,6 +125,24 @@ func (a *API) Import(ctx context.Context, subscription int, database int, reques
 	return a.taskWaiter.Wait(ctx, *task.ID)
 }
 
+// GetCertificate retrieves the TLS certificate for the specified database within a subscription.
+func (a *API) GetCertificate(ctx context.Context, subscription int, database int) (*DatabaseCertificate, error) {
+	var certificate DatabaseCertificate
+	getCertificateUrl := "/subscriptions/%d/databases/%d/certificate"
+
+	path := fmt.Sprintf(getCertificateUrl, subscription, database)
+	err := a.client.Get(
+		ctx,
+		fmt.Sprintf("get TLS certificate for database %d in subscription %d", database, subscription),
+		path,
+		&certificate,
+	)
+	if err != nil {
+		return nil, wrap404Error(subscription, database, err)
+	}
+	return &certificate, nil
+}
+
 type ListDatabase struct {
 	client       HttpClient
 	subscription int

service/databases/service_test.go

@@ -85,6 +85,39 @@ func TestListDatabase_recordsError(t *testing.T) {
 	assert.Nil(t, subject.Value())
 }
 
+func TestGetCertificate_success(t *testing.T) {
+	client := &mockHttpClient{}
+	api := NewAPI(client, nil, nil)
+
+	expected := &DatabaseCertificate{
+		PublicCertificatePEMString: "test-certificate",
+	}
+
+	client.On("Get", context.TODO(), "get TLS certificate for database 123 in subscription 456", "/subscriptions/456/databases/123/certificate", mock.AnythingOfType("*databases.DatabaseCertificate")).
+		Run(func(args mock.Arguments) {
+			cert := args.Get(3).(*DatabaseCertificate)
+			cert.PublicCertificatePEMString = "test-certificate"
+		}).Return(nil)
+
+	result, err := api.GetCertificate(context.TODO(), 456, 123)
+	assert.NoError(t, err)
+	assert.Equal(t, expected, result)
+}
+
+func TestGetCertificate_error(t *testing.T) {
+	client := &mockHttpClient{}
+	api := NewAPI(client, nil, nil)
+
+	expected := fmt.Errorf("test error")
+	client.On("Get", context.TODO(), "get TLS certificate for database 123 in subscription 456", "/subscriptions/456/databases/123/certificate", mock.AnythingOfType("*databases.DatabaseCertificate")).
+		Return(expected)
+
+	result, err := api.GetCertificate(context.TODO(), 456, 123)
+	assert.Error(t, err)
+	assert.Equal(t, expected, err)
+	assert.Nil(t, result)
+}
+
 type mockHttpClient struct {
 	mock.Mock
 }