enable auth in migration params schema (#1432)

threatgrid · Jun 26, 2024 · 43ca304 · 43ca304
1 parent b4ac7ee
commit 43ca304
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/src/ctia/task/migration/store.clj b/src/ctia/task/migration/store.clj
@@ -20,7 +20,8 @@
              [conn :as conn]
              [document :as ductile.doc]
              [index :as ductile.index]
-             [query :as ductile.query]]
+             [query :as ductile.query]
+             [schemas :refer [AuthParams]]]
             [ductile.schemas :refer [ESConn ESQuery Refresh]]
             [schema-tools.core :as st]
             [schema.core :as s]))
@@ -41,7 +42,8 @@
     :aliased  s/Bool
     :default_operator (s/enum "OR" "AND")
     :timeout s/Num
-    :version s/Num}))
+    :version s/Num
+    :auth AuthParams}))
 
 (s/defschema MigrationParams
   {:migration-id s/Str

diff --git a/test/ctia/task/migration/migrate_es_stores_test.clj b/test/ctia/task/migration/migrate_es_stores_test.clj
@@ -159,8 +159,11 @@
         (with-each-fixtures #(-> %
                                  (assoc-in [:ctia :store :es :investigation :indexname]
                                            investigation-indexname)
-                                 (assoc-in [:malware 0 :state :props :indexname]
-                                           malware-indexname))
+                                 (assoc-in [:malware 0 :state :props]
+                                           {:indexname malware-indexname
+                                            :auth {:type "basic"
+                                                   :params {:user "basic"
+                                                            :pwd "ductile"}}}))
           app
           (let [{:keys [get-in-config]} (helpers/get-service-map app :ConfigService)]
             (let [v (get-in-config [:ctia :store :es :investigation :indexname])]