add sleep to storage_spec test in analytics

amplitude · Aug 18, 2023 · f79a95c · f79a95c
1 parent 724a7be
commit f79a95c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/spec/amplitude/storage_spec.rb b/spec/amplitude/storage_spec.rb
@@ -4,9 +4,9 @@ module AmplitudeAnalytics
       @provider = InMemoryStorageProvider.new
       @storage = @provider.storage
       @workers = Workers.new
-      config = Config.new
-      @storage.setup(config, @workers)
-      @workers.setup(config, @storage)
+      @config = Config.new
+      @storage.setup(@config, @workers)
+      @workers.setup(@config, @storage)
     end
 
     it 'checks if storage is empty after pull' do
@@ -68,6 +68,7 @@ module AmplitudeAnalytics
 
     it 'exceeds max capacity and fails' do
       push_event(@storage, Set.new, MAX_BUFFER_CAPACITY)
+      sleep(1)
       expect(@storage.total_events).to eq(MAX_BUFFER_CAPACITY)
 
       event = BaseEvent.new('test_event', user_id: 'test_user')

diff --git a/spec/amplitude/worker_spec.rb b/spec/amplitude/worker_spec.rb
@@ -257,7 +257,6 @@ module AmplitudeAnalytics
         while @workers.storage.total_events > 0
           sleep(0.1)
         end
-        sleep(5)
         total_events = @events_dict.values.sum(&:length)
         expect(@workers.storage.total_events).to eq(0)
         expect(total_events).to eq(5000)