Add support for throwing exceptions on get(), update() and delete()

lib/src/mock_document_reference.dart

@@ -96,6 +96,7 @@ class MockDocumentReference<T extends Object?> implements DocumentReference<T> {
   @override
   Future<void> update(Map<Object, Object?> data) async {
     await _firestore.maybeThrowSecurityException(path, Method.update);
+    maybeThrowException(this, Invocation.method(#update, [data]));
     if (!_exists()) {
       return Future.error(FirebaseException(
         plugin: 'FakeFirestore',
@@ -215,6 +216,7 @@ class MockDocumentReference<T extends Object?> implements DocumentReference<T> {
   @override
   Future<DocumentSnapshot<T>> get([GetOptions? options]) async {
     await _firestore.maybeThrowSecurityException(path, Method.read);
+    maybeThrowException(this, Invocation.method(#get, [options]));
     return _getSync(options);
   }
 
@@ -268,6 +270,7 @@ class MockDocumentReference<T extends Object?> implements DocumentReference<T> {
   @override
   Future<void> delete() async {
     await _firestore.maybeThrowSecurityException(path, Method.delete);
+    maybeThrowException(this, Invocation.method(#delete, null));
     rootParent.remove(id);
     _firestore.removeSavedDocument(path);
     docsData.remove(path);

lib/src/mock_query.dart

@@ -4,6 +4,7 @@ import 'dart:math';
 import 'package:cloud_firestore/cloud_firestore.dart';
 import 'package:fake_cloud_firestore/src/util.dart';
 import 'package:flutter/services.dart';
+import 'package:mock_exceptions/mock_exceptions.dart';
 import 'package:quiver/core.dart';
 
 import 'converter.dart';
@@ -39,6 +40,7 @@ class MockQuery<T extends Object?> extends FakeQueryWithParent<T> {
     // Collection references: parent query is null.
     // Regular queries: _parentQuery and _operation are not null.
     assert(_parentQuery != null && _operation != null);
+    maybeThrowException(this, Invocation.method(#get, [options]));
     final parentQueryResult = await _parentQuery!.get(options);
     final docs = _operation!(parentQueryResult.docs);
     return MockQuerySnapshot<T>(docs, options?.source == Source.cache);

test/fake_cloud_firestore_test.dart

@@ -89,6 +89,34 @@ void main() {
           .thenThrow(FirebaseException(plugin: 'firestore'));
       expect(() => doc.set({'name': 'Bob'}), throwsA(isA<FirebaseException>()));
     });
+
+    test('DocumentReference.get throws exceptions', () async {
+      final instance = FakeFirebaseFirestore();
+      final doc = instance.collection('users').doc(uid);
+      whenCalling(Invocation.method(#get, null))
+          .on(doc)
+          .thenThrow(FirebaseException(plugin: 'firestore'));
+      expect(() => doc.get(), throwsA(isA<FirebaseException>()));
+    });
+
+    test('DocumentReference.delete throws exceptions', () async {
+      final instance = FakeFirebaseFirestore();
+      final doc = instance.collection('users').doc(uid);
+      whenCalling(Invocation.method(#delete, null))
+          .on(doc)
+          .thenThrow(FirebaseException(plugin: 'firestore'));
+      expect(() => doc.delete(), throwsA(isA<FirebaseException>()));
+    });
+
+    test('DocumentReference.update throws exceptions', () async {
+      final instance = FakeFirebaseFirestore();
+      final doc = instance.collection('users').doc(uid);
+      whenCalling(Invocation.method(#update, null))
+          .on(doc)
+          .thenThrow(FirebaseException(plugin: 'firestore'));
+      expect(() => doc.update({'data' : 'new'}), throwsA(isA<FirebaseException>()));
+    });
+
     test('DocumentReference.set throws exceptions on certain conditions',
         () async {
       final instance = FakeFirebaseFirestore();

test/mock_query_test.dart

@@ -2,6 +2,7 @@ import 'package:cloud_firestore/cloud_firestore.dart';
 import 'package:fake_cloud_firestore/fake_cloud_firestore.dart';
 import 'package:fake_cloud_firestore/src/util.dart';
 import 'package:flutter/services.dart';
+import 'package:mock_exceptions/mock_exceptions.dart';
 import 'package:test/test.dart';
 
 import 'document_snapshot_matcher.dart';
@@ -1654,4 +1655,27 @@ void main() {
       expect(query.metadata.isFromCache, false);
     });
   });
+
+  group('exceptions', () {
+    test('get', () async {
+      final firestore = FakeFirebaseFirestore();
+      final movies = firestore.collection('movies');
+      final query = firestore
+          .collection('movies')
+          .where('title', isEqualTo: 'Test Movie');
+
+      await movies.add({
+        'title': 'Test Movie',
+      });
+
+      whenCalling(Invocation.method(#get, null))
+          .on(query)
+          .thenThrow(FirebaseException(plugin: 'firestore'));
+
+      expect(
+        () async => await query.get(GetOptions(source: Source.server)),
+        throwsA(isA<FirebaseException>()),
+      );
+    });
+  });
 }