If you want to keep the test simpler, with just one notification instead of two almost identical ones, then do the subscrition before creating the entity and remove the updateContext step (current step 3).