Skip to content

Commit

Permalink
Fix gl-testing
Browse files Browse the repository at this point in the history 
I've broken the `Makefile` in a previous commit.
After this commit

```
make gen
```

works again
  • Loading branch information
@ErikDeSmedt
ErikDeSmedt committed Mar 4, 2024
1 parent ba4e106 commit aa386f9
Show file tree
Hide file tree
Showing 3 changed files with 301 additions and 4 deletions.
4 changes: 2 additions & 2 deletions libs/gl-testing/Makefile
View file
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ GEN_TESTING = \

GENALL += ${GEN_TESTING}

testgrpc ${TESTINGDIR}/gltesting/scheduler_grpc.py: ${REPO}/libs/proto/scheduler.proto
python -m grpc_tools.protoc ${TESTPROTOC_OPTS} ${REPO}/libs/proto/scheduler.proto
testgrpc ${TESTINGDIR}/gltesting/scheduler_grpc.py: ${REPO}/libs/proto/glclient/scheduler.proto
python -m grpc_tools.protoc ${TESTPROTOC_OPTS} ${REPO}/libs/proto/glclient/scheduler.proto
sed -E -i 's/import (.+)_pb2 as .+__pb2/from glclient import \1_pb2 as \1__pb2/g' ${TESTINGDIR}/gltesting/scheduler_grpc.py

297 changes: 297 additions & 0 deletions libs/gl-testing/gltesting/glclient/scheduler_grpc.py
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,297 @@
import purerpc
import glclient.scheduler_pb2 as glclient_dot_scheduler__pb2
import glclient.greenlight_pb2 as glclient_dot_greenlight__pb2


class SchedulerServicer(purerpc.Servicer):
async def Register(self, input_message):
raise NotImplementedError()

async def Recover(self, input_message):
raise NotImplementedError()

async def GetChallenge(self, input_message):
raise NotImplementedError()

async def Schedule(self, input_message):
raise NotImplementedError()

async def GetNodeInfo(self, input_message):
raise NotImplementedError()

async def MaybeUpgrade(self, input_message):
raise NotImplementedError()

async def ListInviteCodes(self, input_message):
raise NotImplementedError()

async def ExportNode(self, input_message):
raise NotImplementedError()

async def AddOutgoingWebhook(self, input_message):
raise NotImplementedError()

async def ListOutgoingWebhooks(self, input_message):
raise NotImplementedError()

async def DeleteWebhooks(self, input_message):
raise NotImplementedError()

async def RotateOutgoingWebhookSecret(self, input_message):
raise NotImplementedError()

@property
def service(self) -> purerpc.Service:
service_obj = purerpc.Service(
"scheduler.Scheduler"
)
service_obj.add_method(
"Register",
self.Register,
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.RegistrationRequest,
glclient_dot_scheduler__pb2.RegistrationResponse,
)
)
service_obj.add_method(
"Recover",
self.Recover,
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.RecoveryRequest,
glclient_dot_scheduler__pb2.RecoveryResponse,
)
)
service_obj.add_method(
"GetChallenge",
self.GetChallenge,
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.ChallengeRequest,
glclient_dot_scheduler__pb2.ChallengeResponse,
)
)
service_obj.add_method(
"Schedule",
self.Schedule,
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.ScheduleRequest,
glclient_dot_scheduler__pb2.NodeInfoResponse,
)
)
service_obj.add_method(
"GetNodeInfo",
self.GetNodeInfo,
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.NodeInfoRequest,
glclient_dot_scheduler__pb2.NodeInfoResponse,
)
)
service_obj.add_method(
"MaybeUpgrade",
self.MaybeUpgrade,
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.UpgradeRequest,
glclient_dot_scheduler__pb2.UpgradeResponse,
)
)
service_obj.add_method(
"ListInviteCodes",
self.ListInviteCodes,
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.ListInviteCodesRequest,
glclient_dot_scheduler__pb2.ListInviteCodesResponse,
)
)
service_obj.add_method(
"ExportNode",
self.ExportNode,
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.ExportNodeRequest,
glclient_dot_scheduler__pb2.ExportNodeResponse,
)
)
service_obj.add_method(
"AddOutgoingWebhook",
self.AddOutgoingWebhook,
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.AddOutgoingWebhookRequest,
glclient_dot_scheduler__pb2.AddOutgoingWebhookResponse,
)
)
service_obj.add_method(
"ListOutgoingWebhooks",
self.ListOutgoingWebhooks,
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.ListOutgoingWebhooksRequest,
glclient_dot_scheduler__pb2.ListOutgoingWebhooksResponse,
)
)
service_obj.add_method(
"DeleteWebhooks",
self.DeleteWebhooks,
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.DeleteOutgoingWebhooksRequest,
glclient_dot_greenlight__pb2.Empty,
)
)
service_obj.add_method(
"RotateOutgoingWebhookSecret",
self.RotateOutgoingWebhookSecret,
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.RotateOutgoingWebhookSecretRequest,
glclient_dot_scheduler__pb2.WebhookSecretResponse,
)
)
return service_obj


