In recent versions of token simulation this changed to what we initially posted. You now have to mock the ElementColors service:
new BpmnJS({
additionalModules: [
bpmnTokenSimulationModule,
{
elementColors: [ 'value', {
add() { },
remove() { }
} ]
}
]
});