crate 0.35.2__py2.py3-none-any.whl → 1.0.0.dev0__py2.py3-none-any.whl

crate/client/sqlalchemy/tests/match_test.py

@@ -1,137 +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 unittest import TestCase
-from unittest.mock import MagicMock
-
-import sqlalchemy as sa
-from sqlalchemy.orm import Session
-try:
-    from sqlalchemy.orm import declarative_base
-except ImportError:
-    from sqlalchemy.ext.declarative import declarative_base
-
-from crate.client.sqlalchemy.types import ObjectType
-from crate.client.sqlalchemy.predicates import match
-from crate.client.cursor import Cursor
-
-
-fake_cursor = MagicMock(name='fake_cursor')
-FakeCursor = MagicMock(name='FakeCursor', spec=Cursor)
-FakeCursor.return_value = fake_cursor
-
-
-class SqlAlchemyMatchTest(TestCase):
-
-    def setUp(self):
-        self.engine = sa.create_engine('crate://')
-        metadata = sa.MetaData()
-        self.quotes = sa.Table('quotes', metadata,
-                               sa.Column('author', sa.String),
-                               sa.Column('quote', sa.String))
-        self.session, self.Character = self.set_up_character_and_session()
-        self.maxDiff = None
-
-    def assertSQL(self, expected_str, actual_expr):
-        self.assertEqual(expected_str, str(actual_expr).replace('\n', ''))
-
-    def set_up_character_and_session(self):
-        Base = declarative_base()
-
-        class Character(Base):
-            __tablename__ = 'characters'
-            name = sa.Column(sa.String, primary_key=True)
-            info = sa.Column(ObjectType)
-
-        session = Session(bind=self.engine)
-        return session, Character
-
-    def test_simple_match(self):
-        query = self.session.query(self.Character.name) \
-                    .filter(match(self.Character.name, 'Trillian'))
-        self.assertSQL(
-            "SELECT characters.name AS characters_name FROM characters " +
-            "WHERE match(characters.name, ?)",
-            query
-        )
-
-    def test_match_boost(self):
-        query = self.session.query(self.Character.name) \
-            .filter(match({self.Character.name: 0.5}, 'Trillian'))
-        self.assertSQL(
-            "SELECT characters.name AS characters_name FROM characters " +
-            "WHERE match((characters.name 0.5), ?)",
-            query
-        )
-
-    def test_muli_match(self):
-        query = self.session.query(self.Character.name) \
-            .filter(match({self.Character.name: 0.5,
-                           self.Character.info['race']: 0.9},
-                          'Trillian'))
-        self.assertSQL(
-            "SELECT characters.name AS characters_name FROM characters " +
-            "WHERE match(" +
-            "(characters.info['race'] 0.9, characters.name 0.5), ?" +
-            ")",
-            query
-        )
-
-    def test_match_type_options(self):
-        query = self.session.query(self.Character.name) \
-            .filter(match({self.Character.name: 0.5,
-                           self.Character.info['race']: 0.9},
-                          'Trillian',
-                          match_type='phrase',
-                          options={'fuzziness': 3, 'analyzer': 'english'}))
-        self.assertSQL(
-            "SELECT characters.name AS characters_name FROM characters " +
-            "WHERE match(" +
-            "(characters.info['race'] 0.9, characters.name 0.5), ?" +
-            ") using phrase with (analyzer=english, fuzziness=3)",
-            query
-        )
-
-    def test_score(self):
-        query = self.session.query(self.Character.name,
-                                   sa.literal_column('_score')) \
-                    .filter(match(self.Character.name, 'Trillian'))
-        self.assertSQL(
-            "SELECT characters.name AS characters_name, _score " +
-            "FROM characters WHERE match(characters.name, ?)",
-            query
-        )
-
-    def test_options_without_type(self):
-        query = self.session.query(self.Character.name).filter(
-            match({self.Character.name: 0.5, self.Character.info['race']: 0.9},
-                  'Trillian',
-                  options={'boost': 10.0})
-        )
-        err = None
-        try:
-            str(query)
-        except ValueError as e:
-            err = e
-        msg = "missing match_type. " + \
-              "It's not allowed to specify options without match_type"
-        self.assertEqual(str(err), msg)

