diff options
| -rw-r--r-- | app/models/chouette/area_type.rb | 2 | ||||
| -rw-r--r-- | spec/models/chouette/area_type_spec.rb | 4 |
2 files changed, 6 insertions, 0 deletions
diff --git a/app/models/chouette/area_type.rb b/app/models/chouette/area_type.rb index 43d96b391..4703ea646 100644 --- a/app/models/chouette/area_type.rb +++ b/app/models/chouette/area_type.rb @@ -13,6 +13,8 @@ class Chouette::AreaType @@instances = {} def self.find(code) + return unless code + code = code.to_sym @@instances[code] ||= new(code) if ALL.include? code end diff --git a/spec/models/chouette/area_type_spec.rb b/spec/models/chouette/area_type_spec.rb index 82c3f96bc..67d218df8 100644 --- a/spec/models/chouette/area_type_spec.rb +++ b/spec/models/chouette/area_type_spec.rb @@ -9,6 +9,10 @@ RSpec.describe Chouette::AreaType do end describe ".find" do + it "returns nil if the given code is nil" do + expect(Chouette::AreaType.find(nil)).to be_nil + end + it "returns nil if the given code is unknown" do expect(Chouette::AreaType.find('dummy')).to be_nil end |