@@ -973,8 +973,32 @@ class MockAPIPlugin<T> extends APIPluginInterface {
973973class MockNativeAuthBridgeCodec extends StandardMessageCodec {
974974 @override
975975 void writeValue (WriteBuffer buffer, Object ? value) {
976- if (value is NativeGraphQLSubscriptionResponse ) {
977- buffer.putUint8 (128 );
976+ if (value is int ) {
977+ buffer.putUint8 (4 );
978+ buffer.putInt64 (value);
979+ } else if (value is NativeAuthSession ) {
980+ buffer.putUint8 (129 );
981+ writeValue (buffer, value.encode ());
982+ } else if (value is NativeAuthUser ) {
983+ buffer.putUint8 (130 );
984+ writeValue (buffer, value.encode ());
985+ } else if (value is NativeUserPoolTokens ) {
986+ buffer.putUint8 (131 );
987+ writeValue (buffer, value.encode ());
988+ } else if (value is NativeAWSCredentials ) {
989+ buffer.putUint8 (132 );
990+ writeValue (buffer, value.encode ());
991+ } else if (value is LegacyCredentialStoreData ) {
992+ buffer.putUint8 (133 );
993+ writeValue (buffer, value.encode ());
994+ } else if (value is NativeGraphQLResponse ) {
995+ buffer.putUint8 (134 );
996+ writeValue (buffer, value.encode ());
997+ } else if (value is NativeGraphQLSubscriptionResponse ) {
998+ buffer.putUint8 (135 );
999+ writeValue (buffer, value.encode ());
1000+ } else if (value is NativeGraphQLRequest ) {
1001+ buffer.putUint8 (136 );
9781002 writeValue (buffer, value.encode ());
9791003 } else {
9801004 super .writeValue (buffer, value);
@@ -984,8 +1008,22 @@ class MockNativeAuthBridgeCodec extends StandardMessageCodec {
9841008 @override
9851009 Object ? readValueOfType (int type, ReadBuffer buffer) {
9861010 switch (type) {
987- case 128 :
1011+ case 129 :
1012+ return NativeAuthSession .decode (readValue (buffer)! );
1013+ case 130 :
1014+ return NativeAuthUser .decode (readValue (buffer)! );
1015+ case 131 :
1016+ return NativeUserPoolTokens .decode (readValue (buffer)! );
1017+ case 132 :
1018+ return NativeAWSCredentials .decode (readValue (buffer)! );
1019+ case 133 :
1020+ return LegacyCredentialStoreData .decode (readValue (buffer)! );
1021+ case 134 :
1022+ return NativeGraphQLResponse .decode (readValue (buffer)! );
1023+ case 135 :
9881024 return NativeGraphQLSubscriptionResponse .decode (readValue (buffer)! );
1025+ case 136 :
1026+ return NativeGraphQLRequest .decode (readValue (buffer)! );
9891027 default :
9901028 return super .readValueOfType (type, buffer);
9911029 }
0 commit comments