crate/client/sqlalchemy/tests/query_caching.py

@@ -1,143 +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
-from unittest import TestCase, skipIf
-
-import sqlalchemy as sa
-from sqlalchemy.orm import Session
-from sqlalchemy.sql.operators import eq
-
-from crate.client.sqlalchemy import SA_VERSION, SA_1_4
-from crate.testing.settings import crate_host
-
-try:
-    from sqlalchemy.orm import declarative_base
-except ImportError:
-    from sqlalchemy.ext.declarative import declarative_base
-
-from crate.client.sqlalchemy.types import ObjectType, ObjectArray
-
-
-class SqlAlchemyQueryCompilationCaching(TestCase):
-
-    def setUp(self):
-        self.engine = sa.create_engine(f"crate://{crate_host}")
-        self.metadata = sa.MetaData(schema="testdrive")
-        self.session = Session(bind=self.engine)
-        self.Character = self.setup_entity()
-
-    def setup_entity(self):
-        """
-        Define ORM entity.
-        """
-        Base = declarative_base(metadata=self.metadata)
-
-        class Character(Base):
-            __tablename__ = 'characters'
-            name = sa.Column(sa.String, primary_key=True)
-            age = sa.Column(sa.Integer)
-            data = sa.Column(ObjectType)
-            data_list = sa.Column(ObjectArray)
-
-        return Character
-
-    def setup_data(self):
-        """
-        Insert two records into the `characters` table.
-        """
-        self.metadata.drop_all(self.engine)
-        self.metadata.create_all(self.engine)
-
-        Character = self.Character
-        char1 = Character(name='Trillian', data={'x': 1}, data_list=[{'foo': 1, 'bar': 10}])
-        char2 = Character(name='Slartibartfast', data={'y': 2}, data_list=[{'bar': 2}])
-        self.session.add(char1)
-        self.session.add(char2)
-        self.session.commit()
-        self.session.execute(sa.text("REFRESH TABLE testdrive.characters;"))
-
-    @skipIf(SA_VERSION < SA_1_4, "On SA13, the 'ResultProxy' object has no attribute 'scalar_one'")
-    def test_object_multiple_select_legacy(self):
-        """
-        The SQLAlchemy implementation of CrateDB's `OBJECT` type offers indexed
-        access to the instance's content in form of a dictionary. Thus, it must
-        not use `cache_ok = True` on its implementation, i.e. this part of the
-        compiled SQL clause must not be cached.
-
-        This test verifies that two subsequent `SELECT` statements are translated
-        well, and don't trip on incorrect SQL compiled statement caching.
-
-        This variant uses direct value matching on the `OBJECT`s attribute.
-        """
-        self.setup_data()
-        Character = self.Character
-
-        selectable = sa.select(Character).where(Character.data['x'] == 1)
-        result = self.session.execute(selectable).scalar_one().data
-        self.assertEqual({"x": 1}, result)
-
-        selectable = sa.select(Character).where(Character.data['y'] == 2)
-        result = self.session.execute(selectable).scalar_one().data
-        self.assertEqual({"y": 2}, result)
-
-    @skipIf(SA_VERSION < SA_1_4, "On SA13, the 'ResultProxy' object has no attribute 'scalar_one'")
-    def test_object_multiple_select_modern(self):
-        """
-        The SQLAlchemy implementation of CrateDB's `OBJECT` type offers indexed
-        access to the instance's content in form of a dictionary. Thus, it must
-        not use `cache_ok = True` on its implementation, i.e. this part of the
-        compiled SQL clause must not be cached.
-
-        This test verifies that two subsequent `SELECT` statements are translated
-        well, and don't trip on incorrect SQL compiled statement caching.
-
-        This variant uses comparator method matching on the `OBJECT`s attribute.
-        """
-        self.setup_data()
-        Character = self.Character
-
-        selectable = sa.select(Character).where(Character.data['x'].as_integer() == 1)
-        result = self.session.execute(selectable).scalar_one().data
-        self.assertEqual({"x": 1}, result)
-
-        selectable = sa.select(Character).where(Character.data['y'].as_integer() == 2)
-        result = self.session.execute(selectable).scalar_one().data
-        self.assertEqual({"y": 2}, result)
-
-    @skipIf(SA_VERSION < SA_1_4, "On SA13, the 'ResultProxy' object has no attribute 'scalar_one'")
-    def test_objectarray_multiple_select(self):
-        """
-        The SQLAlchemy implementation of CrateDB's `ARRAY` type in form of the
-        `ObjectArray`, does *not* offer indexed access to the instance's content.
-        Thus, using `cache_ok = True` on that type should be sane, and not mess
-        up SQLAlchemy's SQL compiled statement caching.
-        """
-        self.setup_data()
-        Character = self.Character
-
-        selectable = sa.select(Character).where(Character.data_list['foo'].any(1, operator=eq))
-        result = self.session.execute(selectable).scalar_one().data
-        self.assertEqual({"x": 1}, result)
-
-        selectable = sa.select(Character).where(Character.data_list['bar'].any(2, operator=eq))
-        result = self.session.execute(selectable).scalar_one().data
-        self.assertEqual({"y": 2}, result)

