Commit 4aacffc9 authored by Dom Sekotill's avatar Dom Sekotill
Browse files

Ensure test container fixtures clean up in the event of errors

parent e5bb6936
Loading
Loading
Loading
Loading
+5 −5
Original line number Diff line number Diff line
@@ -129,11 +129,11 @@ class Site:
		test_dir = Path(__file__).parent
		db_init = test_dir / "mysql-init.sql"

		with Network() as network:
			database = Mysql(network=network, init_files=[db_init])
		with Network() as network, Mysql(network=network, init_files=[db_init]) as database:
			database.start()  # Get a head start on initialising the database
			backend = Wordpress(site_url, database, network=network)
			frontend = Nginx(backend, network=network)
			with \
				Wordpress(site_url, database, network=network) as backend, \
				Nginx(backend, network=network) as frontend:
					yield cls(site_url, network, frontend, backend, database)

	@contextmanager