src/main/java/com/xtremelabs/robolectric/shadows/ShadowAnimationDrawable.java

package com.xtremelabs.robolectric.shadows;

import android.graphics.drawable.AnimationDrawable;
import android.graphics.drawable.Drawable;
import com.xtremelabs.robolectric.internal.Implementation;
import com.xtremelabs.robolectric.internal.Implements;

import java.util.ArrayList;
import java.util.List;

@Implements(AnimationDrawable.class)
public class ShadowAnimationDrawable extends ShadowDrawable {

    private List<Drawable> frames = new ArrayList<Drawable>();

    @Implementation
    public void addFrame(Drawable frame, int duration) {
        frames.add(frame);
    }

    @Implementation
    public int getNumberOfFrames() {
        return frames.size();
    }
}