class SchedulerStub:
def __init__(self, channel):
self._client = purerpc.Client(
"scheduler.Scheduler",
channel
)
self.Register = self._client.get_method_stub(
"Register",
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.RegistrationRequest,
glclient_dot_scheduler__pb2.RegistrationResponse,
)
)
self.Recover = self._client.get_method_stub(
"Recover",
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.RecoveryRequest,
glclient_dot_scheduler__pb2.RecoveryResponse,
)
)
self.GetChallenge = self._client.get_method_stub(
"GetChallenge",
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.ChallengeRequest,
glclient_dot_scheduler__pb2.ChallengeResponse,
)
)
self.Schedule = self._client.get_method_stub(
"Schedule",
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.ScheduleRequest,
glclient_dot_scheduler__pb2.NodeInfoResponse,
)
)
self.GetNodeInfo = self._client.get_method_stub(
"GetNodeInfo",
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.NodeInfoRequest,
glclient_dot_scheduler__pb2.NodeInfoResponse,
)
)
self.MaybeUpgrade = self._client.get_method_stub(
"MaybeUpgrade",
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.UpgradeRequest,
glclient_dot_scheduler__pb2.UpgradeResponse,
)
)
self.ListInviteCodes = self._client.get_method_stub(
"ListInviteCodes",
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.ListInviteCodesRequest,
glclient_dot_scheduler__pb2.ListInviteCodesResponse,
)
)
self.ExportNode = self._client.get_method_stub(
"ExportNode",
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.ExportNodeRequest,
glclient_dot_scheduler__pb2.ExportNodeResponse,
)
)
self.AddOutgoingWebhook = self._client.get_method_stub(
"AddOutgoingWebhook",
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.AddOutgoingWebhookRequest,
glclient_dot_scheduler__pb2.AddOutgoingWebhookResponse,
)
)
self.ListOutgoingWebhooks = self._client.get_method_stub(
"ListOutgoingWebhooks",
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.ListOutgoingWebhooksRequest,
glclient_dot_scheduler__pb2.ListOutgoingWebhooksResponse,
)
)
self.DeleteWebhooks = self._client.get_method_stub(
"DeleteWebhooks",
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.DeleteOutgoingWebhooksRequest,
glclient_dot_greenlight__pb2.Empty,
)
)
self.RotateOutgoingWebhookSecret = self._client.get_method_stub(
"RotateOutgoingWebhookSecret",
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.RotateOutgoingWebhookSecretRequest,
glclient_dot_scheduler__pb2.WebhookSecretResponse,
)
)


class DebugServicer(purerpc.Servicer):
async def ReportSignerRejection(self, input_message):
raise NotImplementedError()

@property
def service(self) -> purerpc.Service:
service_obj = purerpc.Service(
"scheduler.Debug"
)
service_obj.add_method(
"ReportSignerRejection",
self.ReportSignerRejection,
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.SignerRejection,
glclient_dot_greenlight__pb2.Empty,
)
)
return service_obj


class DebugStub:
def __init__(self, channel):
self._client = purerpc.Client(
"scheduler.Debug",
channel
)
self.ReportSignerRejection = self._client.get_method_stub(
"ReportSignerRejection",
purerpc.RPCSignature(
purerpc.Cardinality.UNARY_UNARY,
glclient_dot_scheduler__pb2.SignerRejection,
glclient_dot_greenlight__pb2.Empty,
)
)
4 changes: 2 additions & 2 deletions libs/gl-testing/gltesting/scheduler_grpc.py
View file
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import purerpc
from glclient import scheduler_pb2 as scheduler__pb2
from glclient import greenlight_pb2 as greenlight__pb2
from glclient from glclient import scheduler_pb2 as scheduler__pb2
from glclient from glclient import greenlight_pb2 as greenlight__pb2


class SchedulerServicer(purerpc.Servicer):
Expand Down

0 comments on commit aa386f9

Please sign in to comment.