I have refactored https://github.com/heruan/testcase-entitygraph as unit test, as suggested.