From 6579ea9269d9def65e544357d843ebf1f73cbb15 Mon Sep 17 00:00:00 2001
From: Craig Reese <109101548+craigrva@users.noreply.github.com>
Date: Tue, 21 May 2024 16:20:25 -0500
Subject: [PATCH] Craig/appeals 46196 (#21689)

* fix query, tested locally

* add basic test to verify csv downloads aren't broken
---
 ...se_distribution_levers_tests_controller.rb |  1 +
 app/models/vacols/case_docket.rb              |  3 ++-
 ...se_distribution_levers_tests_controller.rb | 26 +++++++++++++++++++
 3 files changed, 29 insertions(+), 1 deletion(-)
 create mode 100644 spec/controllers/case_distribution_levers_tests_controller.rb

diff --git a/app/controllers/case_distribution_levers_tests_controller.rb b/app/controllers/case_distribution_levers_tests_controller.rb
index f9be94cfeb6..6b452ff1998 100644
--- a/app/controllers/case_distribution_levers_tests_controller.rb
+++ b/app/controllers/case_distribution_levers_tests_controller.rb
@@ -57,6 +57,7 @@ def appeals_distributed
 
   def check_environment
     return true if Rails.env.development?
+    return true if Rails.env.test?
     return true if Rails.deploy_env?(:demo)
 
     redirect_to "/unauthorized"
diff --git a/app/models/vacols/case_docket.rb b/app/models/vacols/case_docket.rb
index 5120733a0ed..f55307d65dc 100644
--- a/app/models/vacols/case_docket.rb
+++ b/app/models/vacols/case_docket.rb
@@ -85,7 +85,8 @@ class DocketNumberCentennialLoop < StandardError; end
 
   # this version of the query should not be used during distribution it is only intended for reporting usage
   SELECT_READY_APPEALS_ADDITIONAL_COLS = "
-    select BFKEY, BFD19, BFDLOOUT, BFMPRO, BFCURLOC, BFAC, BFHINES, TINUM, TITRNUM, AOD, BFCORKEY, BFCORLID
+    select BFKEY, BFD19, BFDLOOUT, BFMPRO, BFCURLOC, BFAC, BFHINES, TINUM, TITRNUM, AOD, BFMEMID, BFDPDCN,
+    BFCORKEY, BFCORLID
     #{FROM_READY_APPEALS}
   "
 
diff --git a/spec/controllers/case_distribution_levers_tests_controller.rb b/spec/controllers/case_distribution_levers_tests_controller.rb
new file mode 100644
index 00000000000..a8dca5fe726
--- /dev/null
+++ b/spec/controllers/case_distribution_levers_tests_controller.rb
@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+describe CaseDistributionLeversTestsController do
+  before do
+    Timecop.freeze(Time.utc(2024, 1, 1, 12, 0, 0))
+    User.authenticate!(user: User.system_user)
+  end
+
+  context "#appeals_ready_to_distribute" do
+    it "downloads a properly named CSV file" do
+      get :appeals_ready_to_distribute, format: :csv
+
+      expect(response.headers["Content-Type"]).to eq("text/csv")
+      expect(response.headers["Content-Disposition"]).to include("appeals_ready_to_distribute_20240101-0700.csv")
+    end
+  end
+
+  context "#appeals_distributed" do
+    it "downloads a properly named CSV file" do
+      get :appeals_distributed, format: :csv
+
+      expect(response.headers["Content-Type"]).to eq("text/csv")
+      expect(response.headers["Content-Disposition"]).to include("distributed_appeals_20240101-0700.csv")
+    end
+  end
+end