Stay up to date with master branch

3 files changed, 21 insertions, 11 deletions

diff --git a/app/models/chouette/vehicle_journey.rb b/app/models/chouette/vehicle_journey.rb
index c8754e958..329b37a6c 100644
--- a/app/models/chouette/vehicle_journey.rb
+++ b/app/models/chouette/vehicle_journey.rb
@@ -3,7 +3,7 @@ module Chouette
     include ChecksumSupport
     include VehicleJourneyRestrictions
     include ObjectidSupport
-    extend StifTransportModeEnumerations
+    include StifTransportModeEnumerations
     # FIXME http://jira.codehaus.org/browse/JRUBY-6358
     self.primary_key = "id"
 
diff --git a/app/models/concerns/stif_transport_mode_enumerations.rb b/app/models/concerns/stif_transport_mode_enumerations.rb
index 74621dfc5..7c72d5468 100644
--- a/app/models/concerns/stif_transport_mode_enumerations.rb
+++ b/app/models/concerns/stif_transport_mode_enumerations.rb
@@ -1,5 +1,15 @@
 module StifTransportModeEnumerations
   extend ActiveSupport::Concern
+  
+  included do
+    extend Enumerize
+    enumerize :transport_mode, in: StifTransportModeEnumerations.transport_modes
+  end
+
+  module ClassMethods
+    def transport_modes; StifTransportModeEnumerations.transport_modes end
+    def sorted_transport_modes; StifTransportModeEnumerations.sorted_transport_modes end
+  end
 
   class << self
     def transport_modes
@@ -9,10 +19,5 @@ module StifTransportModeEnumerations
       transport_modes.sort_by{|m| I18n.t("enumerize.transport_mode.#{m}").parameterize }
     end
   end
-  
-  included do
-    extend Enumerize
-    enumerize :transport_mode, in: StifTransportModeEnumerations.transport_modes
-  end
 
 end
diff --git a/app/models/concerns/stif_transport_submode_enumerations.rb b/app/models/concerns/stif_transport_submode_enumerations.rb
index eb3e56eac..363faadf0 100644
--- a/app/models/concerns/stif_transport_submode_enumerations.rb
+++ b/app/models/concerns/stif_transport_submode_enumerations.rb
@@ -1,6 +1,16 @@
 module StifTransportSubmodeEnumerations
   extend ActiveSupport::Concern
 
+  included do
+    extend Enumerize
+    enumerize :transport_submode, in: StifTransportSubmodeEnumerations.transport_submodes
+  end
+
+  module ClassMethods
+    def transport_submodes; StifTransportSubmodeEnumerations.transport_submodes end
+    def sorted_transport_submodes; StifTransportSubmodeEnumerations.sorted_transport_submodes end
+  end
+
   class << self
     def transport_submodes
       %w(
@@ -19,10 +29,5 @@ module StifTransportSubmodeEnumerations
       transport_submodes.sort_by{|m| I18n.t("enumerize.transport_submode.#{m}").parameterize }
     end
   end
-
-  included do
-    extend Enumerize
-    enumerize :transport_submode, in: StifTransportSubmodeEnumerations.transport_submodes
-  end
   
 end