enterprise-registration: fail-safe improvement; region= can receive region or id

Rodrigo Souto
1 parent f0f667bb
Showing 1 changed file with 7 additions and 1 deletions Show diff stats
app/models/create_enterprise.rb
@@ -73,7 +73,13 @@ class CreateEnterprise &lt; Task
  
   # sets the associated region for the enterprise creation
   def region=(value)
-    raise ArgumentError.new("Region expected, but got #{value.class}") unless value.kind_of?(Region)
+    unless value.kind_of?(Region)
+      begin
+        value = Region.find(value)
+      rescue
+        raise ArgumentError.new("Could not find any region with the id #{value}")
+      end
+    end
  
     @region = value
     self.region_id = value.id
...	...	@@ -73,7 +73,13 @@ class CreateEnterprise < Task
73	73
74	74	# sets the associated region for the enterprise creation
75	75	def region=(value)
76		- raise ArgumentError.new("Region expected, but got #{value.class}") unless value.kind_of?(Region)
	76	+ unless value.kind_of?(Region)
	77	+ begin
	78	+ value = Region.find(value)
	79	+ rescue
	80	+ raise ArgumentError.new("Could not find any region with the id #{value}")
	81	+ end
	82	+ end
77	83
78	84	@region = value
79	85	self.region_id = value.id
...	...