stable-diffusion-webui/test/server_poll.py

import unittest
import requests
import time


def run_tests():
    timeout_threshold = 240
    start_time = time.time()
    while time.time()-start_time < timeout_threshold:
        try:
            requests.head("http://localhost:7860/")
            break
        except requests.exceptions.ConnectionError:
            pass
    if time.time()-start_time < timeout_threshold:
        suite = unittest.TestLoader().discover('', pattern='*_test.py')
        result = unittest.TextTestRunner(verbosity=2).run(suite)
    else:
        print("Launch unsuccessful")