lib/kb/fake/api.rb in barkibu-kb-0.16.2 vs lib/kb/fake/api.rb in barkibu-kb-0.17.0

@@ -6,23 +6,25 @@
 require 'kb/fake/bounded_context/pet_family/hubspot_relationship'
 
 module KB
   module Fake
     class ApiState
-      attr_accessor :petparents, :pets, :consultations, :petcontracts, :plans, :breeds, :products, :hubspot_relationship
+      attr_accessor :petparents, :pets, :consultations, :petcontracts, :plans, :breeds, :products,
+                    :hubspot_relationship, :referrals
 
       # rubocop:disable Metrics/ParameterLists
       def initialize(petparents: [], pets: [], consultations: [], petcontracts: [], plans: [], breeds: [],
-                     products: [], hubspot_relationship: [])
+                     products: [], hubspot_relationship: [], referrals: [])
         @petparents = petparents
         @pets = pets
         @consultations = consultations
         @petcontracts = petcontracts
         @plans = plans
         @breeds = breeds
         @products = products
         @hubspot_relationship = hubspot_relationship
+        @referrals = referrals
       end
       # rubocop:enable Metrics/ParameterLists
 
       def to_snapshot
         {
@@ -31,10 +33,11 @@
           consultations: @consultations.clone,
           petcontracts: @petcontracts.clone,
           plans: @plans.clone,
           breeds: @breeds.clone,
           products: @products.clone,
-          hubspot_relationship: @hubspot_relationship.clone
+          hubspot_relationship: @hubspot_relationship.clone,
+          referrals: @referrals.clone
         }
       end
     end
 
     class Api < Sinatra::Base