Merge "Fix crash due to unregistered receiver"

        }

    }

    @VisibleForTesting

    protected void unregisterBroadcastReceiver() {

        mBroadcastDispatcher.unregisterReceiver(mLockEventReceiver);

    }

    @Override

    public void onDestroy() {

        unregisterReceiver(mLockEventReceiver);

        unregisterBroadcastReceiver();

        super.onDestroy();

    }

import static android.app.ActivityManager.TaskDescription;

import static org.junit.Assert.assertEquals;

import static org.mockito.ArgumentMatchers.any;

import static org.mockito.Mockito.eq;

import static org.mockito.Mockito.never;

import static org.mockito.Mockito.verify;

import static org.mockito.Mockito.when;

import android.annotation.ColorInt;

import com.android.systemui.SysuiTestCase;

import com.android.systemui.broadcast.BroadcastDispatcher;

import com.android.systemui.keyguard.WorkLockActivity;

import org.junit.Before;

import org.junit.Test;

                .putExtra(Intent.EXTRA_USER_ID, USER_ID));

        assertEquals(orgColor, mActivity.getPrimaryColor());

    }

    @Test

    public void testUnregisteredFromDispatcher() {

        mActivity.unregisterBroadcastReceiver();

        verify(mBroadcastDispatcher).unregisterReceiver(any());

        verify(mContext, never()).unregisterReceiver(any());

    }

}