Added tests for Country model