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

Commit 7681275d authored by Lucas Dupin's avatar Lucas Dupin
Browse files

Fixed flaky test 

Change-Id: I88a39a8eb4fd6da6dc5b156aa475330299ffb9b9
Fixes: 62568528
Test: runtest systemui
parent a9205480

packages/SystemUI/tests/src/com/android/systemui/colorextraction/SysuiColorExtractorTests.java

+18 −5
Original line number Diff line number Diff line
@@ -18,24 +18,27 @@ package com.android.systemui.colorextraction; 


import static org.junit.Assert.assertEquals;



import android.app.WallpaperColors;

import android.app.WallpaperManager;

import android.graphics.Color;

import android.support.test.filters.SmallTest;

import android.support.test.runner.AndroidJUnit4;

import android.util.Pair;



import com.android.systemui.SysuiTestCase;



import com.google.android.colorextraction.ColorExtractor;

import com.google.android.colorextraction.types.Tonal;



import org.junit.Ignore;

import org.junit.Test;

import org.junit.runner.RunWith;



import java.util.ArrayList;

import java.util.List;



/**

 * Tests color extraction generation.

 */

@Ignore

@SmallTest

@RunWith(AndroidJUnit4.class)

public class SysuiColorExtractorTests extends SysuiTestCase {
@@ -55,6 +58,7 @@ public class SysuiColorExtractorTests extends SysuiTestCase { 
        fallbackColors.setSecondaryColor(ColorExtractor.FALLBACK_COLOR);



        SysuiColorExtractor extractor = new SysuiColorExtractor(getContext(), new Tonal(), false);

        simulateEvent(extractor);

        extractor.setWallpaperVisible(false);



        for (int which : sWhich) {
@@ -79,6 +83,7 @@ public class SysuiColorExtractorTests extends SysuiTestCase { 
                    outGradientColorsExtraDark.set(colors);

                    return true;

                }, false);

        simulateEvent(extractor);

        extractor.setWallpaperVisible(true);



        for (int which : sWhich) {
@@ -88,4 +93,12 @@ public class SysuiColorExtractorTests extends SysuiTestCase { 
            }

        }

    }



    private void simulateEvent(SysuiColorExtractor extractor) {

        // Let's fake a color event

        List<Pair<Color, Integer>> dummyColors = new ArrayList<>();

        dummyColors.add(new Pair<>(Color.valueOf(Color.BLACK), 1));

        extractor.onColorsChanged(new WallpaperColors(dummyColors),

                WallpaperManager.FLAG_SYSTEM | WallpaperManager.FLAG_LOCK);

    }

}
 
 No newline at end of file