| /* |
| * Copyright (C) 2007 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package com.android.unit_tests.os; |
| |
| import android.os.Handler; |
| import android.os.Looper; |
| import android.os.Message; |
| import android.os.MessageQueue.IdleHandler; |
| import android.test.suitebuilder.annotation.MediumTest; |
| import junit.framework.TestCase; |
| |
| public class IdleHandlerTest extends TestCase { |
| |
| private static class BaseTestHandler extends TestHandlerThread { |
| Handler mHandler; |
| |
| public BaseTestHandler() { |
| } |
| |
| public void go() { |
| mHandler = new Handler() { |
| public void handleMessage(Message msg) { |
| BaseTestHandler.this.handleMessage(msg); |
| } |
| }; |
| } |
| |
| public void addIdleHandler() { |
| Looper.myQueue().addIdleHandler(new IdleHandler() { |
| public boolean queueIdle() { |
| return BaseTestHandler.this.queueIdle(); |
| } |
| }); |
| } |
| |
| public void handleMessage(Message msg) { |
| } |
| |
| public boolean queueIdle() { |
| return false; |
| } |
| } |
| |
| @MediumTest |
| public void testOneShotFirst() throws Exception { |
| TestHandlerThread tester = new BaseTestHandler() { |
| int mCount; |
| |
| public void go() { |
| super.go(); |
| mCount = 0; |
| mHandler.sendMessageDelayed(mHandler.obtainMessage(0), 100); |
| addIdleHandler(); |
| } |
| |
| public void handleMessage(Message msg) { |
| if (msg.what == 0) { |
| mHandler.sendMessageDelayed(mHandler.obtainMessage(1), 100); |
| } else if (msg.what == 1) { |
| if (mCount == 1) { |
| success(); |
| } else { |
| failure(new RuntimeException( |
| "Idle handler called " + mCount + " times")); |
| } |
| } |
| } |
| |
| public boolean queueIdle() { |
| mCount++; |
| return false; |
| } |
| }; |
| |
| tester.doTest(1000); |
| } |
| |
| @MediumTest |
| public void testOneShotLater() throws Exception { |
| TestHandlerThread tester = new BaseTestHandler() { |
| int mCount; |
| |
| public void go() { |
| super.go(); |
| mCount = 0; |
| mHandler.sendMessage(mHandler.obtainMessage(0)); |
| } |
| |
| public void handleMessage(Message msg) { |
| if (msg.what == 0) { |
| addIdleHandler(); |
| mHandler.sendMessageDelayed(mHandler.obtainMessage(1), 100); |
| } else if (msg.what == 1) { |
| mHandler.sendMessageDelayed(mHandler.obtainMessage(2), 100); |
| } else if (msg.what == 2) { |
| if (mCount == 1) { |
| success(); |
| } else { |
| failure(new RuntimeException( |
| "Idle handler called " + mCount + " times")); |
| } |
| } |
| } |
| |
| public boolean queueIdle() { |
| mCount++; |
| return false; |
| } |
| }; |
| |
| tester.doTest(1000); |
| } |
| |
| |
| @MediumTest |
| public void testRepeatedFirst() throws Exception { |
| TestHandlerThread tester = new BaseTestHandler() { |
| int mCount; |
| |
| public void go() { |
| super.go(); |
| mCount = 0; |
| mHandler.sendMessageDelayed(mHandler.obtainMessage(0), 100); |
| addIdleHandler(); |
| } |
| |
| public void handleMessage(Message msg) { |
| if (msg.what == 0) { |
| mHandler.sendMessageDelayed(mHandler.obtainMessage(1), 100); |
| } else if (msg.what == 1) { |
| if (mCount == 2) { |
| success(); |
| } else { |
| failure(new RuntimeException( |
| "Idle handler called " + mCount + " times")); |
| } |
| } |
| } |
| |
| public boolean queueIdle() { |
| mCount++; |
| return true; |
| } |
| }; |
| |
| tester.doTest(1000); |
| } |
| |
| @MediumTest |
| public void testRepeatedLater() throws Exception { |
| TestHandlerThread tester = new BaseTestHandler() { |
| int mCount; |
| |
| public void go() { |
| super.go(); |
| mCount = 0; |
| mHandler.sendMessage(mHandler.obtainMessage(0)); |
| } |
| |
| public void handleMessage(Message msg) { |
| if (msg.what == 0) { |
| addIdleHandler(); |
| mHandler.sendMessageDelayed(mHandler.obtainMessage(1), 100); |
| } else if (msg.what == 1) { |
| mHandler.sendMessageDelayed(mHandler.obtainMessage(2), 100); |
| } else if (msg.what == 2) { |
| if (mCount == 2) { |
| success(); |
| } else { |
| failure(new RuntimeException( |
| "Idle handler called " + mCount + " times")); |
| } |
| } |
| } |
| |
| public boolean queueIdle() { |
| mCount++; |
| return true; |
| } |
| }; |
| |
| tester.doTest(1000); |
| } |
| } |
| |