Refactored code

faker-ruby · Jul 30, 2020 · 8b8b3e7 · 8b8b3e7
1 parent ee2dc2d
commit 8b8b3e7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/lib/faker/default/driving_licence.rb b/lib/faker/default/driving_licence.rb
@@ -79,7 +79,7 @@ def uk_driving_licence(*args)
       end
 
       def usa_driving_licence(state = 'CA')
-        bothify(fetch("driving_licence.usa_state_formats.#{state.upcase}"))
+        bothify(fetch("driving_licence.usa_state_formats.#{state.to_s.upcase}"))
       rescue I18n::MissingTranslationData => _e
         raise InvalidStatePassed, "Invalid state code passed for USA, '#{state}'"
       end

diff --git a/test/faker/default/test_faker_driving_licence.rb b/test/faker/default/test_faker_driving_licence.rb
@@ -84,7 +84,11 @@ def test_usa_driving_licence_for_different_states
 
   def test_usa_driving_licence_with_faker_code
     assert_raises(Faker::InvalidStatePassed) do
-      @tester.usa_driving_licence('123')
+      @tester.usa_driving_licence('abc')
+    end
+
+    assert_raises(Faker::InvalidStatePassed) do
+      @tester.usa_driving_licence(123)
     end
   end
 end