1 files changed, 57 insertions, 0 deletions
diff --git a/spec/models/user_custom_attribute_spec.rb b/spec/models/user_custom_attribute_spec.rb
index 1a51ad662b0..67c144d7caa 100644
--- a/spec/models/user_custom_attribute_spec.rb
+++ b/spec/models/user_custom_attribute_spec.rb
@@ -15,4 +15,61 @@ RSpec.describe UserCustomAttribute do
     it { is_expected.to validate_presence_of(:value) }
     it { is_expected.to validate_uniqueness_of(:key).scoped_to(:user_id) }
   end
+
+  describe 'scopes' do
+    let(:user) { create(:user) }
+    let(:blocked_at) { DateTime.now }
+    let(:custom_attribute) { create(:user_custom_attribute, key: 'blocked_at', value: blocked_at, user_id: user.id) }
+
+    describe '.by_user_id' do
+      subject { UserCustomAttribute.by_user_id(user.id) }
+
+      it { is_expected.to match_array([custom_attribute]) }
+    end
+
+    describe '.by_updated_at' do
+      subject { UserCustomAttribute.by_updated_at(Date.today.all_day) }
+
+      it { is_expected.to match_array([custom_attribute]) }
+    end
+
+    describe '.by_key' do
+      subject { UserCustomAttribute.by_key('blocked_at') }
+
+      it { is_expected.to match_array([custom_attribute]) }
+    end
+  end
+
+  describe '#upsert_custom_attributes' do
+    subject { UserCustomAttribute.upsert_custom_attributes(custom_attributes) }
+
+    let_it_be_with_reload(:user) { create(:user) }
+
+    let(:arkose_session) { '22612c147bb418c8.2570749403' }
+    let(:risk_band) { 'Low' }
+    let(:global_score) { '0' }
+    let(:custom_score) { '0' }
+
+    let(:custom_attributes) do
+      custom_attributes = []
+      custom_attributes.push({ key: 'arkose_session', value: arkose_session })
+      custom_attributes.push({ key: 'arkose_risk_band', value: risk_band })
+      custom_attributes.push({ key: 'arkose_global_score', value: global_score })
+      custom_attributes.push({ key: 'arkose_custom_score', value: custom_score })
+
+      custom_attributes.map! { |custom_attribute| custom_attribute.merge({ user_id: user.id }) }
+      custom_attributes
+    end
+
+    it 'adds arkose data to custom attributes' do
+      subject
+
+      expect(user.custom_attributes.count).to eq(4)
+
+      expect(user.custom_attributes.find_by(key: 'arkose_session').value).to eq(arkose_session)
+      expect(user.custom_attributes.find_by(key: 'arkose_risk_band').value).to eq(risk_band)
+      expect(user.custom_attributes.find_by(key: 'arkose_global_score').value).to eq(global_score)
+      expect(user.custom_attributes.find_by(key: 'arkose_custom_score').value).to eq(custom_score)
+    end
+  end
 end