xref: /external/robolectric/v1/src/main/java/com/xtremelabs/robolectric/shadows/ShadowHandler.java

package com.xtremelabs.droidsugar.fakes;

import android.os.Handler;
import android.os.Looper;
import android.os.Message;
import com.xtremelabs.droidsugar.ProxyDelegatingHandler;
import com.xtremelabs.droidsugar.util.Implementation;
import com.xtremelabs.droidsugar.util.Implements;

@SuppressWarnings({"UnusedDeclaration"})
@Implements(Handler.class)
public class FakeHandler {
    private Handler realHandler;
    private Looper looper;

    public FakeHandler(Handler realHandler) {
        this.realHandler = realHandler;
    }

    public void __constructor__() {
        this.looper = Looper.myLooper();
    }

    public void __constructor__(Looper looper) {
        this.looper = looper;
    }

    @Implementation
    public boolean post(Runnable r) {
        return postDelayed(r, 0);
    }

    @Implementation
    public boolean postDelayed(Runnable r, long delayMillis) {
        proxyFor(looper).post(r, delayMillis);
        return true;
    }

    @Implementation
    public Message obtainMessage(int what, Object obj) {
        Message message = new Message();
        message.what = what;
        message.obj = obj;
        return message;
    }

    @Implementation
    public final boolean sendMessage(final Message msg) {
        post(new Runnable() {
            @Override
            public void run() {
                realHandler.handleMessage(msg);
            }
        });
        return true;
    }

    public static void flush() {
        proxyFor(Looper.myLooper()).idle();
    }

    private static FakeLooper proxyFor(Looper looper) {
        return (FakeLooper) ProxyDelegatingHandler.getInstance().proxyFor(looper);
    }
}