Support all restrictions for setRestriction

Refs #3f3366a15b082fcc4ac3efe02375a789bb3d18d9