From 0b16fea42a0aa80a3d3b564768358e9d9611afaf Mon Sep 17 00:00:00 2001
From: DatDang <tddang@linagora.com>
Date: Tue, 18 Mar 2025 10:55:08 +0700
Subject: [PATCH] TF-3564 E2E Mailbox count real time update

---
 .../null_inbox_unread_count_exception.dart    |  1 +
 .../robots/mailbox_menu_robot.dart            | 19 ++++++++
 ...ilbox_count_real_time_update_scenario.dart | 47 +++++++++++++++++++
 .../mailbox_count_real_time_update_test.dart  |  9 ++++
 4 files changed, 76 insertions(+)
 create mode 100644 integration_test/exceptions/mailbox/null_inbox_unread_count_exception.dart
 create mode 100644 integration_test/scenarios/mailbox/mailbox_count_real_time_update_scenario.dart
 create mode 100644 integration_test/tests/mailbox/mailbox_count_real_time_update_test.dart

diff --git a/integration_test/exceptions/mailbox/null_inbox_unread_count_exception.dart b/integration_test/exceptions/mailbox/null_inbox_unread_count_exception.dart
new file mode 100644
index 000000000..da6dfa130
--- /dev/null
+++ b/integration_test/exceptions/mailbox/null_inbox_unread_count_exception.dart
@@ -0,0 +1 @@
+class NullInboxUnreadCountException implements Exception {}
diff --git a/integration_test/robots/mailbox_menu_robot.dart b/integration_test/robots/mailbox_menu_robot.dart
index 5aed6b101..edbf18917 100644
--- a/integration_test/robots/mailbox_menu_robot.dart
+++ b/integration_test/robots/mailbox_menu_robot.dart
@@ -4,12 +4,15 @@ import 'package:core/presentation/views/button/tmail_button_widget.dart';
 import 'package:core/presentation/views/text/text_field_builder.dart';
 import 'package:flutter/material.dart';
 import 'package:flutter_test/flutter_test.dart';
+import 'package:get/get.dart';
 import 'package:tmail_ui_user/features/mailbox/presentation/widgets/label_mailbox_item_widget.dart';
 import 'package:tmail_ui_user/features/mailbox/presentation/widgets/mailbox_item_widget.dart';
 import 'package:tmail_ui_user/features/mailbox_creator/presentation/mailbox_creator_view.dart';
 import 'package:tmail_ui_user/main/localizations/app_localizations.dart';
+import 'package:tmail_ui_user/features/mailbox_dashboard/presentation/controller/mailbox_dashboard_controller.dart';
 
 import '../base/core_robot.dart';
+import '../exceptions/mailbox/null_inbox_unread_count_exception.dart';
 
 class MailboxMenuRobot extends CoreRobot {
   MailboxMenuRobot(super.$);
@@ -74,4 +77,20 @@ class MailboxMenuRobot extends CoreRobot {
       })
       .tap();
   }
+
+  int getCurrentInboxCount() {
+    final inboxCount = Get.find<MailboxDashBoardController>()
+      .selectedMailbox
+      .value
+      ?.unreadEmails
+      ?.value
+      .value
+      .toInt();
+
+    if (inboxCount == null) {
+      throw NullInboxUnreadCountException();
+    }
+
+    return inboxCount;
+  }
 }
\ No newline at end of file
diff --git a/integration_test/scenarios/mailbox/mailbox_count_real_time_update_scenario.dart b/integration_test/scenarios/mailbox/mailbox_count_real_time_update_scenario.dart
new file mode 100644
index 000000000..f03cacf91
--- /dev/null
+++ b/integration_test/scenarios/mailbox/mailbox_count_real_time_update_scenario.dart
@@ -0,0 +1,47 @@
+import '../../base/base_test_scenario.dart';
+import '../../models/provisioning_email.dart';
+import '../../robots/mailbox_menu_robot.dart';
+import '../../robots/thread_robot.dart';
+
+class MailboxCountRealTimeUpdateScenario extends BaseTestScenario {
+  const MailboxCountRealTimeUpdateScenario(super.$);
+
+  @override
+  Future<void> runTestLogic() async {
+    const email = String.fromEnvironment('BASIC_AUTH_EMAIL');
+    const subject = 'mailbox unread realtime update';
+
+    final threadRobot = ThreadRobot($);
+    final mailboxMenuRobot = MailboxMenuRobot($);
+
+    await provisionEmail([ProvisioningEmail(
+      toEmail: email,
+      subject: 'dummy email',
+      content: '',
+    )]);
+    await $.pumpAndSettle(duration: const Duration(seconds: 2));
+
+    await threadRobot.openMailbox();
+    final currentInboxCount = mailboxMenuRobot.getCurrentInboxCount();
+    await _expectInboxWithCount(currentInboxCount);
+
+    await provisionEmail([ProvisioningEmail(
+      toEmail: email,
+      subject: subject,
+      content: '',
+    )]);
+    await $.pumpAndSettle(duration: const Duration(seconds: 2));
+    await _expectInboxWithCount(currentInboxCount + 1);
+
+    await simulateUpdateFlagsOfEmailsWithSubjectsFromOutsideCurrentClient(
+      subjects: [subject],
+      isRead: true,
+    );
+    await $.pumpAndSettle(duration: const Duration(seconds: 5));
+    await _expectInboxWithCount(currentInboxCount);
+  }
+
+  Future<void> _expectInboxWithCount(int currentInboxCount) async {
+    await expectViewVisible($(currentInboxCount.toString()));
+  }
+}
diff --git a/integration_test/tests/mailbox/mailbox_count_real_time_update_test.dart b/integration_test/tests/mailbox/mailbox_count_real_time_update_test.dart
new file mode 100644
index 000000000..fb1ab78c0
--- /dev/null
+++ b/integration_test/tests/mailbox/mailbox_count_real_time_update_test.dart
@@ -0,0 +1,9 @@
+import '../../base/test_base.dart';
+import '../../scenarios/mailbox/mailbox_count_real_time_update_scenario.dart';
+
+void main() {
+  TestBase().runPatrolTest(
+    description: 'Should see mailbox unread count update real time',
+    scenarioBuilder: ($) => MailboxCountRealTimeUpdateScenario($),
+  );
+}