[Fix #50] Add type hints for ResourceReaper (#51)

This commit is contained in:
Andrea Richiardi 2021-03-09 00:26:47 -08:00 committed by GitHub
parent ff1215e710
commit 12c4b6c1e9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

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