added mapp_aggregate_type_to_event_store and mapp_aggregate_types_to_…

…event_store to event_stores

lib/sandthorn/event_stores.rb

@@ -40,6 +40,16 @@ def stores
       store_map.values
     end
 
+    def mapp_aggregate_type_to_event_store(aggregate_type, event_store)
+      aggregate_type.event_store(event_store)
+    end
+
+    def mapp_aggregate_types_to_event_store(array = [])
+      array.each do |item|
+        mapp_aggregate_type_to_event_store(item[:aggregate_type], item[:event_store]) if(item[:aggregate_type] && item[:event_store])
+      end
+    end
+
     private
 
     attr_reader :store_map

spec/event_stores_spec.rb

@@ -4,6 +4,13 @@ module Sandthorn
   describe EventStores do
     let(:stores) { EventStores.new }
 
+    before do
+      class AnAggregate 
+        include Sandthorn::AggregateRoot
+      end
+    end
+
+
     describe "#initialize" do
       context "when given a single event_store" do
         it "sets it as the default event store" do
@@ -77,5 +84,37 @@ module Sandthorn
         expect(stores[:foo]).to eq(store)
       end
     end
+
+    describe "#mapp_aggregate_type_to_event_store" do
+
+      let(:klass) {
+        class AnAggregate 
+          include Sandthorn::AggregateRoot
+        end
+      }
+
+      it "mapps the aggregate_type to the event_store" do
+        store = double
+        stores.add(:foo, store)
+        stores.mapp_aggregate_type_to_event_store(klass, :foo)
+        expect(klass.event_store).to eq(:foo)
+      end
+    end
+
+    describe "#mapp_aggregate_types_to_event_store" do
+
+      let(:klass) {
+        class AnAggregate 
+          include Sandthorn::AggregateRoot
+        end
+      }
+
+      it "mapps an array of aggregate_types to event_stores" do
+        store = double
+        stores.add(:foo, store)
+        stores.mapp_aggregate_types_to_event_store([{ aggregate_type: klass, event_store: :foo }])
+        expect(klass.event_store).to eq(:foo)
+      end
+    end
   end
 end

spec/get_events_spec.rb

@@ -6,7 +6,7 @@ class AnAggregate
 
 class AnotherAggregate
   include Sandthorn::AggregateRoot
-  event_store :other
+  event_store :should_override_this
 end
 
 describe Sandthorn do
@@ -66,6 +66,7 @@ def setup_secondary_db
     url = "sqlite://spec/db/other_db.sqlite3"
     driver = SandthornDriverSequel.driver_from_url(url: url)
     Sandthorn.event_stores.add(:other, driver)
+    Sandthorn.event_stores.mapp_aggregate_type_to_event_store(AnotherAggregate, :other)
     migrator = SandthornDriverSequel::Migration.new url: url
     SandthornDriverSequel.migrate_db url: url
     migrator.send(:clear_for_test)