org.mockito.internal.invocation.AllInvocationsFinder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mockito-all

Mock objects library for java

There is a newer version: 2.0.2-beta

/*
 * Copyright (c) 2007 Mockito contributors
 * This program is made available under the terms of the MIT License.
 */
package org.mockito.internal.invocation;

import java.util.Comparator;
import java.util.LinkedList;
import java.util.List;
import java.util.Set;
import java.util.TreeSet;

import org.mockito.internal.util.MockUtil;

public class AllInvocationsFinder {
    
    /**
     * gets all invocations from mocks. Invocations are ordered earlier first. 
     * 
     * @param mocks mocks
     * @return invocations
     */
    public List getAllInvocations(List mocks) {
        Set invocationsInOrder = new TreeSet(new SequenceNumberComparator());
        for (Object mock : mocks) {
            List fromSingleMock = new MockUtil().getMockHandler(mock).getRegisteredInvocations();
            invocationsInOrder.addAll(fromSingleMock);
        }
        
        return new LinkedList(invocationsInOrder);
    }

    private final class SequenceNumberComparator implements Comparator {
        public int compare(Invocation o1, Invocation o2) {
            return o1.getSequenceNumber().compareTo(o2.getSequenceNumber());
        }
    }
}