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

crate/client/sqlalchemy/tests/create_table_test.py

@@ -1,313 +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 sqlalchemy as sa
-try:
-    from sqlalchemy.orm import declarative_base
-except ImportError:
-    from sqlalchemy.ext.declarative import declarative_base
-
-from crate.client.sqlalchemy.types import ObjectType, ObjectArray, Geopoint
-from crate.client.cursor import Cursor
-
-from unittest import TestCase
-from unittest.mock import patch, MagicMock
-
-
-fake_cursor = MagicMock(name='fake_cursor')
-FakeCursor = MagicMock(name='FakeCursor', spec=Cursor)
-FakeCursor.return_value = fake_cursor
-
-
-@patch('crate.client.connection.Cursor', FakeCursor)
-class SqlAlchemyCreateTableTest(TestCase):
-
-    def setUp(self):
-        self.engine = sa.create_engine('crate://')
-        self.Base = declarative_base()
-
-    def test_table_basic_types(self):
-        class User(self.Base):
-            __tablename__ = 'users'
-            string_col = sa.Column(sa.String, primary_key=True)
-            unicode_col = sa.Column(sa.Unicode)
-            text_col = sa.Column(sa.Text)
-            int_col = sa.Column(sa.Integer)
-            long_col1 = sa.Column(sa.BigInteger)
-            long_col2 = sa.Column(sa.NUMERIC)
-            bool_col = sa.Column(sa.Boolean)
-            short_col = sa.Column(sa.SmallInteger)
-            datetime_col = sa.Column(sa.DateTime)
-            date_col = sa.Column(sa.Date)
-            float_col = sa.Column(sa.Float)
-            double_col = sa.Column(sa.DECIMAL)
-
-        self.Base.metadata.create_all(bind=self.engine)
-        fake_cursor.execute.assert_called_with(
-            ('\nCREATE TABLE users (\n\tstring_col STRING NOT NULL, '
-             '\n\tunicode_col STRING, \n\ttext_col STRING, \n\tint_col INT, '
-             '\n\tlong_col1 LONG, \n\tlong_col2 LONG, '
-             '\n\tbool_col BOOLEAN, '
-             '\n\tshort_col SHORT, '
-             '\n\tdatetime_col TIMESTAMP, \n\tdate_col TIMESTAMP, '
-             '\n\tfloat_col FLOAT, \n\tdouble_col DOUBLE, '
-             '\n\tPRIMARY KEY (string_col)\n)\n\n'),
-            ())
-
-    def test_column_obj(self):
-        class DummyTable(self.Base):
-            __tablename__ = 'dummy'
-            pk = sa.Column(sa.String, primary_key=True)
-            obj_col = sa.Column(ObjectType)
-        self.Base.metadata.create_all(bind=self.engine)
-        fake_cursor.execute.assert_called_with(
-            ('\nCREATE TABLE dummy (\n\tpk STRING NOT NULL, \n\tobj_col OBJECT, '
-             '\n\tPRIMARY KEY (pk)\n)\n\n'),
-            ())
-
-    def test_table_clustered_by(self):
-        class DummyTable(self.Base):
-            __tablename__ = 't'
-            __table_args__ = {
-                'crate_clustered_by': 'p'
-            }
-            pk = sa.Column(sa.String, primary_key=True)
-            p = sa.Column(sa.String)
-        self.Base.metadata.create_all(bind=self.engine)
-        fake_cursor.execute.assert_called_with(
-            ('\nCREATE TABLE t (\n\t'
-             'pk STRING NOT NULL, \n\t'
-             'p STRING, \n\t'
-             'PRIMARY KEY (pk)\n'
-             ') CLUSTERED BY (p)\n\n'),
-            ())
-
-    def test_column_computed(self):
-        class DummyTable(self.Base):
-            __tablename__ = 't'
-            ts = sa.Column(sa.BigInteger, primary_key=True)
-            p = sa.Column(sa.BigInteger, sa.Computed("date_trunc('day', ts)"))
-        self.Base.metadata.create_all(bind=self.engine)
-        fake_cursor.execute.assert_called_with(
-            ('\nCREATE TABLE t (\n\t'
-             'ts LONG NOT NULL, \n\t'
-             'p LONG GENERATED ALWAYS AS (date_trunc(\'day\', ts)), \n\t'
-             'PRIMARY KEY (ts)\n'
-             ')\n\n'),
-            ())
-
-    def test_column_computed_virtual(self):
-        class DummyTable(self.Base):
-            __tablename__ = 't'
-            ts = sa.Column(sa.BigInteger, primary_key=True)
-            p = sa.Column(sa.BigInteger, sa.Computed("date_trunc('day', ts)", persisted=False))
-        with self.assertRaises(sa.exc.CompileError):
-            self.Base.metadata.create_all(bind=self.engine)
-
-    def test_table_partitioned_by(self):
-        class DummyTable(self.Base):
-            __tablename__ = 't'
-            __table_args__ = {
-                'crate_partitioned_by': 'p',
-                'invalid_option': 1
-            }
-            pk = sa.Column(sa.String, primary_key=True)
-            p = sa.Column(sa.String)
-        self.Base.metadata.create_all(bind=self.engine)
-        fake_cursor.execute.assert_called_with(
-            ('\nCREATE TABLE t (\n\t'
-             'pk STRING NOT NULL, \n\t'
-             'p STRING, \n\t'
-             'PRIMARY KEY (pk)\n'
-             ') PARTITIONED BY (p)\n\n'),
-            ())
-
-    def test_table_number_of_shards_and_replicas(self):
-        class DummyTable(self.Base):
-            __tablename__ = 't'
-            __table_args__ = {
-                'crate_number_of_replicas': '2',
-                'crate_number_of_shards': 3
-            }
-            pk = sa.Column(sa.String, primary_key=True)
-
-        self.Base.metadata.create_all(bind=self.engine)
-        fake_cursor.execute.assert_called_with(
-            ('\nCREATE TABLE t (\n\t'
-             'pk STRING NOT NULL, \n\t'
-             'PRIMARY KEY (pk)\n'
-             ') CLUSTERED INTO 3 SHARDS WITH (NUMBER_OF_REPLICAS = 2)\n\n'),
-            ())
-
-    def test_table_clustered_by_and_number_of_shards(self):
-        class DummyTable(self.Base):
-            __tablename__ = 't'
-            __table_args__ = {
-                'crate_clustered_by': 'p',
-                'crate_number_of_shards': 3
-            }
-            pk = sa.Column(sa.String, primary_key=True)
-            p = sa.Column(sa.String, primary_key=True)
-        self.Base.metadata.create_all(bind=self.engine)
-        fake_cursor.execute.assert_called_with(
-            ('\nCREATE TABLE t (\n\t'
-             'pk STRING NOT NULL, \n\t'
-             'p STRING NOT NULL, \n\t'
-             'PRIMARY KEY (pk, p)\n'
-             ') CLUSTERED BY (p) INTO 3 SHARDS\n\n'),
-            ())
-
-    def test_column_object_array(self):
-        class DummyTable(self.Base):
-            __tablename__ = 't'
-            pk = sa.Column(sa.String, primary_key=True)
-            tags = sa.Column(ObjectArray)
-
-        self.Base.metadata.create_all(bind=self.engine)
-        fake_cursor.execute.assert_called_with(
-            ('\nCREATE TABLE t (\n\t'
-             'pk STRING NOT NULL, \n\t'
-             'tags ARRAY(OBJECT), \n\t'
-             'PRIMARY KEY (pk)\n)\n\n'), ())
-
-    def test_column_nullable(self):
-        class DummyTable(self.Base):
-            __tablename__ = 't'
-            pk = sa.Column(sa.String, primary_key=True)
-            a = sa.Column(sa.Integer, nullable=True)
-            b = sa.Column(sa.Integer, nullable=False)
-
-        self.Base.metadata.create_all(bind=self.engine)
-        fake_cursor.execute.assert_called_with(
-            ('\nCREATE TABLE t (\n\t'
-             'pk STRING NOT NULL, \n\t'
-             'a INT, \n\t'
-             'b INT NOT NULL, \n\t'
-             'PRIMARY KEY (pk)\n)\n\n'), ())
-
-    def test_column_pk_nullable(self):
-        class DummyTable(self.Base):
-            __tablename__ = 't'
-            pk = sa.Column(sa.String, primary_key=True, nullable=True)
-        with self.assertRaises(sa.exc.CompileError):
-            self.Base.metadata.create_all(bind=self.engine)
-
-    def test_column_crate_index(self):
-        class DummyTable(self.Base):
-            __tablename__ = 't'
-            pk = sa.Column(sa.String, primary_key=True)
-            a = sa.Column(sa.Integer, crate_index=False)
-            b = sa.Column(sa.Integer, crate_index=True)
-
-        self.Base.metadata.create_all(bind=self.engine)
-        fake_cursor.execute.assert_called_with(
-            ('\nCREATE TABLE t (\n\t'
-             'pk STRING NOT NULL, \n\t'
-             'a INT INDEX OFF, \n\t'
-             'b INT, \n\t'
-             'PRIMARY KEY (pk)\n)\n\n'), ())
-
-    def test_column_geopoint_without_index(self):
-        class DummyTable(self.Base):
-            __tablename__ = 't'
-            pk = sa.Column(sa.String, primary_key=True)
-            a = sa.Column(Geopoint, crate_index=False)
-        with self.assertRaises(sa.exc.CompileError):
-            self.Base.metadata.create_all(bind=self.engine)
-
-    def test_text_column_without_columnstore(self):
-        class DummyTable(self.Base):
-            __tablename__ = 't'
-            pk = sa.Column(sa.String, primary_key=True)
-            a = sa.Column(sa.String, crate_columnstore=False)
-            b = sa.Column(sa.String, crate_columnstore=True)
-            c = sa.Column(sa.String)
-
-        self.Base.metadata.create_all(bind=self.engine)
-
-        fake_cursor.execute.assert_called_with(
-            ('\nCREATE TABLE t (\n\t'
-             'pk STRING NOT NULL, \n\t'
-             'a STRING STORAGE WITH (columnstore = false), \n\t'
-             'b STRING, \n\t'
-             'c STRING, \n\t'
-             'PRIMARY KEY (pk)\n)\n\n'), ())
-
-    def test_non_text_column_without_columnstore(self):
-        class DummyTable(self.Base):
-            __tablename__ = 't'
-            pk = sa.Column(sa.String, primary_key=True)
-            a = sa.Column(sa.Integer, crate_columnstore=False)
-
-        with self.assertRaises(sa.exc.CompileError):
-            self.Base.metadata.create_all(bind=self.engine)
-
-    def test_column_server_default_text_func(self):
-        class DummyTable(self.Base):
-            __tablename__ = 't'
-            pk = sa.Column(sa.String, primary_key=True)
-            a = sa.Column(sa.DateTime, server_default=sa.text("now()"))
-
-        self.Base.metadata.create_all(bind=self.engine)
-        fake_cursor.execute.assert_called_with(
-            ('\nCREATE TABLE t (\n\t'
-             'pk STRING NOT NULL, \n\t'
-             'a TIMESTAMP DEFAULT now(), \n\t'
-             'PRIMARY KEY (pk)\n)\n\n'), ())
-
-    def test_column_server_default_string(self):
-        class DummyTable(self.Base):
-            __tablename__ = 't'
-            pk = sa.Column(sa.String, primary_key=True)
-            a = sa.Column(sa.String, server_default="Zaphod")
-
-        self.Base.metadata.create_all(bind=self.engine)
-        fake_cursor.execute.assert_called_with(
-            ('\nCREATE TABLE t (\n\t'
-             'pk STRING NOT NULL, \n\t'
-             'a STRING DEFAULT \'Zaphod\', \n\t'
-             'PRIMARY KEY (pk)\n)\n\n'), ())
-
-    def test_column_server_default_func(self):
-        class DummyTable(self.Base):
-            __tablename__ = 't'
-            pk = sa.Column(sa.String, primary_key=True)
-            a = sa.Column(sa.DateTime, server_default=sa.func.now())
-
-        self.Base.metadata.create_all(bind=self.engine)
-        fake_cursor.execute.assert_called_with(
-            ('\nCREATE TABLE t (\n\t'
-             'pk STRING NOT NULL, \n\t'
-             'a TIMESTAMP DEFAULT now(), \n\t'
-             'PRIMARY KEY (pk)\n)\n\n'), ())
-
-    def test_column_server_default_text_constant(self):
-        class DummyTable(self.Base):
-            __tablename__ = 't'
-            pk = sa.Column(sa.String, primary_key=True)
-            answer = sa.Column(sa.Integer, server_default=sa.text("42"))
-
-        self.Base.metadata.create_all(bind=self.engine)
-        fake_cursor.execute.assert_called_with(
-            ('\nCREATE TABLE t (\n\t'
-             'pk STRING NOT NULL, \n\t'
-             'answer INT DEFAULT 42, \n\t'
-             'PRIMARY KEY (pk)\n)\n\n'), ())

