From 4764d2cdc34e44a0040115b99e8984aea5a6b569 Mon Sep 17 00:00:00 2001
From: cedricnjanga
Date: Fri, 29 Sep 2017 15:32:12 +0200
Subject: Add transport_submode attribute to ComplianceControlBlock +
 validation on transport mode presence

---
 app/models/compliance_control_block.rb | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

(limited to 'app/models/compliance_control_block.rb')

diff --git a/app/models/compliance_control_block.rb b/app/models/compliance_control_block.rb
index e03c4ce85..7213e0812 100644
--- a/app/models/compliance_control_block.rb
+++ b/app/models/compliance_control_block.rb
@@ -2,9 +2,17 @@ class ComplianceControlBlock < ActiveRecord::Base
   belongs_to :compliance_control_set
   has_many :compliance_controls, dependent: :destroy
 
-  hstore_accessor :condition_attributes, transport_mode: :string
+  hstore_accessor :condition_attributes, 
+    transport_mode: :string, 
+    transport_sub_mode: :string
+
+    validates_presence_of :transport_mode
 
   def self.transport_modes
-    ["all"] + StifTransportModeEnumerations.transport_mode.values
+    StifTransportModeEnumerations.transport_mode.options
+  end
+
+  def self.transport_sub_modes
+    StifTransportSubmodeEnumerations.transport_submode.options
   end
 end
-- 
cgit v1.2.3