[Fix #50] Add type hints for ResourceReaper

This commit is contained in:
Andrea Richiardi 2021-03-08 17:14:46 -08:00
parent ff1215e710
commit 062052075d
No known key found for this signature in database
GPG key ID: 3D0E9F0881052A48

View file

@ -17,7 +17,8 @@
(org.testcontainers.images.builder (org.testcontainers.images.builder
ImageFromDockerfile) ImageFromDockerfile)
(org.testcontainers.utility (org.testcontainers.utility
MountableFile))) MountableFile
ResourceReaper)))
(defn- resolve-bind-mode (defn- resolve-bind-mode
(^BindMode [bind-mode] (^BindMode [bind-mode]
@ -27,7 +28,7 @@
(defn- reaper-instance (defn- reaper-instance
[] []
(org.testcontainers.utility.ResourceReaper/instance)) (ResourceReaper/instance))
(defmulti wait (defmulti wait
"Sets a wait strategy to the container. Supports :http, :health and :log as "Sets a wait strategy to the container. Supports :http, :health and :log as
@ -278,10 +279,10 @@
[port] [port]
[port (.getMappedPort container port)]) [port (.getMappedPort container port)])
mapped-ports (into {} (map map-port) exposed-ports) mapped-ports (into {} (map map-port) exposed-ports)
container-id (.getContainerId container) container-id ^String (.getContainerId container)
image-name (.getDockerImageName container) image-name ^String (.getDockerImageName container)
logger (log log-to container)] logger (log log-to container)]
(.registerContainerForCleanup (reaper-instance) (.registerContainerForCleanup ^ResourceReaper (reaper-instance)
container-id container-id
image-name) image-name)
(-> container-config (-> container-config
@ -317,7 +318,7 @@
(let [network (.build builder) (let [network (.build builder)
network-name (.getName network)] network-name (.getName network)]
(.registerNetworkIdForCleanup (reaper-instance) network-name) (.registerNetworkIdForCleanup ^ResourceReaper (reaper-instance) network-name)
{:network network {:network network
:name network-name :name network-name
:ipv6 (.getEnableIpv6 network) :ipv6 (.getEnableIpv6 network)
@ -328,7 +329,7 @@
(defn perform-cleanup! (defn perform-cleanup!
"Stops and removes all container instances which were created in the active JVM or REPL session" "Stops and removes all container instances which were created in the active JVM or REPL session"
[] []
(.performCleanup (reaper-instance))) (.performCleanup ^ResourceReaper (reaper-instance)))
;;; REPL Helpers ;;; REPL Helpers