ValueExpectations

gistfile1.java

    private User activeUser(Customer customer) {
       return mock(User.class,
                "isAccountNonLocked", true,
                "isEnabled", true,
                "isAccountNonExpired", true,
                "isPasswordProvided", true,
                "isCredentialsNonExpired", true,
                "getCustomer", customer,
                "getAuthorities", ImmutableList.of()
        );
    }


    public <T> T mock(Class<T> clas, Object... pairs) {
        T result = mockery.mock(clas);
        mockery.checking(new ValueExpectations(result, pairs));
        return result;
    }



    public class ValueExpectations implements ExpectationBuilder {

        private final Iterable<Map.Entry<String, Object>> values;
        private final Object subject;

        public ValueExpectations(Object subject, Iterable<Map.Entry<String, Object>> values) {
            this.subject = subject;
            this.values = values;
        }

        public <T> ValueExpectations(T subject, Object... pairs) {
            this(subject, mapEntriesFor(pairs));
        }

        @Override
        public void buildExpectations(Action action, ExpectationCollector expectationCollector) {
            for (Map.Entry<String, Object> value : values) {
                expectationCollector.add(expecationBuilderFor(value.getKey(), value.getValue()).toExpectation(action));
            }
        }

        private InvocationExpectationBuilder expecationBuilderFor(String name, Object value) {
            InvocationExpectationBuilder builder = new InvocationExpectationBuilder();
            builder.of(subject);
            builder.method(name);
            builder.setAction(new ReturnValueAction(value));
            return builder;
        }

        private static Iterable<Map.Entry<String, Object>> mapEntriesFor(Object[] pairs) {
            if (pairs.length % 2 != 0)
                throw new IllegalArgumentException("argument must be pairs");
            List<Map.Entry<String, Object>> result = Lists.newArrayListWithCapacity(pairs.length / 2);
            for (int i = 0; i < pairs.length;) {
                result.add(Pair.of((String)pairs[i++], pairs[i++]));
            }
            return result;
        }
    }