;; This Source Code Form is subject to the terms of the Mozilla Public
;; License, v. 2.0. If a copy of the MPL was not distributed with this
;; file, You can obtain one at http://mozilla.org/MPL/2.0/.
;;
;; Copyright (c) KALEIDOS SUBSIDIARY SL

(ns performance-tests.render-wasm.runner-test
  (:require
   [benches.render-wasm.node.options :as options]
   [benches.render-wasm.node.runner :as runner]
   [cljs.test :as t :include-macros true]
   [clojure.string :as str]))

(t/deftest output-options-apply-to-every-command
  (doseq [args [["build"] ["run"] ["purge"] ["compare" "base" "candidate"]
                ["merge" "base" "additional"]
                ["ab" "--features" "a" "--filter" "rects/load"]]]
    (t/is (= :info (options/console-level (:options (runner/parse-args args)))))
    (t/is (= :auto (options/color-mode (:options (runner/parse-args args)))))
    (t/is (= :warn (options/console-level (:options (runner/parse-args (conj args "-q"))))))
    (t/is (= :debug (options/console-level (:options (runner/parse-args (conj args "-v"))))))
    (t/is (= :always (options/color-mode (:options (runner/parse-args (conj args "--color"))))))
    (t/is (= :never (options/color-mode (:options (runner/parse-args (conj args "--no-color")))))))
  (t/is (thrown-with-msg? js/Error #"mutually exclusive"
                          (runner/parse-args ["run" "-q" "-v"]))))

(t/deftest build-parses-defaults-and-explicit-functional-feature-selection
  (let [{:keys [command options]} (runner/parse-args ["build"])]
    (t/is (= "build" command))
    (t/is (true? (:default-features options)))
    (t/is (= options/default-build-timeout-ms (:build-timeout-ms options))))
  (let [{:keys [command options]} (runner/parse-args
                                   ["--" "build" "--no-default-features"
                                    "--features" "functional_b,functional-a,functional_b"
                                    "--build-timeout-ms" "1250"])]
    (t/is (= "build" command))
    (t/is (= {:default-features false
              :features ["functional-a" "functional_b"]
              :build-timeout-ms 1250}
             (select-keys options [:default-features :features :build-timeout-ms])))))

(t/deftest diagnostic-and-malformed-features-fail-at-the-parser-boundary
  (doseq [feature ["stats" "profile" "profile-macros" "profile-raf"]]
    (t/is (thrown-with-msg? js/Error #"diagnostic Cargo features"
                            (runner/parse-args ["build" "--features" (str "functional," feature)]))))
  (doseq [feature ["" "functional,,other" "functional/other"]]
    (t/is (thrown-with-msg? js/Error #"invalid Cargo feature selection"
                            (runner/parse-args ["build" "--features" feature]))))
  (t/is (thrown? js/Error (runner/parse-args ["build" "--features"]))))

(t/deftest preparation-budget-requires-a-complete-positive-safe-integer-number
  ;; Number accepts whitespace and exponent notation when the result is an
  ;; integer; these are valid CLI budgets, not partial numeric parses.
  (doseq [[text expected] [["1" 1] [" 1000 " 1000] ["1e3" 1000]]]
    (t/is (= expected (get-in (runner/parse-args ["build" "--build-timeout-ms" text])
                              [:options :build-timeout-ms]))))
  (doseq [text ["" "0" "-1" "1.5" "1000ms" "NaN" "Infinity" "9007199254740992"]]
    (t/is (thrown-with-msg? js/Error #"Must be a positive integer"
                            (runner/parse-args ["build" "--build-timeout-ms" text]))))
  (t/is (thrown? js/Error (runner/parse-args ["build" "--build-timeout-ms"]))))

