From 394191b5733c4e9e52c554fcbc2304c7982aebd7 Mon Sep 17 00:00:00 2001
From: TrochtaOndrej <TrochtaOndrej@gmail.com>
Date: Wed, 18 Mar 2026 21:08:17 +0100
Subject: [PATCH] =?UTF-8?q?feat:=20Connect=20AddTask=20to=20Drift=20DB=20?=
 =?UTF-8?q?=E2=80=94=20remove=20sample=20tasks=20[AITM-232]?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 lib/features/tasks/screens/add_task_screen.dart | 2 +-
 lib/features/zen_mode/screens/zen_screen.dart   | 7 +++++--
 test/widget_test.dart                           | 2 +-
 3 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/lib/features/tasks/screens/add_task_screen.dart b/lib/features/tasks/screens/add_task_screen.dart
index 2568e95..507e1be 100644
--- a/lib/features/tasks/screens/add_task_screen.dart
+++ b/lib/features/tasks/screens/add_task_screen.dart
@@ -55,7 +55,7 @@ class _AddTaskSheetState extends ConsumerState<_AddTaskSheet> {
     final taskTime = ref.read(selectedTaskTimeProvider);
     final deadline = _resolveDeadline(taskDay, taskTime);
     final now = DateTime.now();
-    final tasks = ref.read(tasksProvider).valueOrNull ?? [];
+    final tasks = ref.read(tasksProvider).value ?? [];
 
     final entity = TaskEntity(
       id: const Uuid().v4(),
diff --git a/lib/features/zen_mode/screens/zen_screen.dart b/lib/features/zen_mode/screens/zen_screen.dart
index e1a2d2c..fa88b55 100644
--- a/lib/features/zen_mode/screens/zen_screen.dart
+++ b/lib/features/zen_mode/screens/zen_screen.dart
@@ -56,7 +56,7 @@ class ZenScreen extends ConsumerWidget {
                     Expanded(
                       child: ElevatedButton(
                         onPressed: () {
-                          ref.read(tasksProvider.notifier).completeTask(task.id);
+                          ref.read(taskRepositoryProvider).completeTask(task.id);
                           ref.read(petStateProvider.notifier).incrementTasksDone();
                         },
                         style: ElevatedButton.styleFrom(
@@ -76,7 +76,10 @@ class ZenScreen extends ConsumerWidget {
                     Expanded(
                       child: TextButton(
                         onPressed: () {
-                          ref.read(tasksProvider.notifier).skipTask(task.id);
+                          ref.read(taskRepositoryProvider).updateTask(
+                                task.copyWith(
+                                    sortOrder: task.sortOrder + 100),
+                              );
                         },
                         style: TextButton.styleFrom(
                           foregroundColor: KittieColors.brownLight,
diff --git a/test/widget_test.dart b/test/widget_test.dart
index 89870b3..79cf60b 100644
--- a/test/widget_test.dart
+++ b/test/widget_test.dart
@@ -9,7 +9,7 @@ void main() {
     await tester.pumpWidget(
       const ProviderScope(child: KittieApp()),
     );
-    await tester.pumpAndSettle();
+    await tester.pump();
 
     // App renders without crashing
     expect(find.byType(MaterialApp), findsOneWidget);