Experimental Async Refreshes API (#34918)

spec/models/async_refresh_spec.rb

@@ -0,0 +1,174 @@
+# frozen_string_literal: true
+
+require 'rails_helper'
+
+RSpec.describe AsyncRefresh do
+  subject { described_class.new(redis_key) }
+
+  let(:redis_key) { 'testjob:key' }
+  let(:status) { 'running' }
+  let(:job_hash) { { 'status' => status, 'result_count' => 23 } }
+
+  describe '::find' do
+    context 'when a matching job in redis exists' do
+      before do
+        redis.hset(redis_key, job_hash)
+      end
+
+      it 'returns a new instance' do
+        id = Rails.application.message_verifier('async_refreshes').generate(redis_key)
+        async_refresh = described_class.find(id)
+
+        expect(async_refresh).to be_a described_class
+      end
+    end
+
+    context 'when no matching job in redis exists' do
+      it 'returns `nil`' do
+        id = Rails.application.message_verifier('async_refreshes').generate('non_existent')
+        expect(described_class.find(id)).to be_nil
+      end
+    end
+  end
+
+  describe '::create' do
+    it 'inserts the given key into redis' do
+      described_class.create(redis_key)
+
+      expect(redis.exists?(redis_key)).to be true
+    end
+
+    it 'sets the status to `running`' do
+      async_refresh = described_class.create(redis_key)
+
+      expect(async_refresh.status).to eq 'running'
+    end
+
+    context 'with `count_results`' do
+      it 'set `result_count` to 0' do
+        async_refresh = described_class.create(redis_key, count_results: true)
+
+        expect(async_refresh.result_count).to eq 0
+      end
+    end
+
+    context 'without `count_results`' do
+      it 'does not set `result_count`' do
+        async_refresh = described_class.create(redis_key)
+
+        expect(async_refresh.result_count).to be_nil
+      end
+    end
+  end
+
+  describe '#id' do
+    before do
+      redis.hset(redis_key, job_hash)
+    end
+
+    it "returns a signed version of the job's redis key" do
+      id = subject.id
+      key_name = Base64.decode64(id.split('-').first)
+
+      expect(key_name).to include redis_key
+    end
+  end
+
+  describe '#status' do
+    before do
+      redis.hset(redis_key, job_hash)
+    end
+
+    context 'when the job is running' do
+      it "returns 'running'" do
+        expect(subject.status).to eq 'running'
+      end
+    end
+
+    context 'when the job is finished' do
+      let(:status) { 'finished' }
+
+      it "returns 'finished'" do
+        expect(subject.status).to eq 'finished'
+      end
+    end
+  end
+
+  describe '#running?' do
+    before do
+      redis.hset(redis_key, job_hash)
+    end
+
+    context 'when the job is running' do
+      it 'returns `true`' do
+        expect(subject.running?).to be true
+      end
+    end
+
+    context 'when the job is finished' do
+      let(:status) { 'finished' }
+
+      it 'returns `false`' do
+        expect(subject.running?).to be false
+      end
+    end
+  end
+
+  describe '#finished?' do
+    before do
+      redis.hset(redis_key, job_hash)
+    end
+
+    context 'when the job is running' do
+      it 'returns `false`' do
+        expect(subject.finished?).to be false
+      end
+    end
+
+    context 'when the job is finished' do
+      let(:status) { 'finished' }
+
+      it 'returns `true`' do
+        expect(subject.finished?).to be true
+      end
+    end
+  end
+
+  describe '#finish!' do
+    before do
+      redis.hset(redis_key, job_hash)
+    end
+
+    it 'sets the status to `finished`' do
+      subject.finish!
+
+      expect(subject).to be_finished
+    end
+  end
+
+  describe '#result_count' do
+    before do
+      redis.hset(redis_key, job_hash)
+    end
+
+    it 'returns the result count from redis' do
+      expect(subject.result_count).to eq 23
+    end
+  end
+
+  describe '#reload' do
+    before do
+      redis.hset(redis_key, job_hash)
+    end
+
+    it 'reloads the current data from redis and returns itself' do
+      expect(subject).to be_running
+      redis.hset(redis_key, { 'status' => 'finished' })
+      expect(subject).to be_running
+
+      expect(subject.reload).to eq subject
+
+      expect(subject).to be_finished
+    end
+  end
+end

spec/models/home_feed_spec.rb

@@ -32,7 +32,7 @@ RSpec.describe HomeFeed do
 
     context 'when feed is being generated' do
       before do
-        redis.set("account:#{account.id}:regeneration", true)
+        redis.hset("account:#{account.id}:regeneration", { 'status' => 'running' })
       end
 
       it 'returns nothing' do
@@ -44,9 +44,19 @@ RSpec.describe HomeFeed do
   end
 
   describe '#regenerating?' do
+    context 'when an old-style string key is still in use' do
+      it 'upgrades the key to a hash' do
+        redis.set("account:#{account.id}:regeneration", true)
+
+        expect(subject.regenerating?).to be true
+
+        expect(redis.type("account:#{account.id}:regeneration")).to eq 'hash'
+      end
+    end
+
     context 'when feed is being generated' do
       before do
-        redis.set("account:#{account.id}:regeneration", true)
+        redis.hset("account:#{account.id}:regeneration", { 'status' => 'running' })
       end
 
       it 'returns `true`' do
@@ -55,13 +65,35 @@ RSpec.describe HomeFeed do
     end
 
     context 'when feed is not being generated' do
-      it 'returns `false`' do
-        expect(subject.regenerating?).to be false
+      context 'when the job is marked as finished' do
+        before do
+          redis.hset("account:#{account.id}:regeneration", { 'status' => 'finished' })
+        end
+
+        it 'returns `false`' do
+          expect(subject.regenerating?).to be false
+        end
+      end
+
+      context 'when the job key is missing' do
+        it 'returns `false`' do
+          expect(subject.regenerating?).to be false
+        end
       end
     end
   end
 
   describe '#regeneration_in_progress!' do
+    context 'when an old-style string key is still in use' do
+      it 'upgrades the key to a hash' do
+        redis.set("account:#{account.id}:regeneration", true)
+
+        subject.regeneration_in_progress!
+
+        expect(redis.type("account:#{account.id}:regeneration")).to eq 'hash'
+      end
+    end
+
     it 'sets the corresponding key in redis' do
       expect(redis.exists?("account:#{account.id}:regeneration")).to be false
 
@@ -72,12 +104,22 @@ RSpec.describe HomeFeed do
   end
 
   describe '#regeneration_finished!' do
-    it 'removes the corresponding key from redis' do
-      redis.set("account:#{account.id}:regeneration", true)
+    context 'when an old-style string key is still in use' do
+      it 'upgrades the key to a hash' do
+        redis.set("account:#{account.id}:regeneration", true)
+
+        subject.regeneration_finished!
+
+        expect(redis.type("account:#{account.id}:regeneration")).to eq 'hash'
+      end
+    end
+
+    it "sets the corresponding key's status to 'finished'" do
+      redis.hset("account:#{account.id}:regeneration", { 'status' => 'running' })
 
       subject.regeneration_finished!
 
-      expect(redis.exists?("account:#{account.id}:regeneration")).to be false
+      expect(redis.hget("account:#{account.id}:regeneration", 'status')).to eq 'finished'
     end
   end
 end