(t/deftest purge-categories-and-build-flags-cannot-be-mixed
  (doseq [[args selected] [[["purge"] {}]
                           [["purge" "--only-builds"] {:only-builds true}]
                           [["purge" "--only-runs"] {:only-runs true}]]]
    (let [{:keys [command options]} (runner/parse-args args)]
      (t/is (= "purge" command))
      (t/is (= selected (select-keys options [:only-builds :only-runs])))))
  (t/is (thrown-with-msg? js/Error #"mutually exclusive"
                          (runner/parse-args ["purge" "--only-builds" "--only-runs"])))
  (doseq [flag ["--only-builds" "--only-runs"]]
    (t/is (thrown-with-msg? js/Error #"purge category flags require purge"
                            (runner/parse-args ["build" flag]))))
  (doseq [args [["purge" "--default-features"]
                ["purge" "--no-default-features"]
                ["purge" "--features" "functional"]
                ["purge" "--build-timeout-ms" "1000"]
                ["purge" "--features=functional"]
                ["purge" "--build-timeout-ms=1000"]]]
    (t/is (thrown-with-msg? js/Error #"build options require build"
                            (runner/parse-args args)))))

(t/deftest help-needs-no-command-and-other-command-lines-must-be-valid
  (doseq [args [["--help"] ["-h"] ["build" "--help"] ["run" "--help"]
                ["compare" "--help"] ["merge" "--help"]]]
    (let [{:keys [options summary]} (runner/parse-args args)
          help                    (runner/usage summary)]
      (t/is (true? (:help options)))
      (t/is (str/includes? help "runner build"))
      (t/is (str/includes? help "runner purge"))
      (t/is (str/includes? help "runner run"))
      (t/is (str/includes? help "runner compare"))
      (t/is (str/includes? help "runner merge"))
      (t/is (str/includes? help "--build-timeout-ms"))
      (t/is (str/includes? help "--cases"))
      (t/is (str/includes? help "--allow-build-differences"))
      (t/is (str/includes? help "--allow-scene-differences"))
      (t/is (str/includes? help "--allow-all-differences"))
      (t/is (not (str/includes? help "--force")))))
  (t/is (thrown-with-msg? js/Error #"expected build, purge, run, ab, compare or merge"
                          (runner/parse-args [])))
  (doseq [args [["unknown"] ["build" "purge"] ["build" "--unknown"]
                ["run" "extra"] ["compare" "only-one"]]]
    (t/is (thrown? js/Error (runner/parse-args args)))))

(t/deftest run-resolves-default-and-explicit-options-before-resources
  (let [{:keys [command options arguments]} (runner/parse-args ["run"])]
    (t/is (= "run" command))
    (t/is (= ["run"] arguments))
    (t/is (= {:seed 42 :warmups 3 :repetitions 10 :timeout-ms 30000
              :build-timeout-ms 300000 :startup-timeout-ms 30000}
             (select-keys options [:seed :warmups :repetitions :timeout-ms
                                   :build-timeout-ms :startup-timeout-ms])))
    (t/is (nil? (:build-id options)))
    (t/is (nil? (:filter options)))
    (t/is (not (:remove options)))
    (t/is (not (:screenshots options))))
  (let [{:keys [options]} (runner/parse-args
                           ["run" "--seed" "7" "--filter" "rects"
                            "--warmups" "0" "--repetitions" "2"
                            "--timeout-ms" "100" "--build-timeout-ms" "200"
                            "--startup-timeout-ms" "300" "--output-dir" "out"
                            "--remove" "--screenshots"])]
    (t/is (= {:seed 7 :filter "rects" :warmups 0 :repetitions 2
              :timeout-ms 100 :build-timeout-ms 200 :startup-timeout-ms 300
              :output-dir "out" :remove true :screenshots true}
             (select-keys options [:seed :filter :warmups :repetitions
                                   :timeout-ms :build-timeout-ms
                                   :startup-timeout-ms :output-dir
                                   :remove :screenshots])))))

