diff --git a/integration_test/dsl/dsl.dart b/integration_test/dsl/dsl.dart
index 6f257011c..94ab32f63 100644
--- 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();