Skip to content

Commit dfaf471

Browse files
committed
chore(): Updated Pigeon codec for unit tests
1 parent 17a2181 commit dfaf471

File tree

1 file changed

+41
-3
lines changed

1 file changed

+41
-3
lines changed

packages/amplify_datastore/test/native_amplify_api_test.dart

Lines changed: 41 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -973,8 +973,32 @@ class MockAPIPlugin<T> extends APIPluginInterface {
973973
class 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

Comments
 (0)