Add a test of the country model