diff --git a/proto-public/pbdns/mock_DNSServiceClient.go b/proto-public/pbdns/mock_DNSServiceClient.go
new file mode 100644
index 000000000..a11f1e963
--- /dev/null
+++ b/proto-public/pbdns/mock_DNSServiceClient.go
@@ -0,0 +1,58 @@
+// Code generated by mockery v2.12.2. DO NOT EDIT.
+
+package pbdns
+
+import (
+	context "context"
+
+	grpc "google.golang.org/grpc"
+
+	mock "github.com/stretchr/testify/mock"
+
+	testing "testing"
+)
+
+// MockDNSServiceClient is an autogenerated mock type for the DNSServiceClient type
+type MockDNSServiceClient struct {
+	mock.Mock
+}
+
+// Query provides a mock function with given fields: ctx, in, opts
+func (_m *MockDNSServiceClient) Query(ctx context.Context, in *QueryRequest, opts ...grpc.CallOption) (*QueryResponse, error) {
+	_va := make([]interface{}, len(opts))
+	for _i := range opts {
+		_va[_i] = opts[_i]
+	}
+	var _ca []interface{}
+	_ca = append(_ca, ctx, in)
+	_ca = append(_ca, _va...)
+	ret := _m.Called(_ca...)
+
+	var r0 *QueryResponse
+	if rf, ok := ret.Get(0).(func(context.Context, *QueryRequest, ...grpc.CallOption) *QueryResponse); ok {
+		r0 = rf(ctx, in, opts...)
+	} else {
+		if ret.Get(0) != nil {
+			r0 = ret.Get(0).(*QueryResponse)
+		}
+	}
+
+	var r1 error
+	if rf, ok := ret.Get(1).(func(context.Context, *QueryRequest, ...grpc.CallOption) error); ok {
+		r1 = rf(ctx, in, opts...)
+	} else {
+		r1 = ret.Error(1)
+	}
+
+	return r0, r1
+}
+
+// NewMockDNSServiceClient creates a new instance of MockDNSServiceClient. It also registers the testing.TB interface on the mock and a cleanup function to assert the mocks expectations.
+func NewMockDNSServiceClient(t testing.TB) *MockDNSServiceClient {
+	mock := &MockDNSServiceClient{}
+	mock.Mock.Test(t)
+
+	t.Cleanup(func() { mock.AssertExpectations(t) })
+
+	return mock
+}
diff --git a/proto-public/pbdns/mock_DNSServiceServer.go b/proto-public/pbdns/mock_DNSServiceServer.go
new file mode 100644
index 000000000..97b98dddb
--- /dev/null
+++ b/proto-public/pbdns/mock_DNSServiceServer.go
@@ -0,0 +1,48 @@
+// Code generated by mockery v2.12.2. DO NOT EDIT.
+
+package pbdns
+
+import (
+	context "context"
+	testing "testing"
+
+	mock "github.com/stretchr/testify/mock"
+)
+
+// MockDNSServiceServer is an autogenerated mock type for the DNSServiceServer type
+type MockDNSServiceServer struct {
+	mock.Mock
+}
+
+// Query provides a mock function with given fields: _a0, _a1
+func (_m *MockDNSServiceServer) Query(_a0 context.Context, _a1 *QueryRequest) (*QueryResponse, error) {
+	ret := _m.Called(_a0, _a1)
+
+	var r0 *QueryResponse
+	if rf, ok := ret.Get(0).(func(context.Context, *QueryRequest) *QueryResponse); ok {
+		r0 = rf(_a0, _a1)
+	} else {
+		if ret.Get(0) != nil {
+			r0 = ret.Get(0).(*QueryResponse)
+		}
+	}
+
+	var r1 error
+	if rf, ok := ret.Get(1).(func(context.Context, *QueryRequest) error); ok {
+		r1 = rf(_a0, _a1)
+	} else {
+		r1 = ret.Error(1)
+	}
+
+	return r0, r1
+}
+
+// NewMockDNSServiceServer creates a new instance of MockDNSServiceServer. It also registers the testing.TB interface on the mock and a cleanup function to assert the mocks expectations.
+func NewMockDNSServiceServer(t testing.TB) *MockDNSServiceServer {
+	mock := &MockDNSServiceServer{}
+	mock.Mock.Test(t)
+
+	t.Cleanup(func() { mock.AssertExpectations(t) })
+
+	return mock
+}
diff --git a/proto-public/pbdns/mock_UnsafeDNSServiceServer.go b/proto-public/pbdns/mock_UnsafeDNSServiceServer.go
new file mode 100644
index 000000000..a56e55bcb
--- /dev/null
+++ b/proto-public/pbdns/mock_UnsafeDNSServiceServer.go
@@ -0,0 +1,29 @@
+// Code generated by mockery v2.12.2. DO NOT EDIT.
+
+package pbdns
+
+import (
+	testing "testing"
+
+	mock "github.com/stretchr/testify/mock"
+)
+
+// MockUnsafeDNSServiceServer is an autogenerated mock type for the UnsafeDNSServiceServer type
+type MockUnsafeDNSServiceServer struct {
+	mock.Mock
+}
+
+// mustEmbedUnimplementedDNSServiceServer provides a mock function with given fields:
+func (_m *MockUnsafeDNSServiceServer) mustEmbedUnimplementedDNSServiceServer() {
+	_m.Called()
+}
+
+// NewMockUnsafeDNSServiceServer creates a new instance of MockUnsafeDNSServiceServer. It also registers the testing.TB interface on the mock and a cleanup function to assert the mocks expectations.
+func NewMockUnsafeDNSServiceServer(t testing.TB) *MockUnsafeDNSServiceServer {
+	mock := &MockUnsafeDNSServiceServer{}
+	mock.Mock.Test(t)
+
+	t.Cleanup(func() { mock.AssertExpectations(t) })
+
+	return mock
+}