summaryrefslogtreecommitdiff
path: root/test/testutil.py
diff options
context:
space:
mode:
authorTaras <voyn1991@gmail.com>2018-03-18 15:56:47 +0200
committerTaras <voyn1991@gmail.com>2018-03-18 15:56:47 +0200
commitef955155f89e9a665a4b636e4535d7b50fce869d (patch)
treecfe10509abaa41d0efa9f841d3a4aa4e7558ec81 /test/testutil.py
parent18e48dce240eaa7cf714c780c02d1d5cf0b8fca2 (diff)
downloadkafka-python-fix_records.tar.gz
Add codec validators to record parser and builder for all formats.fix_records
Also removed `decorator` dependency on tests Fixes #1443
Diffstat (limited to 'test/testutil.py')
-rw-r--r--test/testutil.py7
1 files changed, 5 insertions, 2 deletions
diff --git a/test/testutil.py b/test/testutil.py
index 4e5db47..365e47f 100644
--- a/test/testutil.py
+++ b/test/testutil.py
@@ -1,10 +1,12 @@
+from __future__ import absolute_import
+
+import functools
import operator
import os
import socket
import time
import uuid
-import decorator
import pytest
from . import unittest
@@ -45,6 +47,7 @@ def kafka_versions(*versions):
validators = map(construct_lambda, versions)
def real_kafka_versions(func):
+ @functools.wraps(func)
def wrapper(func, *args, **kwargs):
version = kafka_version()
@@ -56,7 +59,7 @@ def kafka_versions(*versions):
pytest.skip("unsupported kafka version")
return func(*args, **kwargs)
- return decorator.decorator(wrapper, func)
+ return wrapper
return real_kafka_versions