import 'dart:async'; import 'dart:io'; import 'package:appflowy/env/cloud_env.dart'; import 'package:appflowy/env/cloud_env_test.dart'; import 'package:appflowy/startup/entry_point.dart'; import 'package:appflowy/startup/startup.dart'; import 'package:appflowy/user/application/auth/af_cloud_mock_auth_service.dart'; import 'package:appflowy/user/application/auth/auth_service.dart'; import 'package:appflowy/user/application/auth/supabase_mock_auth_service.dart'; import 'package:appflowy/user/presentation/presentation.dart'; import 'package:appflowy/user/presentation/screens/sign_in_screen/widgets/widgets.dart'; import 'package:appflowy/workspace/application/settings/prelude.dart'; import 'package:dartz/dartz.dart'; import 'package:flowy_infra/uuid.dart'; import 'package:flowy_infra_ui/flowy_infra_ui.dart'; import 'package:flutter/gestures.dart'; import 'package:flutter/material.dart'; import 'package:flutter/services.dart'; import 'package:flutter_test/flutter_test.dart'; import 'package:path/path.dart' as p; import 'package:path_provider/path_provider.dart'; class FlowyTestContext { FlowyTestContext({ required this.applicationDataDirectory, }); final String applicationDataDirectory; } extension AppFlowyTestBase on WidgetTester { Future initializeAppFlowy({ // use to append after the application data directory String? pathExtension, Size windowsSize = const Size(1600, 1200), AuthenticatorType? cloudType, String? email, }) async { binding.setSurfaceSize(windowsSize); mockHotKeyManagerHandlers(); final directory = await mockApplicationDataStorage( pathExtension: pathExtension, ); WidgetsFlutterBinding.ensureInitialized(); await FlowyRunner.run( FlowyApp(), IntegrationMode.integrationTest, rustEnvsBuilder: () { final rustEnvs = {}; if (cloudType != null) { switch (cloudType) { case AuthenticatorType.local: break; case AuthenticatorType.supabase: break; case AuthenticatorType.appflowyCloud: rustEnvs["GOTRUE_ADMIN_EMAIL"] = "admin@example.com"; rustEnvs["GOTRUE_ADMIN_PASSWORD"] = "password"; break; } } return rustEnvs; }, didInitGetItCallback: () { return Future( () async { if (cloudType != null) { switch (cloudType) { case AuthenticatorType.local: await useLocal(); break; case AuthenticatorType.supabase: await useSupabaseCloud(); getIt.unregister(); getIt.registerFactory( () => SupabaseMockAuthService(), ); break; case AuthenticatorType.appflowyCloud: await useAppFlowyCloud(); getIt.unregister(); getIt.registerFactory( () => AppFlowyCloudMockAuthService(email: email), ); break; } } }, ); }, ); await waitUntilSignInPageShow(); return FlowyTestContext( applicationDataDirectory: directory, ); } void mockHotKeyManagerHandlers() { TestDefaultBinaryMessengerBinding.instance.defaultBinaryMessenger .setMockMethodCallHandler(const MethodChannel('hotkey_manager'), (MethodCall methodCall) async { if (methodCall.method == 'unregisterAll') { // do nothing } return; }); } Future mockApplicationDataStorage({ // use to append after the application data directory String? pathExtension, }) async { final dir = await getTemporaryDirectory(); // Use a random uuid to avoid conflict. String path = p.join(dir.path, 'appflowy_integration_test', uuid()); if (pathExtension != null && pathExtension.isNotEmpty) { path = '$path/$pathExtension'; } final directory = Directory(path); if (!directory.existsSync()) { await directory.create(recursive: true); } MockApplicationDataStorage.initialPath = directory.path; return directory.path; } Future waitUntilSignInPageShow() async { if (isAuthEnabled) { final finder = find.byType(SignInAnonymousButton); await pumpUntilFound(finder); expect(finder, findsOneWidget); } else { final finder = find.byType(GoButton); await pumpUntilFound(finder); expect(finder, findsOneWidget); } } Future pumpUntilFound( Finder finder, { Duration timeout = const Duration(seconds: 10), }) async { bool timerDone = false; final timer = Timer(timeout, () => timerDone = true); while (timerDone != true) { await pump(); if (any(finder)) { timerDone = true; } } timer.cancel(); } Future tapButton( Finder finder, { int? pointer, int buttons = kPrimaryButton, bool warnIfMissed = false, int milliseconds = 500, }) async { await tap( finder, buttons: buttons, warnIfMissed: warnIfMissed, ); await pumpAndSettle(Duration(milliseconds: milliseconds)); } Future tapButtonWithName( String tr, { int milliseconds = 500, }) async { Finder button = find.text( tr, findRichText: true, skipOffstage: false, ); if (button.evaluate().isEmpty) { button = find.byWidgetPredicate( (widget) => widget is FlowyText && widget.text == tr, ); } await tapButton( button, milliseconds: milliseconds, ); return; } Future tapButtonWithTooltip( String tr, { int milliseconds = 500, }) async { final button = find.byTooltip(tr); await tapButton( button, milliseconds: milliseconds, ); return; } Future doubleTapAt( Offset location, { int? pointer, int buttons = kPrimaryButton, int milliseconds = 500, }) async { await tapAt(location, pointer: pointer, buttons: buttons); await pump(kDoubleTapMinTime); await tapAt(location, pointer: pointer, buttons: buttons); await pumpAndSettle(Duration(milliseconds: milliseconds)); } Future doubleTapButton( Finder finder, { int? pointer, int buttons = kPrimaryButton, bool warnIfMissed = true, int milliseconds = 500, }) async { await tap( finder, pointer: pointer, buttons: buttons, warnIfMissed: warnIfMissed, ); await pump(kDoubleTapMinTime); await tap( finder, buttons: buttons, pointer: pointer, warnIfMissed: warnIfMissed, ); await pumpAndSettle(Duration(milliseconds: milliseconds)); } Future wait(int milliseconds) async { await pumpAndSettle(Duration(milliseconds: milliseconds)); return; } } extension AppFlowyFinderTestBase on CommonFinders { Finder findTextInFlowyText(String text) { return find.byWidgetPredicate( (widget) => widget is FlowyText && widget.text == text, ); } } Future useLocal() async { await setAuthenticatorType(AuthenticatorType.local); } Future useSupabaseCloud() async { await setAuthenticatorType(AuthenticatorType.supabase); await setSupbaseServer( Some(TestEnv.supabaseUrl), Some(TestEnv.supabaseAnonKey), ); } Future useAppFlowyCloud() async { await setAuthenticatorType(AuthenticatorType.appflowyCloud); await setAppFlowyCloudUrl(Some(TestEnv.afCloudUrl)); }