Fix transport modes and submodes modules

Fix Compliance Control Set flotted links to display on show page

4 files changed, 23 insertions, 24 deletions

diff --git a/app/models/chouette/line.rb b/app/models/chouette/line.rb
index 0139bb6a4..6f87a5d08 100644
--- a/app/models/chouette/line.rb
+++ b/app/models/chouette/line.rb
@@ -2,8 +2,8 @@ class Chouette::Line < Chouette::ActiveRecord
   include StifCodifligneAttributesSupport
   include LineRestrictions
   include LineReferentialSupport
-  extend StifTransportModeEnumerations
-  extend StifTransportSubmodeEnumerations
+  include StifTransportModeEnumerations
+  include StifTransportSubmodeEnumerations
 
   extend ActiveModel::Naming
 
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
diff --git a/app/views/compliance_control_sets/show.html.slim b/app/views/compliance_control_sets/show.html.slim
index 71b71f842..a6e2c192a 100644
--- a/app/views/compliance_control_sets/show.html.slim
+++ b/app/views/compliance_control_sets/show.html.slim
@@ -108,15 +108,4 @@
                 cls: 'table has-filter has-search',
                 model: ComplianceControl
 
-    .select_toolbox
-      ul
-        li.st_action.with_text
-          = link_to select_type_compliance_control_set_compliance_controls_path(@compliance_control_set.id)
-            span.fa.fa-plus
-            span
-              = t('compliance_control_sets.actions.add_compliance_control')
-        li.st_action.with_text
-          = link_to new_compliance_control_set_compliance_control_block_path(@compliance_control_set.id)
-            span.fa.fa-plus
-            span
-              = t('compliance_control_sets.actions.add_compliance_control_block')
+    = flotted_links @compliance_control_set.id