1 files changed, 44 insertions, 0 deletions
diff --git a/spec/models/raw_usage_data_spec.rb b/spec/models/raw_usage_data_spec.rb
new file mode 100644
index 00000000000..c10db63da56
--- /dev/null
+++ b/spec/models/raw_usage_data_spec.rb
@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe RawUsageData do
+  describe 'validations' do
+    it { is_expected.to validate_presence_of(:payload) }
+    it { is_expected.to validate_presence_of(:recorded_at) }
+
+    context 'uniqueness validation' do
+      let!(:existing_record) { create(:raw_usage_data) }
+
+      it { is_expected.to validate_uniqueness_of(:recorded_at) }
+    end
+
+    describe '#update_sent_at!' do
+      let(:raw_usage_data) { create(:raw_usage_data) }
+
+      context 'with save_raw_usage_data feature enabled' do
+        before do
+          stub_feature_flags(save_raw_usage_data: true)
+        end
+
+        it 'updates sent_at' do
+          raw_usage_data.update_sent_at!
+
+          expect(raw_usage_data.sent_at).not_to be_nil
+        end
+      end
+
+      context 'with save_raw_usage_data feature disabled' do
+        before do
+          stub_feature_flags(save_raw_usage_data: false)
+        end
+
+        it 'updates sent_at' do
+          raw_usage_data.update_sent_at!
+
+          expect(raw_usage_data.sent_at).to be_nil
+        end
+      end
+    end
+  end
+end