crate/client/sqlalchemy/tests/update_test.py

@@ -1,115 +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 datetime import datetime
-from unittest import TestCase
-from unittest.mock import patch, MagicMock
-
-from crate.client.sqlalchemy.types import ObjectType
-
-import sqlalchemy as sa
-from sqlalchemy.orm import Session
-try:
-    from sqlalchemy.orm import declarative_base
-except ImportError:
-    from sqlalchemy.ext.declarative import declarative_base
-
-from crate.client.cursor import Cursor
-
-
-fake_cursor = MagicMock(name='fake_cursor')
-fake_cursor.rowcount = 1
-FakeCursor = MagicMock(name='FakeCursor', spec=Cursor)
-FakeCursor.return_value = fake_cursor
-
-
-class SqlAlchemyUpdateTest(TestCase):
-
-    def setUp(self):
-        self.engine = sa.create_engine('crate://')
-        self.base = declarative_base()
-
-        class Character(self.base):
-            __tablename__ = 'characters'
-
-            name = sa.Column(sa.String, primary_key=True)
-            age = sa.Column(sa.Integer)
-            obj = sa.Column(ObjectType)
-            ts = sa.Column(sa.DateTime, onupdate=datetime.utcnow)
-
-        self.character = Character
-        self.session = Session(bind=self.engine)
-
-    @patch('crate.client.connection.Cursor', FakeCursor)
-    def test_onupdate_is_triggered(self):
-        char = self.character(name='Arthur')
-        self.session.add(char)
-        self.session.commit()
-        now = datetime.utcnow()
-
-        fake_cursor.fetchall.return_value = [('Arthur', None)]
-        fake_cursor.description = (
-            ('characters_name', None, None, None, None, None, None),
-            ('characters_ts', None, None, None, None, None, None),
-        )
-
-        char.age = 40
-        self.session.commit()
-
-        expected_stmt = ("UPDATE characters SET age = ?, "
-                         "ts = ? WHERE characters.name = ?")
-        args, kwargs = fake_cursor.execute.call_args
-        stmt = args[0]
-        args = args[1]
-        self.assertEqual(expected_stmt, stmt)
-        self.assertEqual(40, args[0])
-        dt = datetime.strptime(args[1], '%Y-%m-%dT%H:%M:%S.%fZ')
-        self.assertIsInstance(dt, datetime)
-        self.assertGreater(dt, now)
-        self.assertEqual('Arthur', args[2])
-
-    @patch('crate.client.connection.Cursor', FakeCursor)
-    def test_bulk_update(self):
-        """
-            Checks whether bulk updates work correctly
-            on native types and Crate types.
-        """
-        before_update_time = datetime.utcnow()
-
-        self.session.query(self.character).update({
-            # change everyone's name to Julia
-            self.character.name: 'Julia',
-            self.character.obj: {'favorite_book': 'Romeo & Juliet'}
-        })
-
-        self.session.commit()
-
-        expected_stmt = ("UPDATE characters SET "
-                         "name = ?, obj = ?, ts = ?")
-        args, kwargs = fake_cursor.execute.call_args
-        stmt = args[0]
-        args = args[1]
-        self.assertEqual(expected_stmt, stmt)
-        self.assertEqual('Julia', args[0])
-        self.assertEqual({'favorite_book': 'Romeo & Juliet'}, args[1])
-        dt = datetime.strptime(args[2], '%Y-%m-%dT%H:%M:%S.%fZ')
-        self.assertIsInstance(dt, datetime)
-        self.assertGreater(dt, before_update_time)

