Use ContextTypeAttribute during migration discovery

/cc @rowanmiller & @emilcicos 

I also did some manual bashing on this for additional functional verification.
I would prefer to have the contextType parameter after model in all the methods below, because model is the primary input.