crate 1.0.0.dev1__py3-none-any.whl → 1.0.1__py3-none-any.whl

crate/client/test_util.py

@@ -1,69 +0,0 @@
-# -*- coding: utf-8; -*-
-#
-# Licensed to CRATE Technology GmbH ("Crate") under one or more contributor
-# license agreements.  See the NOTICE file distributed with this work for
-# additional information regarding copyright ownership.  Crate licenses
-# this file to you under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.  You may
-# obtain a copy of the License at
-#
-#   http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
-# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
-# License for the specific language governing permissions and limitations
-# under the License.
-#
-# However, if you have executed another commercial license agreement
-# with Crate these terms will supersede the license and you may use the
-# software solely pursuant to the terms of the relevant commercial agreement.
-import unittest
-
-
-class ClientMocked(object):
-
-    active_servers = ["http://localhost:4200"]
-
-    def __init__(self):
-        self.response = {}
-        self._server_infos = ("http://localhost:4200", "my server", "2.0.0")
-
-    def sql(self, stmt=None, parameters=None, bulk_parameters=None):
-        return self.response
-
-    def server_infos(self, server):
-        return self._server_infos
-
-    def set_next_response(self, response):
-        self.response = response
-
-    def set_next_server_infos(self, server, server_name, version):
-        self._server_infos = (server, server_name, version)
-
-    def close(self):
-        pass
-
-
-class ParametrizedTestCase(unittest.TestCase):
-    """
-    TestCase classes that want to be parametrized should
-    inherit from this class.
-
-    https://eli.thegreenplace.net/2011/08/02/python-unit-testing-parametrized-test-cases
-    """
-    def __init__(self, methodName="runTest", param=None):
-        super(ParametrizedTestCase, self).__init__(methodName)
-        self.param = param
-
-    @staticmethod
-    def parametrize(testcase_klass, param=None):
-        """ Create a suite containing all tests taken from the given
-            subclass, passing them the parameter 'param'.
-        """
-        testloader = unittest.TestLoader()
-        testnames = testloader.getTestCaseNames(testcase_klass)
-        suite = unittest.TestSuite()
-        for name in testnames:
-            suite.addTest(testcase_klass(name, param=param))
-        return suite

crate/client/tests.py