crate/client/sqlalchemy/tests/warnings_test.py

@@ -1,64 +0,0 @@
-# -*- coding: utf-8; -*-
-import sys
-import warnings
-from unittest import TestCase, skipIf
-
-from crate.client.sqlalchemy import SA_1_4, SA_VERSION
-from crate.testing.util import ExtraAssertions
-
-
-class SqlAlchemyWarningsTest(TestCase, ExtraAssertions):
-    """
-    Verify a few `DeprecationWarning` spots.
-
-    https://docs.python.org/3/library/warnings.html#testing-warnings
-    """
-
-    @skipIf(SA_VERSION >= SA_1_4, "There is no deprecation warning for "
-                                  "SQLAlchemy 1.3 on higher versions")
-    def test_sa13_deprecation_warning(self):
-        """
-        Verify that a `DeprecationWarning` is issued when running SQLAlchemy 1.3.
-        """
-        with warnings.catch_warnings(record=True) as w:
-
-            # Cause all warnings to always be triggered.
-            warnings.simplefilter("always")
-
-            # Trigger a warning by importing the SQLAlchemy dialect module.
-            # Because it already has been loaded, unload it beforehand.
-            del sys.modules["crate.client.sqlalchemy"]
-            import crate.client.sqlalchemy  # noqa: F401
-
-            # Verify details of the SA13 EOL/deprecation warning.
-            self.assertEqual(len(w), 1)
-            self.assertIsSubclass(w[-1].category, DeprecationWarning)
-            self.assertIn("SQLAlchemy 1.3 is effectively EOL.", str(w[-1].message))
-
-    def test_craty_object_deprecation_warning(self):
-        """
-        Verify that a `DeprecationWarning` is issued when accessing the deprecated
-        module variables `Craty`, and `Object`. The new type is called `ObjectType`.
-        """
-
-        with warnings.catch_warnings(record=True) as w:
-
-            # Import the deprecated symbol.
-            from crate.client.sqlalchemy.types import Craty  # noqa: F401
-
-            # Verify details of the deprecation warning.
-            self.assertEqual(len(w), 1)
-            self.assertIsSubclass(w[-1].category, DeprecationWarning)
-            self.assertIn("Craty is deprecated and will be removed in future releases. "
-                          "Please use ObjectType instead.", str(w[-1].message))
-
-        with warnings.catch_warnings(record=True) as w:
-
-            # Import the deprecated symbol.
-            from crate.client.sqlalchemy.types import Object  # noqa: F401
-
-            # Verify details of the deprecation warning.
-            self.assertEqual(len(w), 1)
-            self.assertIsSubclass(w[-1].category, DeprecationWarning)
-            self.assertIn("Object is deprecated and will be removed in future releases. "
-                          "Please use ObjectType instead.", str(w[-1].message))