1 files changed, 10 insertions, 0 deletions
diff --git a/spec/models/custom_field_spec.rb b/spec/models/custom_field_spec.rb
index b93d98187..80873683c 100644
--- a/spec/models/custom_field_spec.rb
+++ b/spec/models/custom_field_spec.rb
@@ -1,6 +1,7 @@
 require 'rails_helper'
 
 RSpec.describe CustomField, type: :model do
+  let( :vj ){ create :vehicle_journey, custom_field_values: {energy: 99} }
 
   context "validates" do
     it { should validate_uniqueness_of(:name).scoped_to(:resource_type) }
@@ -12,6 +13,15 @@ RSpec.describe CustomField, type: :model do
     it "option's values can be accessed by a key" do
       expect( custom_field.options['capacity'] ).to eq("0")
     end
+  end
+
+
+  context "custom fields for a resource" do
+    let!( :fields ){ (1..2).map{ create :custom_field } }
+    it { expect(vj.custom_fields).to eq(fields) }
+  end
 
+  context "custom field_values for a resource" do
+    it { expect(vj.custom_field_value("energy")).to eq(99) }
   end
 end