crate/client/sqlalchemy/tests/datetime_test.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.
-
-from __future__ import absolute_import
-from datetime import datetime, tzinfo, timedelta
-from unittest import TestCase
-from unittest.mock import patch, MagicMock
-
-import sqlalchemy as sa
-from sqlalchemy.exc import DBAPIError
-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')
-FakeCursor = MagicMock(name='FakeCursor', spec=Cursor)
-FakeCursor.return_value = fake_cursor
-
-
-class CST(tzinfo):
-    """
-    Timezone object for CST
-    """
-
-    def utcoffset(self, date_time):
-        return timedelta(seconds=-3600)
-
-    def dst(self, date_time):
-        return timedelta(seconds=-7200)
-
-
-@patch('crate.client.connection.Cursor', FakeCursor)
-class SqlAlchemyDateAndDateTimeTest(TestCase):
-
-    def setUp(self):
-        self.engine = sa.create_engine('crate://')
-        Base = declarative_base()
-
-        class Character(Base):
-            __tablename__ = 'characters'
-            name = sa.Column(sa.String, primary_key=True)
-            date = sa.Column(sa.Date)
-            timestamp = sa.Column(sa.DateTime)
-
-        fake_cursor.description = (
-            ('characters_name', None, None, None, None, None, None),
-            ('characters_date', None, None, None, None, None, None)
-        )
-        self.session = Session(bind=self.engine)
-        self.Character = Character
-
-    def test_date_can_handle_datetime(self):
-        """ date type should also be able to handle iso datetime strings.
-
-        this verifies that the fallback in the Date result_processor works.
-        """
-        fake_cursor.fetchall.return_value = [
-            ('Trillian', '2013-07-16T00:00:00.000Z')
-        ]
-        self.session.query(self.Character).first()
-
-    def test_date_cannot_handle_tz_aware_datetime(self):
-        character = self.Character()
-        character.name = "Athur"
-        character.timestamp = datetime(2009, 5, 13, 19, 19, 30, tzinfo=CST())
-        self.session.add(character)
-        self.assertRaises(DBAPIError, self.session.commit)

