Mar 3, 2018 · Feb 6, 2018 · Feb 14, 2018 · Mar 1, 2018 · Mar 1, 2018 · Mar 1, 2018
diff --git a/testgres/connection.py b/testgres/connection.py
    Transaction isolation level for NodeConnection
    """

    ReadUncommitted, ReadCommitted, RepeatableRead, Serializable = range(4)
    ReadUncommitted = 'read uncommitted'
    ReadCommitted = 'read committed'
    RepeatableRead = 'repeatable read'
    Serializable = 'serializable'


 class NodeConnection(object):
        self.close()

    def begin(self, isolation_level=IsolationLevel.ReadCommitted):
        # yapf: disable
        levels = [
            'read uncommitted',
            'read committed',
            'repeatable read',
            'serializable'
        ]

        # Check if level is an IsolationLevel
        if (isinstance(isolation_level, IsolationLevel)):

            # Get index of isolation level
            level_idx = isolation_level.value
            assert level_idx in range(4)

            # Replace isolation level with its name
            isolation_level = levels[level_idx]

        else:
        # Check if level isn't an IsolationLevel
        if not isinstance(isolation_level, IsolationLevel):
            # Get name of isolation level
            level_str = str(isolation_level).lower()

            # Validate level string
            if level_str not in levels:
            try:
                isolation_level = IsolationLevel(level_str)
            except ValueError:
                error = 'Invalid isolation level "{}"'
                raise QueryException(error.format(level_str))

            # Replace isolation level with its name
            isolation_level = level_str

        # Set isolation level
        cmd = 'SET TRANSACTION ISOLATION LEVEL {}'
        self.cursor.execute(cmd.format(isolation_level))
        self.cursor.execute(cmd.format(isolation_level.value))

        return self

diff --git a/testgres/node.py b/testgres/node.py
 import tempfile
 import time

 from enum importEnum
 from enum importIntEnum
 from six import raise_from

 from .cache import cached_initdb
    positional_args_hack


 class NodeStatus(Enum):
 class NodeStatus(IntEnum):
    """
    Status of a PostgresNode
    """

    # for Python 3.x
    def __bool__(self):
        return self.value == NodeStatus.Running.value
        return self == NodeStatus.Running

    # for Python 2.x
    __nonzero__ = __bool__
Original file line number	Diff line number	Diff line change
Expand Up		@@ -27,7 +27,10 @@ class IsolationLevel(Enum):
		Transaction isolation level for NodeConnection
		"""

		ReadUncommitted, ReadCommitted, RepeatableRead, Serializable = range(4)
		ReadUncommitted = 'read uncommitted'
		ReadCommitted = 'read committed'
		RepeatableRead = 'repeatable read'
		Serializable = 'serializable'


		class NodeConnection(object):
Expand DownExpand Up		@@ -71,39 +74,21 @@ def __exit__(self, type, value, traceback):
		self.close()

		def begin(self, isolation_level=IsolationLevel.ReadCommitted):
		# yapf: disable
		levels = [
		'read uncommitted',
		'read committed',
		'repeatable read',
		'serializable'
		]

		# Check if level is an IsolationLevel
		if (isinstance(isolation_level, IsolationLevel)):

		# Get index of isolation level
		level_idx = isolation_level.value
		assert level_idx in range(4)

		# Replace isolation level with its name
		isolation_level = levels[level_idx]

		else:
		# Check if level isn't an IsolationLevel
		if not isinstance(isolation_level, IsolationLevel):
		# Get name of isolation level
		level_str = str(isolation_level).lower()

		# Validate level string
		if level_str not in levels:
		try:
		isolation_level = IsolationLevel(level_str)
		except ValueError:
		error = 'Invalid isolation level "{}"'
		raise QueryException(error.format(level_str))

		# Replace isolation level with its name
		isolation_level = level_str

		# Set isolation level
		cmd = 'SET TRANSACTION ISOLATION LEVEL {}'
		self.cursor.execute(cmd.format(isolation_level))
		self.cursor.execute(cmd.format(isolation_level.value))

		return self

Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -8,7 +8,7 @@
		import tempfile
		import time

		from enum importEnum
		from enum importIntEnum
		from six import raise_from

		from .cache import cached_initdb
Expand DownExpand Up		@@ -53,7 +53,7 @@
		positional_args_hack


		class NodeStatus(Enum):
		class NodeStatus(IntEnum):
		"""
		Status of a PostgresNode
		"""
Expand All		@@ -62,7 +62,7 @@ class NodeStatus(Enum):

		# for Python 3.x
		def __bool__(self):
		return self.value == NodeStatus.Running.value
		return self == NodeStatus.Running

		# for Python 2.x
		__nonzero__ = __bool__
Expand Down