Make Container.started return the same type as self (eb99b6fd) · Commits · Dom Sekotill / Behave Utils

behave_utils/docker.py

+1 −1

Original line number	Diff line number	Diff line
		@@ -222,7 +222,7 @@ class Container(Item):
		self.stop(rm=True)

		@contextmanager
		def started(self) -> Iterator[Container]:
		def started(self: T) -> Iterator[T]:
		"""
		Return a context manager that ensures the container is started when the context is entered
		"""

+4 −1

Original line number	Diff line number	Diff line
		@@ -16,6 +16,7 @@ from time import sleep
		from typing import TYPE_CHECKING
		from typing import Iterator
		from typing import Sequence
		from typing import TypeVar

		from behave import fixture

		@@ -39,6 +40,8 @@ class Mysql(Container):
		Container subclass for a database container
		"""

		T = TypeVar('T', bound='Mysql')

		def __init__(
		self,
		version: str = "latest",
		@@ -95,7 +98,7 @@ class Mysql(Container):
		return Cli(self, "mysqldump")

		@contextmanager
		def started(self) -> Iterator[Container]:
		def started(self: T) -> Iterator[T]:
		"""
		Return a context manager that only enters once the database is initialised
		"""