(t/deftest retained-runs-reject-feature-selection-and-build-runs-accept-it
  (t/is (= "retained-id" (get-in (runner/parse-args ["run" "--build" "retained-id"])
                                 [:options :build-id])))
  (doseq [args [["run" "--build" "retained-id" "--no-default-features"]
                ["run" "--no-default-features" "--build" "retained-id"]
                ["run" "--build" "retained-id" "--features" "functional"]
                ["run" "--build" "retained-id" "--default-features"]]]
    (t/is (thrown-with-msg? js/Error #"cannot be combined"
                            (runner/parse-args args))))
  (let [{:keys [options]} (runner/parse-args
                           ["run" "--no-default-features" "--features" "functional"])]
    (t/is (false? (:default-features options)))
    (t/is (= ["functional"] (:features options)))))

(t/deftest compare-takes-two-file-paths-and-its-own-options
  (let [{:keys [command arguments options]}
        (runner/parse-args ["compare" "baseline.transit" "candidate.transit"])]
    (t/is (= "compare" command))
    (t/is (= ["compare" "baseline.transit" "candidate.transit"] arguments))
    (t/is (nil? (:cases options)))
    (t/is (not-any? options [:allow-build-differences :allow-scene-differences
                             :allow-all-differences])))
  (doseq [[flag option] [["--allow-build-differences" :allow-build-differences]
                         ["--allow-scene-differences" :allow-scene-differences]
                         ["--allow-all-differences" :allow-all-differences]]
          selectors [[] ["--cases" "rects/pan,paths/pan"]]]
    (let [{:keys [options]}
          (runner/parse-args (into ["compare" "a.transit" "b.transit" flag] selectors))]
      (t/is (= (when (seq selectors) "rects/pan,paths/pan") (:cases options)))
      (t/is (true? (get options option)))
      (t/is (not-any? options (remove #{option} [:allow-build-differences
                                                 :allow-scene-differences
                                                 :allow-all-differences])))))
  (let [{:keys [options]}
        (runner/parse-args ["compare" "a.transit" "b.transit"
                            "--allow-build-differences" "--allow-scene-differences"])]
    (t/is (true? (:allow-build-differences options)))
    (t/is (true? (:allow-scene-differences options)))
    (t/is (not (:allow-all-differences options)))))

(t/deftest merge-takes-two-paths-and-only-output-dir
  (let [{:keys [command arguments options]}
        (runner/parse-args ["merge" "base.transit" "additional.transit"
                            "--output-dir" "merged-home"])]
    (t/is (= "merge" command))
    (t/is (= ["merge" "base.transit" "additional.transit"] arguments))
    (t/is (= "merged-home" (:output-dir options))))
  (doseq [args [["merge"] ["merge" "only-base"]
                ["merge" "base" "additional" "extra"]]]
    (t/is (thrown? js/Error (runner/parse-args args))))
  (doseq [flags [["--build" "id"]
                 ["--filter" "rects"]
                 ["--seed" "1"]
                 ["--require-gpu"]
                 ["--features" "functional"]
                 ["--cases" "a,b"]
                 ["--allow-build-differences"]
                 ["--only-builds"]
                 ["--dry-run"]]]
    (t/is (thrown? js/Error (runner/parse-args (into ["merge" "base" "additional"] flags))))))

(t/deftest force-is-rejected-without-an-alias
  (doseq [args [["compare" "a.transit" "b.transit" "--force"]
                ["compare" "a.transit" "b.transit" "--force" "--allow-all-differences"]
                ["compare" "a.transit" "b.transit" "--cases" "a,b" "--force"]]]
    (t/is (thrown? js/Error (runner/parse-args args)))))