@@ -1,340 +0,0 @@
-# -*- coding: utf-8; -*-
-#
-# Licensed to CRATE Technology GmbH ("Crate") under one or more contributor
-# license agreements.  See the NOTICE file distributed with this work for
-# additional information regarding copyright ownership.  Crate licenses
-# this file to you under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.  You may
-# obtain a copy of the License at
-#
-#   http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
-# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
-# License for the specific language governing permissions and limitations
-# under the License.
-#
-# However, if you have executed another commercial license agreement
-# with Crate these terms will supersede the license and you may use the
-# software solely pursuant to the terms of the relevant commercial agreement.
-
-from __future__ import absolute_import
-
-import json
-import os
-import socket
-import unittest
-import doctest
-from pprint import pprint
-from http.server import HTTPServer, BaseHTTPRequestHandler
-import ssl
-import time
-import threading
-import logging
-
-import stopit
-
-from crate.testing.layer import CrateLayer
-from crate.testing.settings import \
-    crate_host, crate_path, crate_port, \
-    crate_transport_port, docs_path, localhost
-from crate.client import connect
-
-from .test_cursor import CursorTest
-from .test_connection import ConnectionTest
-from .test_http import (
-    HttpClientTest,
-    ThreadSafeHttpClientTest,
-    KeepAliveClientTest,
-    ParamsTest,
-    RetryOnTimeoutServerTest,
-    RequestsCaBundleTest,
-    TestUsernameSentAsHeader,
-    TestCrateJsonEncoder,
-    TestDefaultSchemaHeader,
-)
-
-makeSuite = unittest.TestLoader().loadTestsFromTestCase
-
-log = logging.getLogger('crate.testing.layer')
-ch = logging.StreamHandler()
-ch.setLevel(logging.ERROR)
-log.addHandler(ch)
-
-
-def cprint(s):
-    if isinstance(s, bytes):
-        s = s.decode('utf-8')
-    print(s)
-
-
-settings = {
-    'udc.enabled': 'false',
-    'lang.js.enabled': 'true',
-    'auth.host_based.enabled': 'true',
-    'auth.host_based.config.0.user': 'crate',
-    'auth.host_based.config.0.method': 'trust',
-    'auth.host_based.config.98.user': 'trusted_me',
-    'auth.host_based.config.98.method': 'trust',
-    'auth.host_based.config.99.user': 'me',
-    'auth.host_based.config.99.method': 'password',
-}
-crate_layer = None
-
-
-def ensure_cratedb_layer():
-    """
-    In order to skip individual tests by manually disabling them within
-    `def test_suite()`, it is crucial make the test layer not run on each
-    and every occasion. So, things like this will be possible::
-
-        ./bin/test -vvvv --ignore_dir=testing
-
-    TODO: Through a subsequent patch, the possibility to individually
-          unselect specific tests might be added to `def test_suite()`
-          on behalf of environment variables.
-          A blueprint for this kind of logic can be found at
-          https://github.com/crate/crate/commit/414cd833.
-    """
-    global crate_layer
-
-    if crate_layer is None:
-        crate_layer = CrateLayer('crate',
-                                 crate_home=crate_path(),
-                                 port=crate_port,
-                                 host=localhost,
-                                 transport_port=crate_transport_port,
-                                 settings=settings)
-    return crate_layer
-
-
-def setUpCrateLayerBaseline(test):
-    test.globs['crate_host'] = crate_host
-    test.globs['pprint'] = pprint
-    test.globs['print'] = cprint
-
-    with connect(crate_host) as conn:
-        cursor = conn.cursor()
-
-        with open(docs_path('testing/testdata/mappings/locations.sql')) as s:
-            stmt = s.read()
-            cursor.execute(stmt)
-            stmt = ("select count(*) from information_schema.tables "
-                    "where table_name = 'locations'")
-            cursor.execute(stmt)
-            assert cursor.fetchall()[0][0] == 1
-
-        data_path = docs_path('testing/testdata/data/test_a.json')
-        # load testing data into crate
-        cursor.execute("copy locations from ?", (data_path,))
-        # refresh location table so imported data is visible immediately
-        cursor.execute("refresh table locations")
-        # create blob table
-        cursor.execute("create blob table myfiles clustered into 1 shards " +
-                       "with (number_of_replicas=0)")
-
-        # create users
-        cursor.execute("CREATE USER me WITH (password = 'my_secret_pw')")
-        cursor.execute("CREATE USER trusted_me")
-
-        cursor.close()
-
-
-def tearDownDropEntitiesBaseline(test):
-    """
-    Drop all tables, views, and users created by `setUpWithCrateLayer*`.
-    """
-    ddl_statements = [
-        "DROP TABLE locations",
-        "DROP BLOB TABLE myfiles",
-        "DROP USER me",
-        "DROP USER trusted_me",
-    ]
-    _execute_statements(ddl_statements)
-
-
-class HttpsTestServerLayer:
-    PORT = 65534
-    HOST = "localhost"
-    CERT_FILE = os.path.abspath(os.path.join(os.path.dirname(__file__),
-                                "pki/server_valid.pem"))
-    CACERT_FILE = os.path.abspath(os.path.join(os.path.dirname(__file__),
-                                  "pki/cacert_valid.pem"))
-
-    __name__ = "httpsserver"
-    __bases__ = tuple()
-
-    class HttpsServer(HTTPServer):
-        def get_request(self):
-
-            # Prepare SSL context.
-            context = ssl._create_unverified_context(
-                protocol=ssl.PROTOCOL_TLS_SERVER,
-                cert_reqs=ssl.CERT_OPTIONAL,
-                check_hostname=False,
-                purpose=ssl.Purpose.CLIENT_AUTH,
-                certfile=HttpsTestServerLayer.CERT_FILE,
-                keyfile=HttpsTestServerLayer.CERT_FILE,
-                cafile=HttpsTestServerLayer.CACERT_FILE)
-
-            # Set minimum protocol version, TLSv1 and TLSv1.1 are unsafe.
-            context.minimum_version = ssl.TLSVersion.TLSv1_2
-
-            # Wrap TLS encryption around socket.
-            socket, client_address = HTTPServer.get_request(self)
-            socket = context.wrap_socket(socket, server_side=True)
-
-            return socket, client_address
-
-    class HttpsHandler(BaseHTTPRequestHandler):
-
-        payload = json.dumps({"name": "test", "status": 200, })
-
-        def do_GET(self):
-            self.send_response(200)
-            payload = self.payload.encode('UTF-8')
-            self.send_header("Content-Length", len(payload))
-            self.send_header("Content-Type", "application/json; charset=UTF-8")
-            self.end_headers()
-            self.wfile.write(payload)
-
-    def setUp(self):
-        self.server = self.HttpsServer(
-            (self.HOST, self.PORT),
-            self.HttpsHandler
-        )
-        thread = threading.Thread(target=self.serve_forever)
-        thread.daemon = True  # quit interpreter when only thread exists
-        thread.start()
-        self.waitForServer()
-
-    def serve_forever(self):
-        print("listening on", self.HOST, self.PORT)
-        self.server.serve_forever()
-        print("server stopped.")
-
-    def tearDown(self):
-        self.server.shutdown()
-        self.server.server_close()
-
-    def isUp(self):
-        """
-        Test if a host is up.
-        """
-        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-        ex = s.connect_ex((self.HOST, self.PORT))
-        s.close()
-        return ex == 0
-
-    def waitForServer(self, timeout=5):
-        """
-        Wait for the host to be available.
-        """
-        with stopit.ThreadingTimeout(timeout) as to_ctx_mgr:
-            while True:
-                if self.isUp():
-                    break
-                time.sleep(0.001)
-
-        if not to_ctx_mgr:
-            raise TimeoutError("Could not properly start embedded webserver "
-                               "within {} seconds".format(timeout))
-
-
-def setUpWithHttps(test):
-    test.globs['crate_host'] = "https://{0}:{1}".format(
-        HttpsTestServerLayer.HOST, HttpsTestServerLayer.PORT
-    )
-    test.globs['pprint'] = pprint
-    test.globs['print'] = cprint
-
-    test.globs['cacert_valid'] = os.path.abspath(
-        os.path.join(os.path.dirname(__file__), "pki/cacert_valid.pem")
-    )
-    test.globs['cacert_invalid'] = os.path.abspath(
-        os.path.join(os.path.dirname(__file__), "pki/cacert_invalid.pem")
-    )
-    test.globs['clientcert_valid'] = os.path.abspath(
-        os.path.join(os.path.dirname(__file__), "pki/client_valid.pem")
-    )
-    test.globs['clientcert_invalid'] = os.path.abspath(
-        os.path.join(os.path.dirname(__file__), "pki/client_invalid.pem")
-    )
-
-
-def _execute_statements(statements, on_error="ignore"):
-    with connect(crate_host) as conn:
-        cursor = conn.cursor()
-        for stmt in statements:
-            _execute_statement(cursor, stmt, on_error=on_error)
-        cursor.close()
-
-
-def _execute_statement(cursor, stmt, on_error="ignore"):
-    try:
-        cursor.execute(stmt)
-    except Exception:  # pragma: no cover
-        # FIXME: Why does this croak on statements like ``DROP TABLE cities``?
-        # Note: When needing to debug the test environment, you may want to
-        #       enable this logger statement.
-        # log.exception("Executing SQL statement failed")
-        if on_error == "ignore":
-            pass
-        elif on_error == "raise":
-            raise
-
-
-def test_suite():
-    suite = unittest.TestSuite()
-    flags = (doctest.NORMALIZE_WHITESPACE | doctest.ELLIPSIS)
-
-    # Unit tests.
-    suite.addTest(makeSuite(CursorTest))
-    suite.addTest(makeSuite(HttpClientTest))
-    suite.addTest(makeSuite(KeepAliveClientTest))
-    suite.addTest(makeSuite(ThreadSafeHttpClientTest))
-    suite.addTest(makeSuite(ParamsTest))
-    suite.addTest(makeSuite(ConnectionTest))
-    suite.addTest(makeSuite(RetryOnTimeoutServerTest))
-    suite.addTest(makeSuite(RequestsCaBundleTest))
-    suite.addTest(makeSuite(TestUsernameSentAsHeader))
-    suite.addTest(makeSuite(TestCrateJsonEncoder))
-    suite.addTest(makeSuite(TestDefaultSchemaHeader))
-    suite.addTest(doctest.DocTestSuite('crate.client.connection'))
-    suite.addTest(doctest.DocTestSuite('crate.client.http'))
-
-    s = doctest.DocFileSuite(
-        'docs/by-example/connection.rst',
-        'docs/by-example/cursor.rst',
-        module_relative=False,
-        optionflags=flags,
-        encoding='utf-8'
-    )
-    suite.addTest(s)
-
-    s = doctest.DocFileSuite(
-        'docs/by-example/https.rst',
-        module_relative=False,
-        setUp=setUpWithHttps,
-        optionflags=flags,
-        encoding='utf-8'
-    )
-    s.layer = HttpsTestServerLayer()
-    suite.addTest(s)
-
-    # Integration tests.
-    s = doctest.DocFileSuite(
-        'docs/by-example/http.rst',
-        'docs/by-example/client.rst',
-        'docs/by-example/blob.rst',
-        module_relative=False,
-        setUp=setUpCrateLayerBaseline,
-        tearDown=tearDownDropEntitiesBaseline,
-        optionflags=flags,
-        encoding='utf-8'
-    )
-    s.layer = ensure_cratedb_layer()
-    suite.addTest(s)
-
-    return suite

