Add token details testing to existing tests instead

Author: andrewheard

packages/firebase_ai/firebase_ai/test/developer_api_test.dart

@@ -39,6 +39,12 @@ void main() {
             'candidatesTokenCount': 5,
             'totalTokenCount': 15,
             'thoughtsTokenCount': 3,
+            'promptTokensDetails': [
+              {'modality': 'TEXT', 'tokenCount': 10}
+            ],
+            'candidatesTokensDetails': [
+              {'modality': 'TEXT', 'tokenCount': 25}
+            ],
           }
         };
         final response =
@@ -48,6 +54,15 @@ void main() {
         expect(response.usageMetadata!.candidatesTokenCount, 5);
         expect(response.usageMetadata!.totalTokenCount, 15);
         expect(response.usageMetadata!.thoughtsTokenCount, 3);
+        expect(response.usageMetadata!.promptTokensDetails, isNotNull);
+        expect(response.usageMetadata!.promptTokensDetails, hasLength(1));
+        expect(
+            response.usageMetadata!.promptTokensDetails!.first.tokenCount, 10);
+        expect(response.usageMetadata!.candidatesTokensDetails, isNotNull);
+        expect(response.usageMetadata!.candidatesTokensDetails, hasLength(1));
+        expect(
+            response.usageMetadata!.candidatesTokensDetails!.first.tokenCount,
+            25);
       });
 
       test('parses usageMetadata when thoughtsTokenCount is missing', () {
@@ -68,6 +83,12 @@ void main() {
             'candidatesTokenCount': 5,
             'totalTokenCount': 15,
             // thoughtsTokenCount is missing
+            'promptTokensDetails': [
+              {'modality': 'TEXT', 'tokenCount': 10}
+            ],
+            'candidatesTokensDetails': [
+              {'modality': 'TEXT', 'tokenCount': 25}
+            ],
           }
         };
         final response =
@@ -153,40 +174,6 @@ void main() {
         expect((part as InlineDataPart).mimeType, 'application/octet-stream');
         expect(part.bytes, inlineData);
       });
-
-      test(
-          'parses usageMetadata with prompt and candidate token details correctly',
-          () {
-        final jsonResponse = {
-          'usageMetadata': {
-            'promptTokenCount': 10,
-            'candidatesTokenCount': 25,
-            'totalTokenCount': 35,
-            'promptTokensDetails': [
-              {'modality': 'TEXT', 'tokenCount': 10}
-            ],
-            'candidatesTokensDetails': [
-              {'modality': 'TEXT', 'tokenCount': 25}
-            ],
-          }
-        };
-
-        final response =
-            DeveloperSerialization().parseGenerateContentResponse(jsonResponse);
-
-        expect(response.usageMetadata, isNotNull);
-        expect(response.usageMetadata!.promptTokenCount, 10);
-        expect(response.usageMetadata!.candidatesTokenCount, 25);
-        expect(response.usageMetadata!.totalTokenCount, 35);
-        expect(response.usageMetadata!.promptTokensDetails, isNotNull);
-        expect(response.usageMetadata!.promptTokensDetails, hasLength(1));
-        expect(response.usageMetadata!.promptTokensDetails!.first.tokenCount,
-            10);
-        expect(response.usageMetadata!.candidatesTokensDetails, isNotNull);
-        expect(response.usageMetadata!.candidatesTokensDetails, hasLength(1));
-        expect(response.usageMetadata!.candidatesTokensDetails!.first.tokenCount,
-            25);
-      });
     });
   });
 }