(t/deftest commands-reject-flags-owned-by-other-commands
  (doseq [[args message]
          [[["run" "--only-builds"] #"purge category flags require purge"]
           [["run" "--cases" "a,b"] #"compare options require compare"]
           [["compare" "a" "b" "--seed" "1"] #"run options require run"]
           [["compare" "a" "b" "--features" "functional"] #"build options require build"]
           [["build" "--build" "retained"] #"run options require run"]
           [["build" "--only-runs"] #"purge category flags require purge"]
           [["purge" "--seed" "1"] #"run options require run"]
           [["purge" "--cases" "a,b"] #"compare options require compare"]
           [["merge" "a" "b" "--seed" "1"] #"run options require run"]
           [["merge" "a" "b" "--features" "functional"] #"build options require build"]
           [["merge" "a" "b" "--cases" "a,b"] #"compare options require compare"]
           [["merge" "a" "b" "--only-builds"] #"purge category flags require purge"]
           [["merge" "a" "b" "--dry-run"] #"ab planning options require ab"]]]
    (t/is (thrown-with-msg? js/Error message (runner/parse-args args))))
  (doseq [command ["build" "purge" "run"]
          flag ["--allow-build-differences" "--allow-scene-differences"
                "--allow-all-differences"]]
    (t/is (thrown-with-msg? js/Error #"compare options require compare"
                            (runner/parse-args [command flag])))))

(t/deftest numeric-and-path-options-reject-invalid-bounds
  (doseq [args [["run" "--seed" "4294967296"]
                ["run" "--seed" "-1"]
                ["run" "--warmups" "-1"]
                ["run" "--warmups" "1.5"]
                ["run" "--repetitions" "0"]
                ["run" "--timeout-ms" "0"]
                ["run" "--timeout-ms" "2147483648"]
                ["run" "--startup-timeout-ms" "0"]
                ["run" "--build" ""]
                ["run" "--output-dir" ""]]]
    (t/is (thrown? js/Error (runner/parse-args args))))
  (t/is (= 4294967295 (:seed (:options (runner/parse-args ["run" "--seed" "4294967295"])))))
  (t/is (= 0 (:warmups (:options (runner/parse-args ["run" "--warmups" "0"]))))))

(t/deftest option-looking-values-and-filenames-are-not-command-flags
  (let [parsed (runner/parse-args ["compare" "--" "--seed" "candidate.transit"])]
    (t/is (= ["compare" "--seed" "candidate.transit"] (:arguments parsed))))
  (let [parsed (runner/parse-args ["run" "--filter" "--allow-all-differences"])]
    (t/is (= "--allow-all-differences" (:filter (:options parsed))))))

(t/deftest numeric-options-reject-blank-values
  (doseq [args [["run" "--seed" ""]
                ["run" "--warmups" " "]
                ["run" "--repetitions" ""]
                ["run" "--timeout-ms" " "]]]
    (t/is (thrown? js/Error (runner/parse-args args)))))

(t/deftest gpu-policy-is-available-for-runs-and-feature-comparisons
  (doseq [args [["run" "--require-gpu"]
                ["ab" "--features" "a" "--filter" "rects/load" "--require-gpu"]
                ["ab" "--features" "a" "--require-gpu" "--dry-run"]]]
    (t/is (true? (get-in (runner/parse-args args) [:options :require-gpu]))))
  (doseq [args [["build" "--require-gpu"]
                ["purge" "--require-gpu"]
                ["compare" "a" "b" "--require-gpu"]]]
    (t/is (thrown-with-msg? js/Error #"run options require run" (runner/parse-args args))))
  (t/is (re-find #"--require-gpu" (runner/usage ""))))

(t/deftest ab-accepts-run-budgets-and-comparison-permissions
  (let [{:keys [command options]}
        (runner/parse-args ["ab" "--features" " b,a,b " "--filter" "rects/load"
                            "--seed" "7" "--warmups" "0" "--repetitions" "2"
                            "--timeout-ms" "1000" "--build-timeout-ms" "2000"
                            "--startup-timeout-ms" "3000" "--output-dir" "new-home"
                            "--screenshots" "--allow-build-differences"
                            "--allow-scene-differences" "--allow-all-differences"])]
    (t/is (= "ab" command))
    (t/is (= {:features ["a" "b"] :filter "rects/load" :seed 7
              :warmups 0 :repetitions 2 :timeout-ms 1000 :build-timeout-ms 2000
              :startup-timeout-ms 3000 :output-dir "new-home" :screenshots true
              :allow-build-differences true :allow-scene-differences true :allow-all-differences true
              :max-features 3 :confirm false :dry-run false :default-features false :color nil}
             options))))

(t/deftest ab-rejects-missing-features-and-incompatible-flags-before-execution
  (t/is (thrown-with-msg? js/Error #"at least one" (runner/parse-args ["ab" "--dry-run"])))
  (doseq [args [["--build" "id"] ["--remove"] ["--cases" "a,b"]
                ["--default-features"] ["--no-default-features"]]]
    (t/is (thrown-with-msg? js/Error #"ab rejects"
                            (runner/parse-args (into ["ab" "--features" "a" "--dry-run"] args)))))
  (doseq [features ["" "a," "profile" "a/b"]]
    (t/is (thrown? js/Error (runner/parse-args ["ab" "--features" features "--dry-run"]))))
  (t/is (thrown-with-msg? js/Error #"purge category flags require purge"
                          (runner/parse-args ["ab" "--features" "a" "--only-comparisons"])))
  (t/is (thrown? js/Error (runner/parse-args ["ab" "extra" "--features" "a" "--dry-run"]))))

(t/deftest ab-planning-errors-keep-their-cause
  (let [error (try
                (runner/parse-args ["ab" "--dry-run"])
                nil
                (catch :default cause cause))]
    (t/is (= :benches.render-wasm.node.runner/invalid-args
             (:type (ex-data error))))
    (t/is (some? (ex-cause error)))
    (t/is (str/includes? (ex-message (ex-cause error)) "at least one"))))

(t/deftest ab-limits-and-confirmation-apply-at-the-parser-boundary
  (t/is (thrown-with-msg? js/Error #"requires --confirm" (runner/parse-args ["ab" "--features" "a"])))
  (t/is (:confirm (:options (runner/parse-args ["ab" "--features" "a" "--confirm"]))))
  (t/is (:dry-run (:options (runner/parse-args ["ab" "--features" "a" "--dry-run"]))))
  (t/is (thrown-with-msg? js/Error #"exceed max-features"
                          (runner/parse-args ["ab" "--features" "a,b,c,d" "--dry-run"])))
  (t/is (thrown-with-msg? js/Error #"requires --confirm"
                          (runner/parse-args ["ab" "--features" "a,b,c,d" "--max-features" "4" "--filter" "rects"])))
  (doseq [flag ["--confirm" "--dry-run"]]
    (t/is (= 4 (:max-features (:options (runner/parse-args ["ab" "--features" "a,b,c,d" "--max-features" "4" flag]))))))
  (doseq [limit ["" " " "0" "7" "-1" "1.5" "NaN" "Infinity" "3x"]]
    (t/is (thrown? js/Error (runner/parse-args ["ab" "--features" "a" "--max-features" limit "--dry-run"])))))

(t/deftest planning-flags-are-only-for-ab-and-help-explains-the-contract
  (doseq [command [["build"] ["run"] ["purge"] ["compare" "before" "after"]]
          flags [["--dry-run"] ["--confirm"] ["--max-features" "2"]]]
    (t/is (thrown-with-msg? js/Error #"ab planning options require ab"
                            (runner/parse-args (into command flags)))))
  (let [{:keys [summary]} (runner/parse-args ["ab" "--help"])
        help (runner/usage summary)]
    (doseq [part ["runner ab --features" "--only-comparisons" "--max-features"
                  "--confirm" "--dry-run" "Cargo defaults disabled"
                  "feature-comparisons/<ab-id>/feature-comparison.transit"]]
      (t/is (str/includes? help part)))))

(t/deftest comparison-purge-is-exclusive-and-only-for-purge
  (t/is (true? (get-in (runner/parse-args ["purge" "--only-comparisons"]) [:options :only-comparisons])))
  (doseq [other ["--only-builds" "--only-runs"]]
    (t/is (thrown-with-msg? js/Error #"mutually exclusive"
                            (runner/parse-args ["purge" "--only-comparisons" other]))))
  (doseq [command [["build"] ["run"] ["compare" "before" "after"]]]
    (t/is (thrown-with-msg? js/Error #"purge category flags require purge"
                            (runner/parse-args (conj command "--only-comparisons"))))))