Add validation for bus.id

semmin · Mar 3, 2012 · 84c221c · 84c221c
1 parent ccc62fb
commit 84c221c
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/lib/hrt_bus/bus.rb b/lib/hrt_bus/bus.rb
@@ -11,7 +11,7 @@ class Bus
     validates_numericality_of :lat, greater_than: 0
     validates_numericality_of :lon, less_than: 0
 
-    validates :time, :lat, :lon, :route_id, :presence => true
+    validates :id, :time, :lat, :lon, :route_id, :presence => true
 
     def initialize(attributes={})
       self.attributes = attributes

diff --git a/spec/factories/bus.rb b/spec/factories/bus.rb
@@ -2,6 +2,7 @@
 
 FactoryGirl.define do
   factory :bus, :class => HrtBus::Bus do
+    id      { SecureRandom.random_number(1e9.to_i) }
     time          { Time.now }
     lat           { 36.8642501 }
     lon           { -76.2818951 }

diff --git a/spec/models/bus_validations_spec.rb b/spec/models/bus_validations_spec.rb
@@ -6,6 +6,13 @@
 
     let(:bus) { Factory.build(:bus) }
 
+    it "should validate_presence_of :id" do
+      bus.should be_valid
+      bus.id = nil
+      bus.should_not be_valid
+      bus.errors.messages[:id].first.should match(/can't be blank/)
+    end
+
     it "should validate_presence_of :time" do
       bus.should be_valid
       bus.time = nil