crate/testing/settings.py

@@ -1,51 +0,0 @@
-# vi: set encoding=utf-8
-# -*- coding: utf-8; -*-
-#
-# Licensed to CRATE Technology GmbH ("Crate") under one or more contributor
-# license agreements.  See the NOTICE file distributed with this work for
-# additional information regarding copyright ownership.  Crate licenses
-# this file to you under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.  You may
-# obtain a copy of the License at
-#
-#   http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
-# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
-# License for the specific language governing permissions and limitations
-# under the License.
-#
-# However, if you have executed another commercial license agreement
-# with Crate these terms will supersede the license and you may use the
-# software solely pursuant to the terms of the relevant commercial agreement.
-from __future__ import absolute_import
-
-import os
-
-
-def docs_path(*parts):
-    return os.path.abspath(
-        os.path.join(
-            os.path.dirname(os.path.dirname(__file__)), *parts
-        )
-    )
-
-
-def project_root(*parts):
-    return os.path.abspath(
-        os.path.join(docs_path("..", ".."), *parts)
-    )
-
-
-def crate_path(*parts):
-    return os.path.abspath(
-        project_root("parts", "crate", *parts)
-    )
-
-
-crate_port = 44209
-crate_transport_port = 44309
-localhost = '127.0.0.1'
-crate_host = "{host}:{port}".format(host=localhost, port=crate_port)
-crate_uri = "http://%s" % crate_host

