1 files changed, 5 insertions, 21 deletions
diff --git a/pkg/rpc/server_test.go b/pkg/rpc/server_test.go
index da60f86a..fd6e6237 100644
--- a/pkg/rpc/server_test.go
+++ b/pkg/rpc/server_test.go
@@ -1,35 +1,19 @@
 package rpc
 
 import (
-	"net"
+	http "net/http"
+	"net/http/httptest"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
-	grpc "google.golang.org/grpc"
-	"google.golang.org/grpc/credentials/insecure"
 )
 
 func TestServer(t *testing.T) {
-	listener, err := net.Listen("tcp", "localhost:0")
-	require.NoError(t, err)
-	defer listener.Close()
+	srv := httptest.NewServer(New())
+	defer srv.Close()
 
-	server := New()
-	defer server.Stop()
-
-	go func() {
-		require.NoError(t, server.Serve(listener))
-	}()
-
-	connection, err := grpc.NewClient(
-		listener.Addr().String(),
-		grpc.WithTransportCredentials(insecure.NewCredentials()),
-	)
-	require.NoError(t, err)
-
-	defer connection.Close()
-	client := NewAbilityClient(connection)
+	client := NewAbilityProtobufClient(srv.URL, &http.Client{})
 
 	t.Run("forbids", func(t *testing.T) {
 		reply, err := client.Allowed(t.Context(), &AllowRequest{