Finally found a way to test the interaction with entity contexts, thanks to https://gist.github.com/troufster