From 072e19f192973ca2f5a82b1fcef134888b0dcceb Mon Sep 17 00:00:00 2001
From: Matus Valo <matusvalo@gmail.com>
Date: Fri, 15 Feb 2019 06:09:40 -0800
Subject: Added RabbitMQ windows tests

Conflicts:
	appveyor.yml
---
 setup.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

(limited to 'setup.py')

diff --git a/setup.py b/setup.py
index 12d6e62..3458a07 100644
--- a/setup.py
+++ b/setup.py
@@ -91,17 +91,24 @@ else:
 
 
 class pytest(setuptools.command.test.test):
-    user_options = [('pytest-args=', 'a', 'Arguments to pass to py.test')]
+    user_options = [
+        ('pytest-args=', 'a', 'Arguments to pass to py.test'),
+        ('pytest-env=', 'E', 'Only run tests matching the environment'),
+    ]
 
     def initialize_options(self):
         setuptools.command.test.test.initialize_options(self)
         self.pytest_args = ''
+        self.pytest_env = None
 
     def run_tests(self):
         import pytest
         pytest_args = self.pytest_args.split(' ')
+        if self.pytest_env:
+            pytest_args = pytest_args + ['-E', self.pytest_env]
         sys.exit(pytest.main(pytest_args))
 
+
 setuptools.setup(
     name=NAME,
     packages=setuptools.find_packages(exclude=['ez_setup', 't', 't.*']),
-- 
cgit v1.2.1