Implement ping pong test case

restatedev · Jul 31, 2024 · e85bcaa · e85bcaa
1 parent e04e3c3
commit e85bcaa
Show file tree

Hide file tree

Showing 4 changed files with 43 additions and 4 deletions.
diff --git a/test-services/coordinator.go b/test-services/coordinator.go
@@ -0,0 +1,27 @@
+package main
+
+import (
+	restate "github.com/restatedev/sdk-go"
+)
+
+func RegisterCoordinator() {
+	REGISTRY.AddRouter(
+		restate.NewServiceRouter("Coordinator").
+			Handler("proxy", restate.NewServiceHandler(
+				func(ctx restate.Context, _ restate.Void) (string, error) {
+					key := ctx.Rand().UUID().String()
+					return restate.CallAs[string](ctx.Object("Receiver", key, "ping")).Request(nil)
+				})))
+}
+
+type coordinator struct {
+}
+
+func (s coordinator) ServiceName() string {
+	return "Coordinator"
+}
+
+func (s coordinator) Proxy(ctx restate.Context, _ restate.Void) (string, error) {
+	key := ctx.Rand().UUID().String()
+	return restate.CallAs[string](ctx.Object("Receiver", key, "ping")).Request(nil)
+}
diff --git a/test-services/exclusions.yaml b/test-services/exclusions.yaml
@@ -13,7 +13,6 @@ exclusions:
   - "dev.restate.sdktesting.tests.WorkflowAPI"
   - "dev.restate.sdktesting.tests.UserErrors"
   - "dev.restate.sdktesting.tests.State"
-  - "dev.restate.sdktesting.tests.ServiceToServiceCommunication"
   - "dev.restate.sdktesting.tests.Sleep"
   - "dev.restate.sdktesting.tests.RequestSigning"
   - "dev.restate.sdktesting.tests.CancelInvocation"
@@ -27,7 +26,6 @@ exclusions:
   - "dev.restate.sdktesting.tests.WorkflowAPI"
   - "dev.restate.sdktesting.tests.UserErrors"
   - "dev.restate.sdktesting.tests.State"
-  - "dev.restate.sdktesting.tests.ServiceToServiceCommunication"
   - "dev.restate.sdktesting.tests.Sleep"
   "singleThreadSinglePartition":
   - "dev.restate.sdktesting.tests.KafkaIngressTest"
@@ -43,7 +41,6 @@ exclusions:
   - "dev.restate.sdktesting.tests.WorkflowAPI"
   - "dev.restate.sdktesting.tests.UserErrors"
   - "dev.restate.sdktesting.tests.State"
-  - "dev.restate.sdktesting.tests.ServiceToServiceCommunication"
   - "dev.restate.sdktesting.tests.Sleep"
   - "dev.restate.sdktesting.tests.RequestSigning"
   - "dev.restate.sdktesting.tests.CancelInvocation"

diff --git a/test-services/main.go b/test-services/main.go
@@ -10,7 +10,8 @@ import (
 )
 
 func init() {
-	// TODO register services
+	RegisterCoordinator()
+	RegisterReceiver()
 }
 
 func main() {

diff --git a/test-services/receiver.go b/test-services/receiver.go
@@ -0,0 +1,14 @@
+package main
+
+import (
+	restate "github.com/restatedev/sdk-go"
+)
+
+func RegisterReceiver() {
+	REGISTRY.AddRouter(
+		restate.NewObjectRouter("Receiver").
+			Handler("ping", restate.NewObjectHandler(
+				func(ctx restate.ObjectContext, _ restate.Void) (string, error) {
+					return "pong", nil
+				})))
+}