import 'package:appflowy/plugins/database_view/application/cell/cell_controller.dart'; import 'package:appflowy/plugins/database_view/application/cell/cell_controller_builder.dart'; import 'package:appflowy/plugins/database_view/application/field/field_controller.dart'; import 'package:appflowy/plugins/database_view/application/field/field_editor_bloc.dart'; import 'package:appflowy/plugins/database_view/application/field/field_service.dart'; import 'package:appflowy/plugins/database_view/application/field/type_option/type_option_context.dart'; import 'package:appflowy/plugins/database_view/application/row/row_cache.dart'; import 'package:appflowy/plugins/database_view/application/row/row_data_controller.dart'; import 'package:appflowy/plugins/database_view/grid/application/grid_data_controller.dart'; import 'package:appflowy/plugins/database_view/grid/application/row/row_bloc.dart'; import 'package:appflowy/plugins/database_view/grid/grid.dart'; import 'package:appflowy/workspace/application/app/app_service.dart'; import 'package:appflowy_backend/protobuf/flowy-folder/view.pb.dart'; import 'package:appflowy_backend/protobuf/flowy-database/field_entities.pb.dart'; import '../../util.dart'; class GridTestContext { final ViewPB gridView; final DatabaseController gridController; GridTestContext(this.gridView, this.gridController); List get rowInfos { return gridController.rowInfos; } List get fieldContexts => fieldController.fieldInfos; FieldController get fieldController { return gridController.fieldController; } Future createRow() async { return gridController.createRow(); } FieldEditorBloc createFieldEditor({ FieldInfo? fieldInfo, }) { IFieldTypeOptionLoader loader; if (fieldInfo == null) { loader = NewFieldTypeOptionLoader(viewId: gridView.id); } else { loader = FieldTypeOptionLoader(viewId: gridView.id, field: fieldInfo.field); } final editorBloc = FieldEditorBloc( fieldName: fieldInfo?.name ?? '', isGroupField: fieldInfo?.isGroupField ?? false, loader: loader, viewId: gridView.id, ); return editorBloc; } Future makeCellController( String fieldId, int rowIndex) async { final builder = await makeCellControllerBuilder(fieldId, rowIndex); return builder.build(); } Future makeCellControllerBuilder( String fieldId, int rowIndex, ) async { final RowInfo rowInfo = rowInfos[rowIndex]; final rowCache = gridController.rowCache; final rowDataController = RowDataController( rowInfo: rowInfo, rowCache: rowCache, ); final rowBloc = RowBloc( rowInfo: rowInfo, dataController: rowDataController, )..add(const RowEvent.initial()); await gridResponseFuture(); return CellControllerBuilder( cellId: rowBloc.state.cellByFieldId[fieldId]!, cellCache: rowCache.cellCache, ); } Future createField(FieldType fieldType) async { final editorBloc = createFieldEditor() ..add(const FieldEditorEvent.initial()); await gridResponseFuture(); editorBloc.add(FieldEditorEvent.switchToField(fieldType)); await gridResponseFuture(); return Future(() => editorBloc); } FieldInfo singleSelectFieldContext() { final fieldInfo = fieldContexts .firstWhere((element) => element.fieldType == FieldType.SingleSelect); return fieldInfo; } FieldCellContext singleSelectFieldCellContext() { final field = singleSelectFieldContext().field; return FieldCellContext(viewId: gridView.id, field: field); } FieldInfo textFieldContext() { final fieldInfo = fieldContexts .firstWhere((element) => element.fieldType == FieldType.RichText); return fieldInfo; } FieldInfo checkboxFieldContext() { final fieldInfo = fieldContexts .firstWhere((element) => element.fieldType == FieldType.Checkbox); return fieldInfo; } Future makeSelectOptionCellController( FieldType fieldType, int rowIndex) async { assert(fieldType == FieldType.SingleSelect || fieldType == FieldType.MultiSelect); final field = fieldContexts.firstWhere((element) => element.fieldType == fieldType); final cellController = await makeCellController(field.id, rowIndex) as SelectOptionCellController; return cellController; } Future makeTextCellController(int rowIndex) async { final field = fieldContexts .firstWhere((element) => element.fieldType == FieldType.RichText); final cellController = await makeCellController(field.id, rowIndex) as TextCellController; return cellController; } Future makeCheckboxCellController(int rowIndex) async { final field = fieldContexts .firstWhere((element) => element.fieldType == FieldType.Checkbox); final cellController = await makeCellController(field.id, rowIndex) as TextCellController; return cellController; } } /// Create a empty Grid for test class AppFlowyGridTest { final AppFlowyUnitTest unitTest; AppFlowyGridTest({required this.unitTest}); static Future ensureInitialized() async { final inner = await AppFlowyUnitTest.ensureInitialized(); return AppFlowyGridTest(unitTest: inner); } Future createTestGrid() async { final app = await unitTest.createTestApp(); final builder = GridPluginBuilder(); final context = await AppService() .createView( appId: app.id, name: "Test Grid", layoutType: builder.layoutType!, ) .then((result) { return result.fold( (view) async { final context = GridTestContext(view, DatabaseController(view: view)); final result = await context.gridController.openGrid(); result.fold((l) => null, (r) => throw Exception(r)); return context; }, (error) { throw Exception(); }, ); }); return context; } } /// Create a new Grid for cell test class AppFlowyGridCellTest { late GridTestContext context; final AppFlowyGridTest gridTest; AppFlowyGridCellTest({required this.gridTest}); static Future ensureInitialized() async { final gridTest = await AppFlowyGridTest.ensureInitialized(); return AppFlowyGridCellTest(gridTest: gridTest); } Future createTestGrid() async { context = await gridTest.createTestGrid(); } Future createTestRow() async { await context.createRow(); } Future makeSelectOptionCellController( FieldType fieldType, int rowIndex) async { return await context.makeSelectOptionCellController(fieldType, rowIndex); } } Future gridResponseFuture({int milliseconds = 200}) { return Future.delayed(gridResponseDuration(milliseconds: milliseconds)); } Duration gridResponseDuration({int milliseconds = 200}) { return Duration(milliseconds: milliseconds); }