Added tests for country code generation