fix actor reference marshalling test

richardartoul · May 20, 2023 · 25cd9c8 · 25cd9c8
1 parent 2c50d90
commit 25cd9c8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/virtual/types/ref.go b/virtual/types/ref.go
@@ -35,6 +35,7 @@ func NewActorReference(
 			ServerVersion: serverVersion,
 			ServerState:   serverState,
 		},
+		Type: ReferenceTypeLocal,
 	}, nil
 }
 
@@ -55,7 +56,7 @@ func NewActorReferenceFromJSON(data []byte) (ActorReference, error) {
 // ActorReference abstracts over different forms of ReferenceType. It provides all the
 // necessary information for communicating with an actor. Some of the fields are "logical"
 type ActorReference struct {
-	Type     ReferenceType          `json:"type"`
+	Type     ReferenceType          `json:"-"`
 	Virtual  ActorReferenceVirtual  `json:"virtual"`
 	Physical ActorReferencePhysical `json:"physical"`
 }

diff --git a/virtual/types/ref_test.go b/virtual/types/ref_test.go
@@ -12,7 +12,7 @@ func TestNewActorReference(t *testing.T) {
 	require.NoError(t, err)
 
 	require.Equal(t, "server1", ref.Physical.ServerID)
-	require.Equal(t, "server1path", ref.Physical.ServerID)
+	require.Equal(t, "server1path", ref.Physical.ServerState.Address)
 	require.Equal(t, "a", ref.Virtual.Namespace)
 	require.Equal(t, "c", ref.Virtual.ActorID)
 	require.Equal(t, "b", ref.Virtual.ModuleID)