code support for organization formatting

GitHub Issue #34

Author: cainlevy

lib/snail.rb

@@ -45,6 +45,9 @@ def self.lookup_country_iso(val)
   attr_accessor :name
   alias full_name= name=
 
+  attr_accessor :organization
+  alias company= organization=
+
   attr_accessor :line_1
   alias street=   line_1=
   alias street_1= line_1=
@@ -89,13 +92,14 @@ def international?
   def to_s(with_country: nil)
     with_country = true if with_country.nil? && international?
 
-    [
-      name,
-      line_1,
-      line_2,
-      city_line,
-      (country_line if with_country)
-    ]
+    (
+      recipient + [
+        line_1,
+        line_2,
+        city_line,
+        (country_line if with_country)
+      ]
+    )
       .reject { |line| line.nil? || line.empty? }
       .join("\n")
   end
@@ -164,4 +168,14 @@ def translated_country(origin, country)
     path = File.join(File.dirname(File.expand_path(__FILE__)), "../assets/#{origin}.yml")
     File.read(path).match(/^#{country}: (.*)$/)[1]
   end
+
+  # NOTE: this is likely an incomplete list. please contribute!
+  private def recipient
+    case country
+    when 'NL'
+      [organization, name]
+    else
+      [name, organization]
+    end
+  end
 end

test/snail_test.rb

@@ -164,6 +164,16 @@ def setup
     s.name = 'John & Jane Doe'
     assert_equal "John &amp; Jane Doe<br />12345 5th St<br />Somewheres, Dublin<br />IRELAND", s.to_html
   end
+
+  test 'name and organization in united states' do
+    s = Snail.new(country: 'US', name: 'NAME', organization: 'ORGANIZATION', city: 'CITY', region: 'STATE', postal_code: 'CODE')
+    assert_equal "NAME\nORGANIZATION\nCITY STATE  CODE", s.to_s
+  end
+
+  test 'name and organization in netherlands' do
+    s = Snail.new(country: 'NL', name: 'NAME', organization: 'ORGANIZATION', city: 'CITY', postal_code: 'CODE')
+    assert_equal "ORGANIZATION\nNAME\nCODE  CITY\nNETHERLANDS", s.to_s
+  end
 end