Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 688b506e authored by Zemiao Zhu's avatar Zemiao Zhu
Browse files

Deflake FileCopyUiTest by fixing tearDown and setup. 

Bug: 157092101
Test: atest
DocumentsUIGoogleTests: com.android.documentsui.FileCopyUiTest
Change-Id: I406a6ce0a5eb7f0cf8075b2100cb8a8d4b41e780
parent 8b475e36

tests/common/com/android/documentsui/bots/Bots.java

+1 −1
Original line number Diff line number Diff line
@@ -35,7 +35,7 @@ import androidx.test.InstrumentationRegistry; 
 */

public final class Bots {



    private static final int TIMEOUT = 5000;

    private static final int TIMEOUT = 15000;



    public final BreadBot breadcrumb;

    public final DirectoryListBot directory;

tests/functional/com/android/documentsui/FileCopyUiTest.java

+19 −1
Original line number Diff line number Diff line
@@ -153,8 +153,13 @@ public class FileCopyUiTest extends ActivityTest<FilesActivity> { 
            mIsVirtualSdCard = enableVirtualSdCard();

            assertTrue("Cannot set virtual SD Card", mIsVirtualSdCard);

            // Call initStorageRootInfo() again for setting SD Card root

            int attempts = 0;

            while (mSdCardRoot == null && attempts++ < 15) {

                SystemClock.sleep(1000);

                initStorageRootInfo();

            }

            assertNotNull("Cannot find virtual SD Card", mSdCardRoot);

        }



        IntentFilter filter = new IntentFilter();

        filter.addAction(TestNotificationService.ACTION_OPERATION_RESULT);
@@ -173,8 +178,21 @@ public class FileCopyUiTest extends ActivityTest<FilesActivity> { 
            deleteDocuments(rootAndFolder.root, rootAndFolder.folder);

        }



        // Eject virtual SD card

        if (mIsVirtualSdCard) {

            device.executeShellCommand("sm set-virtual-disk false");

            while (mSdCardRoot != null) {

                List<RootInfo> rootList = mStorageDocsHelper.getRootList();

                for (RootInfo info : rootList) {

                    if (info.isSd()) {

                        SystemClock.sleep(1000);

                        break;

                    }



                    mSdCardRoot = null;

                    mSdCardLabel = null;

                }

            }

        }



        device.executeShellCommand("settings put global stay_on_while_plugged_in "