I don't see why armor restrictions need to be in place when cosmetic sets are a thing.
Yes you could make your death knight wear mage mogs, but you could honestly do that now as well with the Kirin Tor set.
Players naturally gravitate towards making their class look like their class. And having access to every armor piece in the game would just make it more pronounced i feel.