Add FederalState model tests