crate/client/sqlalchemy/tests/dialect_test.py

@@ -1,156 +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, skipIf
-from unittest.mock import MagicMock, patch
-
-import sqlalchemy as sa
-
-from crate.client.cursor import Cursor
-from crate.client.sqlalchemy import SA_VERSION
-from crate.client.sqlalchemy.sa_version import SA_1_4, SA_2_0
-from crate.client.sqlalchemy.types import ObjectType
-from sqlalchemy import inspect
-from sqlalchemy.orm import Session
-try:
-    from sqlalchemy.orm import declarative_base
-except ImportError:
-    from sqlalchemy.ext.declarative import declarative_base
-from sqlalchemy.testing import eq_, in_, is_true
-
-FakeCursor = MagicMock(name='FakeCursor', spec=Cursor)
-
-
-@patch('crate.client.connection.Cursor', FakeCursor)
-class SqlAlchemyDialectTest(TestCase):
-
-    def execute_wrapper(self, query, *args, **kwargs):
-        self.executed_statement = query
-        return self.fake_cursor
-
-    def setUp(self):
-
-        self.fake_cursor = MagicMock(name='fake_cursor')
-        FakeCursor.return_value = self.fake_cursor
-
-        self.engine = sa.create_engine('crate://')
-
-        self.executed_statement = None
-
-        self.connection = self.engine.connect()
-
-        self.fake_cursor.execute = self.execute_wrapper
-
-        self.base = declarative_base()
-
-        class Character(self.base):
-            __tablename__ = 'characters'
-
-            name = sa.Column(sa.String, primary_key=True)
-            age = sa.Column(sa.Integer, primary_key=True)
-            obj = sa.Column(ObjectType)
-            ts = sa.Column(sa.DateTime, onupdate=datetime.utcnow)
-
-        self.session = Session(bind=self.engine)
-
-    def init_mock(self, return_value=None):
-        self.fake_cursor.rowcount = 1
-        self.fake_cursor.description = (
-            ('foo', None, None, None, None, None, None),
-        )
-        self.fake_cursor.fetchall = MagicMock(return_value=return_value)
-
-    def test_primary_keys_2_3_0(self):
-        insp = inspect(self.session.bind)
-        self.engine.dialect.server_version_info = (2, 3, 0)
-
-        self.fake_cursor.rowcount = 3
-        self.fake_cursor.description = (
-            ('foo', None, None, None, None, None, None),
-        )
-        self.fake_cursor.fetchall = MagicMock(return_value=[["id"], ["id2"], ["id3"]])
-
-        eq_(insp.get_pk_constraint("characters")['constrained_columns'], {"id", "id2", "id3"})
-        self.fake_cursor.fetchall.assert_called_once_with()
-        in_("information_schema.key_column_usage", self.executed_statement)
-        in_("table_catalog = ?", self.executed_statement)
-
-    def test_primary_keys_3_0_0(self):
-        insp = inspect(self.session.bind)
-        self.engine.dialect.server_version_info = (3, 0, 0)
-
-        self.fake_cursor.rowcount = 3
-        self.fake_cursor.description = (
-            ('foo', None, None, None, None, None, None),
-        )
-        self.fake_cursor.fetchall = MagicMock(return_value=[["id"], ["id2"], ["id3"]])
-
-        eq_(insp.get_pk_constraint("characters")['constrained_columns'], {"id", "id2", "id3"})
-        self.fake_cursor.fetchall.assert_called_once_with()
-        in_("information_schema.key_column_usage", self.executed_statement)
-        in_("table_schema = ?", self.executed_statement)
-
-    def test_get_table_names(self):
-        self.fake_cursor.rowcount = 1
-        self.fake_cursor.description = (
-            ('foo', None, None, None, None, None, None),
-        )
-        self.fake_cursor.fetchall = MagicMock(return_value=[["t1"], ["t2"]])
-
-        insp = inspect(self.session.bind)
-        self.engine.dialect.server_version_info = (2, 0, 0)
-        eq_(insp.get_table_names(schema="doc"),
-            ['t1', 't2'])
-        in_("WHERE table_schema = ? AND table_type = 'BASE TABLE' ORDER BY", self.executed_statement)
-
-    def test_get_view_names(self):
-        self.fake_cursor.rowcount = 1
-        self.fake_cursor.description = (
-            ('foo', None, None, None, None, None, None),
-        )
-        self.fake_cursor.fetchall = MagicMock(return_value=[["v1"], ["v2"]])
-
-        insp = inspect(self.session.bind)
-        self.engine.dialect.server_version_info = (2, 0, 0)
-        eq_(insp.get_view_names(schema="doc"),
-            ['v1', 'v2'])
-        eq_(self.executed_statement, "SELECT table_name FROM information_schema.views "
-                                     "ORDER BY table_name ASC, table_schema ASC")
-
-    @skipIf(SA_VERSION < SA_1_4, "Inspector.has_table only available on SQLAlchemy>=1.4")
-    def test_has_table(self):
-        self.init_mock(return_value=[["foo"], ["bar"]])
-        insp = inspect(self.session.bind)
-        is_true(insp.has_table("bar"))
-        eq_(self.executed_statement,
-            "SELECT table_name FROM information_schema.tables "
-            "WHERE table_schema = ? AND table_type = 'BASE TABLE' "
-            "ORDER BY table_name ASC, table_schema ASC")
-
-    @skipIf(SA_VERSION < SA_2_0, "Inspector.has_schema only available on SQLAlchemy>=2.0")
-    def test_has_schema(self):
-        self.init_mock(
-            return_value=[["blob"], ["doc"], ["information_schema"], ["pg_catalog"], ["sys"]])
-        insp = inspect(self.session.bind)
-        is_true(insp.has_schema("doc"))
-        eq_(self.executed_statement,
-            "select schema_name from information_schema.schemata order by schema_name asc")