Powermock：如何模拟枚举常量中的方法

Question

I have an enumeration type which I used as a collection of factories for other objects that implement a common interface. A simplified version of the code is:

interface Operation { 
    void execute();
}

enum Factory {
    TYPE1(Class1::new),
    TYPE2(Class2::new);

    private Supplier<Operation> constructor;

    Factory(Supplier<Operation> constructor) {
        this.constructor = constructor;
    }

    Operation build() {
        return constructor.get(); 
    }
}

A simplified version of the client code looks like:

class Client {
    private void run(EnumSet<Factory> required) {
        for (Factory x : required) {
            x.build().execute(); 
        }
    }

    // Some code that calls run() with the right EnumSet
}

This all appears to work as expected, so I want to write some Unit tests.

Testing the Factory is easy, but the Client is proving more difficult. The problem is that I don't want to start calling the Operation (they do a lot of work). Instead I'd like to get x.build() return a mock object.

I've tried using PowerMocks whenNew to trap the creation of the Operation objects, but this doesn't work (I don't actually have any 'new' operations). I've also tried to use a Powermock 'spy', but this fails because the enumeration constants are real objects.

Any ideas?

Answer 1

Two solutions:

• you insist on using PowerMock(ito). Then the simple issue is: you are probably preparing the wrong class for test. You have to temper with those enum constants , and they are inner classes. But it is possible, see this answer for example.
• you step back, and improve your production code, to make it easier to test.

For example like this:

interface OperationProvider { 
    Operation build();
}

to be used like:

enum Factory implements OperationProvider {

And now you can change your client code to do

OperationProvider ourProvider = ...
whatever.build();...

The point is: when using dependency injection, you simply pass a mocked OperationProvider. This means that you can basically sidestep the need to mock with your enum completely. You just have to make sure that within your production code, you obviously pass one of the enum constants when initializing that ourProvider field for example.