test/answer/parser/whois.nic.hu_test.rb in whois-1.1.1 vs test/answer/parser/whois.nic.hu_test.rb in whois-1.1.2

@@ -136,10 +136,11 @@
   def test_registrant_contact_as_company
     parser    = @klass.new(load_part('/property_registrant_as_company.txt'))
     result    = parser.registrant_contact
 
     assert_instance_of Whois::Answer::Contact,  result
+    assert_equal Whois::Answer::Contact::TYPE_REGISTRANT, result.type
     assert_equal 'Google, Inc.',                result.name
     assert_equal 'Google, Inc.',                result.organization
     assert_equal 'Amphitheatre Pkwy 1600.',     result.address
     assert_equal 'CA-94043',                    result.zip
     assert_equal 'Mountain View',               result.city
@@ -151,10 +152,11 @@
   def test_registrant_contact_as_private_person
     parser    = @klass.new(load_part('/property_registrant_as_private_person.txt'))
     result    = parser.registrant_contact
 
     assert_instance_of Whois::Answer::Contact,  result
+    assert_equal Whois::Answer::Contact::TYPE_REGISTRANT, result.type
     assert_match /Buruzs/,                      result.name             # UTF-8 hack
     assert_equal nil,                           result.organization
     assert_equal nil,                           result.address
     assert_equal nil,                           result.zip
     assert_equal nil,                           result.city
@@ -172,20 +174,21 @@
     assert_equal nil, result.city
     assert_equal nil, result.country_code
   end
 
   def test_admin_contact_with_registered
-    admin = @klass.new(load_part('/registered.txt')).admin_contact
-    assert_instance_of Whois::Answer::Contact, admin
-    assert_equal '2000466366', admin.id
-    assert_equal '3C Kft. (Registrar)', admin.name
-    assert_equal 'Konkoly Thege út 29-33.', admin.address
-    assert_equal 'H-1121', admin.zip
-    assert_equal 'Budapest', admin.city
-    assert_equal 'HU', admin.country_code
-    assert_equal '[phone]', admin.phone
-    assert_equal '[phone]', admin.fax
+    result = @klass.new(load_part('/registered.txt')).admin_contact
+    assert_instance_of Whois::Answer::Contact,    result
+    assert_equal Whois::Answer::Contact::TYPE_ADMIN, result.type
+    assert_equal '2000466366',                    result.id
+    assert_equal '3C Kft. (Registrar)',           result.name
+    assert_equal 'Konkoly Thege út 29-33.',       result.address
+    assert_equal 'H-1121',                        result.zip
+    assert_equal 'Budapest',                      result.city
+    assert_equal 'HU',                            result.country_code
+    assert_equal '[phone]',               result.phone
+    assert_equal '[phone]',               result.fax
   end
 
   def test_admin_contact_with_unregistered
     parser      = @klass.new(load_part('/available.txt'))
     expected    = nil
@@ -197,20 +200,21 @@
     assert_equal  expected, parser.admin_contact
     assert_equal  expected, parser.instance_eval { @admin_contact }
   end
 
   def test_technical_contact_with_registered
-    technical = @klass.new(load_part('/registered.txt')).technical_contact
-    assert_instance_of Whois::Answer::Contact, technical
-    assert_equal '2000578125', technical.id
-    assert_equal 'Markmonitor', technical.name
-    assert_equal 'Overland Road 10400, PMB155', technical.address
-    assert_equal 'ID-83709', technical.zip
-    assert_equal 'Boise', technical.city
-    assert_equal 'US', technical.country_code
-    assert_equal '+ 1 208 389 5798', technical.phone
-    assert_equal '+ 1 208 389 5771', technical.fax
-    assert_equal '[e-mail]', technical.email
+    result = @klass.new(load_part('/registered.txt')).technical_contact
+    assert_instance_of Whois::Answer::Contact,    result
+    assert_equal Whois::Answer::Contact::TYPE_TECHNICAL, result.type
+    assert_equal '2000578125',                    result.id
+    assert_equal 'Markmonitor',                   result.name
+    assert_equal 'Overland Road 10400, PMB155',   result.address
+    assert_equal 'ID-83709',                      result.zip
+    assert_equal 'Boise',                         result.city
+    assert_equal 'US',                            result.country_code
+    assert_equal '+ 1 208 389 5798',              result.phone
+    assert_equal '+ 1 208 389 5771',              result.fax
+    assert_equal '[e-mail]',         result.email
   end
 
   def test_technical_contact_with_unregistered
     parser      = @klass.new(load_part('/available.txt'))
     expected    = nil
\ No newline at end of file