7 files changed, 260 insertions, 0 deletions
diff --git a/test/integration/targets/wait_for/aliases b/test/integration/targets/wait_for/aliases
new file mode 100644
index 0000000..a4c92ef
--- /dev/null
+++ b/test/integration/targets/wait_for/aliases
@@ -0,0 +1,2 @@
+destructive
+shippable/posix/group1
diff --git a/test/integration/targets/wait_for/files/testserver.py b/test/integration/targets/wait_for/files/testserver.py
new file mode 100644
index 0000000..2b728b6
--- /dev/null
+++ b/test/integration/targets/wait_for/files/testserver.py
@@ -0,0 +1,19 @@
+from __future__ import (absolute_import, division, print_function)
+__metaclass__ = type
+
+import sys
+
+if __name__ == '__main__':
+    if sys.version_info[0] >= 3:
+        import http.server
+        import socketserver
+        PORT = int(sys.argv[1])
+        Handler = http.server.SimpleHTTPRequestHandler
+        httpd = socketserver.TCPServer(("", PORT), Handler)
+        httpd.serve_forever()
+    else:
+        import mimetypes
+        mimetypes.init()
+        mimetypes.add_type('application/json', '.json')
+        import SimpleHTTPServer
+        SimpleHTTPServer.test()
diff --git a/test/integration/targets/wait_for/files/write_utf16.py b/test/integration/targets/wait_for/files/write_utf16.py
new file mode 100644
index 0000000..6079ed3
--- /dev/null
+++ b/test/integration/targets/wait_for/files/write_utf16.py
@@ -0,0 +1,20 @@
+from __future__ import (absolute_import, division, print_function)
+__metaclass__ = type
+
+import sys
+
+# utf16 encoded bytes
+# to ensure wait_for doesn't have any encoding errors
+data = (
+    b'\xff\xfep\x00r\x00e\x00m\x00i\x00\xe8\x00r\x00e\x00 \x00i\x00s\x00 '
+    b'\x00f\x00i\x00r\x00s\x00t\x00\n\x00p\x00r\x00e\x00m\x00i\x00e\x00'
+    b'\x00\x03r\x00e\x00 \x00i\x00s\x00 \x00s\x00l\x00i\x00g\x00h\x00t\x00'
+    b'l\x00y\x00 \x00d\x00i\x00f\x00f\x00e\x00r\x00e\x00n\x00t\x00\n\x00\x1a'
+    b'\x048\x04@\x048\x04;\x04;\x048\x04F\x040\x04 \x00i\x00s\x00 \x00C\x00y'
+    b'\x00r\x00i\x00l\x00l\x00i\x00c\x00\n\x00\x01\xd8\x00\xdc \x00a\x00m'
+    b'\x00 \x00D\x00e\x00s\x00e\x00r\x00e\x00t\x00\n\x00\n'
+    b'completed\n'
+)
+
+with open(sys.argv[1], 'wb') as f:
+    f.write(data)
diff --git a/test/integration/targets/wait_for/files/zombie.py b/test/integration/targets/wait_for/files/zombie.py
new file mode 100644
index 0000000..913074e
--- /dev/null
+++ b/test/integration/targets/wait_for/files/zombie.py
@@ -0,0 +1,13 @@
+from __future__ import absolute_import, division, print_function
+__metaclass__ = type
+
+import os
+import sys
+import time
+
+child_pid = os.fork()
+
+if child_pid > 0:
+    time.sleep(60)
+else:
+    sys.exit()
diff --git a/test/integration/targets/wait_for/meta/main.yml b/test/integration/targets/wait_for/meta/main.yml
new file mode 100644
index 0000000..cb6005d
--- /dev/null
+++ b/test/integration/targets/wait_for/meta/main.yml
@@ -0,0 +1,3 @@
+dependencies:
+  - prepare_tests
+  - setup_remote_tmp_dir
diff --git a/test/integration/targets/wait_for/tasks/main.yml b/test/integration/targets/wait_for/tasks/main.yml
new file mode 100644
index 0000000..f71ddbd
--- /dev/null
+++ b/test/integration/targets/wait_for/tasks/main.yml
@@ -0,0 +1,199 @@
+---
+- name: test wait_for with delegate_to
+  wait_for:
+    timeout: 2
+  delegate_to: localhost
+  register: waitfor
+
+- assert:
+    that:
+      - waitfor is successful
+      - waitfor.elapsed >= 2
+
+- name: setup create a directory to serve files from
+  file:
+    dest: "{{ files_dir }}"
+    state: directory
+
+- name: setup webserver
+  copy:
+    src: "testserver.py"
+    dest: "{{ remote_tmp_dir }}/testserver.py"
+
+- name: setup a path
+  file:
+    path: "{{ remote_tmp_dir }}/wait_for_file"
+    state: touch
+
+- name: setup remove a file after 3s
+  shell: sleep 3 && rm {{ remote_tmp_dir }}/wait_for_file
+  async: 20
+  poll: 0
+
+- name: test for absent path
+  wait_for:
+    path: "{{ remote_tmp_dir }}/wait_for_file"
+    state: absent
+    timeout: 20
+  register: waitfor
+- name: verify test for absent path
+  assert:
+    that:
+      - waitfor is successful
+      - waitfor.path == "{{ remote_tmp_dir | expanduser }}/wait_for_file"
+      - waitfor.elapsed >= 2
+      - waitfor.elapsed <= 15
+
+- name: setup create a file after 3s
+  shell: sleep 3 && touch {{ remote_tmp_dir }}/wait_for_file
+  async: 20
+  poll: 0
+
+- name: test for present path
+  wait_for:
+    path:  "{{ remote_tmp_dir }}/wait_for_file"
+    timeout: 5
+  register: waitfor
+- name: verify test for absent path
+  assert:
+    that:
+      - waitfor is successful
+      - waitfor.path == "{{ remote_tmp_dir | expanduser }}/wait_for_file"
+      - waitfor.elapsed >= 2
+      - waitfor.elapsed <= 15
+
+- name: setup write keyword to file after 3s
+  shell: sleep 3 && echo completed > {{remote_tmp_dir}}/wait_for_keyword
+  async: 20
+  poll: 0
+
+- name: test wait for keyword in file
+  wait_for:
+    path: "{{remote_tmp_dir}}/wait_for_keyword"
+    search_regex: completed
+    timeout: 5
+  register: waitfor
+
+- name: verify test wait for keyword in file
+  assert:
+    that:
+      - waitfor is successful
+      - "waitfor.search_regex == 'completed'"
+      - waitfor.elapsed >= 2
+      - waitfor.elapsed <= 15
+
+- name: setup write keyword to file after 3s
+  shell: sleep 3 && echo "completed data 123" > {{remote_tmp_dir}}/wait_for_keyword
+  async: 20
+  poll: 0
+
+- name: test wait for keyword in file with match groups
+  wait_for:
+    path: "{{remote_tmp_dir}}/wait_for_keyword"
+    search_regex: completed (?P<foo>\w+) ([0-9]+)
+    timeout: 5
+  register: waitfor
+
+- name: verify test wait for keyword in file with match groups
+  assert:
+    that:
+    - waitfor is successful
+    - waitfor.elapsed >= 2
+    - waitfor.elapsed <= 15
+    - waitfor['match_groupdict'] | length == 1
+    - waitfor['match_groupdict']['foo'] == 'data'
+    - waitfor['match_groups'] == ['data', '123']
+
+- name: write non-ascii file
+  script: write_utf16.py "{{remote_tmp_dir}}/utf16.txt"
+  args:
+    executable: '{{ ansible_facts.python.executable }}'
+
+- name: test non-ascii file
+  wait_for:
+    path: "{{remote_tmp_dir}}/utf16.txt"
+    search_regex: completed
+
+- name: test wait for port timeout
+  wait_for:
+    port: 12121
+    timeout: 3
+  register: waitfor
+  ignore_errors: true
+- name: verify test wait for port timeout
+  assert:
+    that:
+      - waitfor is failed
+      - waitfor.elapsed == 3
+      - "waitfor.msg == 'Timeout when waiting for 127.0.0.1:12121'"
+
+- name: test fail with custom msg
+  wait_for:
+    port: 12121
+    msg: fail with custom message
+    timeout: 3
+  register: waitfor
+  ignore_errors: true
+- name: verify test fail with custom msg
+  assert:
+    that:
+      - waitfor is failed
+      - waitfor.elapsed == 3
+      - "waitfor.msg == 'fail with custom message'"
+
+- name: setup start SimpleHTTPServer
+  shell: sleep 3 && cd {{ files_dir }} && {{ ansible_python.executable }} {{ remote_tmp_dir}}/testserver.py {{ http_port }}
+  async: 120 # this test set can take ~1m to run on FreeBSD (via Shippable)
+  poll: 0
+
+- name: test wait for port with sleep
+  wait_for:
+    port: "{{ http_port }}"
+    sleep: 3
+  register: waitfor
+- name: verify test wait for port sleep
+  assert:
+    that:
+      - waitfor is successful
+      - waitfor is not changed
+      - "waitfor.port == {{ http_port }}"
+
+- name: install psutil using pip (non-Linux only)
+  pip:
+    name: psutil==5.8.0
+  when: ansible_system != 'Linux'
+
+- name: Copy zombie.py
+  copy:
+    src: zombie.py
+    dest: "{{ remote_tmp_dir }}"
+
+- name: Create zombie process
+  shell: "{{ ansible_python.executable }} {{ remote_tmp_dir }}/zombie"
+  async: 90
+  poll: 0
+
+- name: test wait for port drained
+  wait_for:
+    port: "{{ http_port }}"
+    state: drained
+  register: waitfor
+
+- name: verify test wait for port
+  assert:
+    that:
+      - waitfor is successful
+      - waitfor is not changed
+      - "waitfor.port == {{ http_port }}"
+
+- name: test wait_for with delay
+  wait_for:
+    timeout: 2
+    delay: 2
+  register: waitfor
+
+- name: verify test wait_for with delay
+  assert:
+    that:
+      - waitfor is successful
+      - waitfor.elapsed >= 4
diff --git a/test/integration/targets/wait_for/vars/main.yml b/test/integration/targets/wait_for/vars/main.yml
new file mode 100644
index 0000000..d15b6d7
--- /dev/null
+++ b/test/integration/targets/wait_for/vars/main.yml
@@ -0,0 +1,4 @@
+---
+http_port: 15261
+files_dir: '{{ remote_tmp_dir|expanduser }}/files'
+checkout_dir: '{{ remote_tmp_dir }}/git'