crate/testing/test_datetime_old.py

@@ -1,90 +0,0 @@
-# -*- coding: utf-8; -*-
-#
-# Licensed to CRATE Technology GmbH ("Crate") under one or more contributor
-# license agreements.  See the NOTICE file distributed with this work for
-# additional information regarding copyright ownership.  Crate licenses
-# this file to you under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.  You may
-# obtain a copy of the License at
-#
-#   http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
-# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
-# License for the specific language governing permissions and limitations
-# under the License.
-#
-# However, if you have executed another commercial license agreement
-# with Crate these terms will supersede the license and you may use the
-# software solely pursuant to the terms of the relevant commercial agreement.
-
-import os
-import sys
-import unittest
-from datetime import datetime, date
-from unittest import TestCase, mock
-
-import time_machine
-
-
-class DateTodayTest(TestCase):
-    """
-    `date.today()` returns the current local date, as advertised in the
-    documentation [1]. Thus, it depends on the system time zone.
-
-    The following test cases demonstrate that the test suite previously
-    failed around midnight, where the UTC vs. non-UTC days overlapped,
-    and when running on machines with non-UTC time zone.
-
-    On the other hand, `datetime.utcnow().date()` works equally well in all
-    situations, so we want to use that within the SQLAlchemy test cases.
-
-    Funny enough, the problem is not observable on Linux?
-
-    [1] https://docs.python.org/3/library/datetime.html#datetime.date.today
-    """
-
-    @mock.patch.dict(os.environ, {"TZ": "UTC"})
-    @time_machine.travel("2022-07-22T00:42:00+0200")
-    def test_date_today_depends_on_system_timezone_success_on_utc(self):
-        today_local = date.today()
-        today_utc = datetime.utcnow().date()
-        self.assertEqual(today_local, today_utc)
-        self.assertEqual(today_local, date(2022, 7, 21))
-        self.assertEqual(today_utc, date(2022, 7, 21))
-
-    @unittest.skipIf(sys.platform == "linux", "Problem not observable on Linux")
-    @mock.patch.dict(os.environ, {"TZ": "Europe/Prague"})
-    @time_machine.travel("2022-07-22T00:42:00+0200")
-    def test_date_today_depends_on_system_timezone_failure_on_non_utc(self):
-        today_local = date.today()
-        today_utc = datetime.utcnow().date()
-        self.assertNotEqual(today_local, today_utc)
-        self.assertEqual(today_local, date(2022, 7, 22))
-        self.assertEqual(today_utc, date(2022, 7, 21))
-
-    @mock.patch.dict(os.environ, {"TZ": "UTC"})
-    @time_machine.travel("2022-07-22T00:42:00+0200")
-    def test_date_today_utc(self):
-        today_local = date.today()
-        self.assertEqual(today_local, date(2022, 7, 21))
-
-    @unittest.skipIf(sys.platform == "linux", "Problem not observable on Linux")
-    @mock.patch.dict(os.environ, {"TZ": "Europe/Prague"})
-    @time_machine.travel("2022-07-22T00:42:00+0200")
-    def test_date_today_non_utc(self):
-        today_local = date.today()
-        self.assertEqual(today_local, date(2022, 7, 22))
-
-    @mock.patch.dict(os.environ, {"TZ": "UTC"})
-    @time_machine.travel("2022-07-22T00:42:00+0200")
-    def test_utcnow_date_utc(self):
-        today_utc = datetime.utcnow().date()
-        self.assertEqual(today_utc, date(2022, 7, 21))
-
-    @mock.patch.dict(os.environ, {"TZ": "Europe/Prague"})
-    @time_machine.travel("2022-07-22T00:42:00+0200")
-    def test_utcnow_date_non_utc(self):
-        today_utc = datetime.utcnow().date()
-        self.assertEqual(today_utc, date(2022, 7, 21))