Fix broken test in WifiDetailPreferenceControllerTest

import android.net.wifi.WifiManager;

import android.os.Handler;

import android.provider.Settings;

import android.util.FeatureFlagUtils;

import android.view.View;

import android.view.View.OnClickListener;

import android.widget.ImageView;

import com.android.settingslib.wifi.AccessPoint;

import org.junit.Before;

import org.junit.Ignore;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.mockito.Answers;

@RunWith(RobolectricTestRunner.class)

@Config(shadows = {ShadowDevicePolicyManager.class, ShadowEntityHeaderController.class})

@Ignore("b/129292549")

public class WifiDetailPreferenceControllerTest {

    private static final int LEVEL = 1;

        when(mockIconInjector.getIcon(anyInt())).thenReturn(new ColorDrawable());

        setupMockedPreferenceScreen();

        // Disable saved network detail page feature for this test

        FeatureFlagUtils.setEnabled(mContext, FeatureFlags.WIFI_DETAILS_SAVED_SCREEN, false);

        when(mockAccessPoint.isActive()).thenReturn(true);

        mController = newWifiDetailPreferenceController();

    }

        when(pref.setButton3Visible(anyBoolean())).thenReturn(pref);

        when(pref.setButton3OnClickListener(any(View.OnClickListener.class))).thenReturn(pref);

        when(pref.setButton4Text(anyInt())).thenReturn(pref);

        when(pref.setButton4Icon(anyInt())).thenReturn(pref);

        when(pref.setButton4Enabled(anyBoolean())).thenReturn(pref);

        when(pref.setButton4Visible(anyBoolean())).thenReturn(pref);

        when(pref.setButton4OnClickListener(any(View.OnClickListener.class))).thenReturn(pref);

        return pref;

    }

}