From e9435cd4aaa990d66ba73858997dfd761010b536 Mon Sep 17 00:00:00 2001
From: Roland Geider <roland@geider.net>
Date: Fri, 9 May 2025 21:46:09 +0200
Subject: [PATCH] Fix test

---
 lib/widgets/routines/forms/session.dart        |  2 ++
 test/workout/gym_mode_session_screen_test.dart | 15 +++++++++++----
 2 files changed, 13 insertions(+), 4 deletions(-)

diff --git a/lib/widgets/routines/forms/session.dart b/lib/widgets/routines/forms/session.dart
index 1c6ab5cc..4bb6d6cd 100644
--- a/lib/widgets/routines/forms/session.dart
+++ b/lib/widgets/routines/forms/session.dart
@@ -130,6 +130,7 @@ class _SessionFormState extends State<SessionForm> {
             children: [
               Flexible(
                 child: TextFormField(
+                  key: const ValueKey('time-start'),
                   decoration: InputDecoration(
                     labelText: AppLocalizations.of(context).timeStart,
                     errorMaxLines: 2,
@@ -172,6 +173,7 @@ class _SessionFormState extends State<SessionForm> {
               const SizedBox(width: 10),
               Flexible(
                 child: TextFormField(
+                  key: const ValueKey('time-end'),
                   decoration: InputDecoration(
                     labelText: AppLocalizations.of(context).timeEnd,
                   ),
diff --git a/test/workout/gym_mode_session_screen_test.dart b/test/workout/gym_mode_session_screen_test.dart
index a865b154..bb5c77f4 100644
--- a/test/workout/gym_mode_session_screen_test.dart
+++ b/test/workout/gym_mode_session_screen_test.dart
@@ -25,6 +25,7 @@ import 'package:provider/provider.dart';
 import 'package:wger/helpers/json.dart';
 import 'package:wger/l10n/generated/app_localizations.dart';
 import 'package:wger/models/workouts/routine.dart';
+import 'package:wger/models/workouts/session.dart';
 import 'package:wger/providers/routines.dart';
 import 'package:wger/widgets/routines/gym_mode/session_page.dart';
 
@@ -77,12 +78,17 @@ void main() {
   testWidgets('Test that data from  session is loaded - null times', (WidgetTester tester) async {
     testRoutine.sessions[0].session.timeStart = null;
     testRoutine.sessions[0].session.timeEnd = null;
-    final timeNow = timeToString(TimeOfDay.now())!;
 
     withClock(Clock.fixed(DateTime(2021, 5, 1)), () async {
       await tester.pumpWidget(renderSessionPage());
-      expect(find.text('13:35'), findsOneWidget);
-      expect(find.text(timeNow), findsOneWidget);
+
+      final startTimeField = find.byKey(const ValueKey('time-start'));
+      expect(startTimeField, findsOneWidget);
+      expect(tester.widget<TextFormField>(startTimeField).controller!.text, '');
+
+      final endTimeField = find.byKey(const ValueKey('time-end'));
+      expect(endTimeField, findsOneWidget);
+      expect(tester.widget<TextFormField>(endTimeField).controller!.text, '');
     });
   });
 
@@ -101,7 +107,8 @@ void main() {
     withClock(Clock.fixed(DateTime(2021, 5, 1)), () async {
       await tester.pumpWidget(renderSessionPage());
       await tester.tap(find.byKey(const ValueKey('save-button')));
-      final captured = verify(mockRoutinesProvider.editSession(captureAny)).captured.single;
+      final captured =
+          verify(mockRoutinesProvider.editSession(captureAny)).captured.single as WorkoutSession;
 
       expect(captured.id, 1);
       expect(captured.impression, 3);