AE-2362: Have the public columns as hardcoded values inside the test …

…clause
solita · Jan 17, 2025 · 690efb7 · 690efb7
1 parent 3bcb77e
commit 690efb7
Showing 1 changed file with 16 additions and 12 deletions.
diff --git a/etp-core/etp-backend/src/test/clj/solita/etp/service/csv_to_s3_test.clj b/etp-core/etp-backend/src/test/clj/solita/etp/service/csv_to_s3_test.clj
@@ -8,8 +8,6 @@
             [clojure.set :as set]
             [solita.etp.test-data.generators :as generators]
             [solita.etp.test-system :as ts]
-            [solita.etp.service.energiatodistus-csv :refer [public-columns]]
-            [solita.etp.service.aineisto-test :as aineistot-test]
             [solita.etp.test-data.laatija :as test-data.laatija]
             [solita.etp.test-data.energiatodistus :as test-data.energiatodistus]))
 
@@ -26,13 +24,6 @@
        (map column-ks->str)
        set))
 
-(def hidden-columns
-  #{[:tila-id]
-    [:korvaava-energiatodistus-id]
-    [:laatija-id]
-    [:laatija-fullname]
-    [:perustiedot :yritys :nimi]})
-
 (t/deftest test-public-csv-to-s3
   (t/testing "Public csv doesn't exist before generating"
     (t/is (false? (file/file-exists? ts/*aws-s3-client* csv-to-s3/public-csv-key))))
@@ -55,9 +46,22 @@
     (t/is (true? (file/file-exists? ts/*aws-s3-client* csv-to-s3/public-csv-key))
           "CSV should exist after generation.")
 
-    (let [[header & _] (get-first-three-lines-from-csv csv-to-s3/public-csv-key)
-          expected-headers (columns->header-strings public-columns)
-          header-set (set header)]
+    (let [[headers & _] (get-first-three-lines-from-csv csv-to-s3/public-csv-key)
+          extra-columns #{[:perustiedot :alakayttotarkoitus-fi]
+                          [:tulokset :e-luokka-rajat :kayttotarkoitus :label-fi]
+                          [:tulokset :e-luokka-rajat :raja-uusi-2018]
+                          [:tulokset :kaytettavat-energiamuodot :kaukolampo-kerroin]
+                          [:tulokset :kaytettavat-energiamuodot :sahko-kerroin]
+                          [:tulokset :kaytettavat-energiamuodot :uusiutuva-polttoaine-kerroin]
+                          [:tulokset :kaytettavat-energiamuodot :fossiilinen-polttoaine-kerroin]
+                          [:tulokset :kaytettavat-energiamuodot :kaukojaahdytys-kerroin]}
+          hidden-columns #{[:laatija-fullname]
+                           [:tila-id]
+                           [:korvaava-energiatodistus-id]
+                           [:laatija-id]
+                           [:perustiedot :yritys :nimi]}
+          expected-headers (columns->header-strings extra-columns)
+          header-set (set headers)]
 
       ;; Verify that all expected headers from public-columns are present in the CSV
       (t/is (set/subset? expected-headers header-set)