Add test

.travis.yml

@@ -13,6 +13,8 @@ matrix:
     env: TOXENV=py37
 install:
 - pip install tox
+before_install:
+- export BOTO_CONFIG=/dev/null
 script:
 - tox
 notifications:

rekcurd_client/__init__.py

@@ -18,4 +18,4 @@
 __project__ = _project.__project__
 __version__ = _version.__version__
 
-from .rekcurd_worker_client import RekcurdWorkerClient
+from .core import RekcurdWorkerClient

rekcurd_client/core/__init__.py

@@ -0,0 +1 @@
+from .rekcurd_worker_client import RekcurdWorkerClient

rekcurd_client/rekcurd_worker_client.py → rekcurd_client/core/rekcurd_worker_client.py

@@ -6,8 +6,8 @@
 import types
 import grpc
 
-from .protobuf import rekcurd_pb2, rekcurd_pb2_grpc
-from .logger import SystemLoggerInterface
+from rekcurd_client.protobuf import rekcurd_pb2, rekcurd_pb2_grpc
+from rekcurd_client.logger import SystemLoggerInterface
 
 
 def error_handling(error_response):

rekcurd_client/logger/logger_fluent.py

@@ -32,6 +32,7 @@ def __init__(self, logger_name: str = 'rekcurd_client', log_level: int = logging
         formatter = handler.FluentRecordFormatter(custom_format)
         fluent_handler.setFormatter(formatter)
         fluent_handler.setLevel(log_level)
+        self.log.handlers = []
         self.log.addHandler(fluent_handler)
 
     def exception(self, message: str) -> None:

rekcurd_client/logger/logger_jsonlogger.py

@@ -36,6 +36,7 @@ def __init__(self, logger_name: str = 'rekcurd_client', log_level: int = logging
         handler = logging.StreamHandler()
         formatter = self.JsonFormatter()
         handler.setFormatter(formatter)
+        self.log.handlers = []
         self.log.addHandler(handler)
         self.log.setLevel(log_level)
 

setup.py

@@ -61,8 +61,8 @@
     install_requires=REQUIRES,
     tests_require=TESTS_REQUIRES,
     extras_require=EXTRAS,
-    packages=['rekcurd_client', 'rekcurd_client.logger',
-              'rekcurd_client.protobuf'],
+    packages=['rekcurd_client', 'rekcurd_client.core',
+              'rekcurd_client.logger', 'rekcurd_client.protobuf'],
     include_package_data=True,
     long_description=LONG_DESCRIPTION,
     classifiers=[

test/_client_application.py → test/core/_client_application.py

@@ -5,7 +5,7 @@
 
 from rekcurd.utils import PredictResult
 
-import rekcurd_client.rekcurd_worker_client
+import rekcurd_client.core.rekcurd_worker_client
 from rekcurd_client.protobuf import rekcurd_pb2_grpc
 from rekcurd_client.logger import logger
 from rekcurd_client import RekcurdWorkerClient
@@ -409,7 +409,7 @@ def _run_arrstring_arrstring(client: RekcurdWorkerClient):
 
 def run(scenario, channel):
     stub = rekcurd_pb2_grpc.RekcurdWorkerStub(channel)
-    client = rekcurd_client.rekcurd_worker_client.RekcurdWorkerClient(logger=logger, domain='example.com', app='rekcurd-sample', env='development')
+    client = rekcurd_client.core.rekcurd_worker_client.RekcurdWorkerClient(logger=logger, domain='example.com', app='rekcurd-sample', env='development')
     client.stub = stub
     try:
         return _IMPLEMENTATIONS[scenario](client)

test/test_worker_client.py → test/core/test_worker_client.py

@@ -6,8 +6,7 @@
 import grpc_testing
 
 from rekcurd_client.protobuf import rekcurd_pb2
-from . import _client_application
-
+from test.core import _client_application
 
 target_service = rekcurd_pb2.DESCRIPTOR.services_by_name['RekcurdWorker']
 

test/test_worker_client_e2e.py → test/core/test_worker_client_e2e.py

@@ -1,14 +1,14 @@
 import grpc
 from concurrent import futures
 
-from . import *
+from test import *
 from rekcurd_client.protobuf import rekcurd_pb2_grpc
 import unittest
 from functools import wraps
 from unittest.mock import patch, Mock
 
 from rekcurd.utils import PredictResult
-import rekcurd_client.rekcurd_worker_client
+import rekcurd_client.core.rekcurd_worker_client
 
 
 def patch_predictor(input_type, output_type):
@@ -87,7 +87,7 @@ def setUpClass(cls):
         server.add_insecure_port("[::]:5000")
         server.start()
         cls.server = server
-        cls.client = rekcurd_client.rekcurd_worker_client.RekcurdWorkerClient(logger=client_logger, host='127.0.0.1:5000')
+        cls.client = rekcurd_client.core.rekcurd_worker_client.RekcurdWorkerClient(logger=client_logger, host='127.0.0.1:5000')
 
     @classmethod
     def tearDownClass(cls):

test/logger/test_logger_fluent.py

@@ -0,0 +1,26 @@
+import unittest
+
+from rekcurd_client.logger import FluentSystemLogger
+
+
+class FluentSystemLoggerTest(unittest.TestCase):
+    """Tests for FluentSystemLogger.
+    """
+
+    def setUp(self):
+        self.logger = FluentSystemLogger()
+
+    def test_exception(self):
+        self.assertIsNone(self.logger.exception("Exception"))
+
+    def test_error(self):
+        self.assertIsNone(self.logger.error("Error"))
+
+    def test_debug(self):
+        self.assertIsNone(self.logger.debug("Debug"))
+
+    def test_info(self):
+        self.assertIsNone(self.logger.info("Info"))
+
+    def test_warn(self):
+        self.assertIsNone(self.logger.warn("Warn"))

test/logger/test_logger_jsonlogger.py

@@ -0,0 +1,26 @@
+import unittest
+
+from rekcurd_client.logger import JsonSystemLogger
+
+
+class JsonSystemLoggerTest(unittest.TestCase):
+    """Tests for JsonSystemLogger.
+    """
+
+    def setUp(self):
+        self.logger = JsonSystemLogger()
+
+    def test_exception(self):
+        self.assertIsNone(self.logger.exception("Exception"))
+
+    def test_error(self):
+        self.assertIsNone(self.logger.error("Error"))
+
+    def test_debug(self):
+        self.assertIsNone(self.logger.debug("Debug"))
+
+    def test_info(self):
+        self.assertIsNone(self.logger.info("Info"))
+
+    def test_warn(self):
+        self.assertIsNone(self.logger.warn("Warn"))

tox.ini

@@ -10,11 +10,14 @@ deps = -r{toxinidir}/test-requirements.txt
 commands =
    python -V
    py.test -vvv -s
+# Patch for https://github.com/travis-ci/travis-ci/issues/7940
+setenv =
+   BOTO_CONFIG = /dev/null
 
 [testenv:coverage]
 commands =
    python -V
-   nosetests --with-coverage --cover-package=rekcurd_client --cover-tests
+   nosetests --with-coverage --cover-package=rekcurd_client.core,rekcurd_client.logger --cover-tests
 
 [testenv:codecov]
 commands =