Update dsl.dart

ardriveapp · Nov 7, 2024 · c618b2e · c618b2e
1 parent cc2410d
commit c618b2e
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/integration_test/dsl/dsl.dart b/integration_test/dsl/dsl.dart
@@ -237,6 +237,18 @@ class See extends Action {
     return multipleTextTest;
   }
 
+  ButtonTestWithKey publicDriveButton(String driveName) {
+    final driveButton = ButtonTestWithKey('public_drives_$driveName');
+    driveButton.expectComponent();
+    return driveButton;
+  }
+
+  ButtonTestWithKey privateDriveButton(String driveName) {
+    final driveButton = ButtonTestWithKey('private_drives_$driveName');
+    driveButton.expectComponent();
+    return driveButton;
+  }
+
   @override
   Future<void> execute(WidgetTester tester, Component component) async {
     component.expectComponent();