Ejemplos de get_cassandra_db_alias en Python

Ejemplo n.º 1

    def test_flush(self):

        db_alias = get_cassandra_db_alias()
        call_command('flush',
                     database=db_alias,
                     noinput=True,
                     interactive=False)

Ejemplo n.º 2

Archivo: test_commands.py Proyecto: isaquealves/django-cassandra-engine

    def test_sync_cassandra_creates_keyspace_and_tables(
            self, sync_table_mock, create_keyspace_mock):

        alias = get_cassandra_db_alias()
        call_command('sync_cassandra', database=alias)
        replication_factor = 1

        all_models = list(chain.from_iterable(
            self.connection.introspection.cql_models.values()))

        create_keyspace_mock.assert_called_once_with(self.keyspace,
                                                     replication_factor)

        for model in all_models:
            sync_table_mock.assert_has_call(call(model))

Ejemplo n.º 3

    def test_sync_cassandra_creates_keyspace_and_tables(self, mock_management):

        alias = get_cassandra_db_alias()
        call_command('sync_cassandra', database=alias)
        replication_factor = 1

        all_models = list(
            chain.from_iterable(
                self.connection.introspection.cql_models.values()))

        mock_management.create_keyspace_simple.assert_called_once_with(
            self.keyspace, replication_factor)

        for model in all_models:
            mock_management.sync_table.assert_has_call(call(model))

Ejemplo n.º 4

Archivo: test_commands.py Proyecto: r4fek/django-cassandra-engine

    def test_sync_cassandra_creates_keyspace_and_tables(self, mock_management):

        alias = get_cassandra_db_alias()
        call_command('sync_cassandra', database=alias)
        replication_factor = 1

        all_models = list(chain.from_iterable(
            self.connection.introspection.cql_models.values()))

        mock_management.create_keyspace_simple.assert_called_once_with(
            self.keyspace,
            replication_factor,
            connections=[alias]
        )

        for model in all_models:
            mock_management.sync_table.assert_has_call(call(model))

Ejemplo n.º 5

Archivo: test_commands.py Proyecto: maria-grigorieva/django_cassandra_engine

    def test_sync_cassandra_creates_keyspace_and_tables(
            self, sync_table_mock, create_keyspace_mock):

        alias = get_cassandra_db_alias()
        call_command('sync_cassandra', database=alias)
        options = self.connection.settings_dict.get('OPTIONS', {})
        replication_opts = options.get('replication', {})
        strategy_class = 'SimpleStrategy'
        replication_factor = 1

        all_models = list(chain.from_iterable(
            self.connection.introspection.cql_models.values()))

        create_keyspace_mock.assert_called_once_with(
            self.keyspace, strategy_class, replication_factor,
            **replication_opts)

        for model in all_models:
            sync_table_mock.assert_has_call(call(model))

Ejemplo n.º 6

Archivo: test_commands.py Proyecto: xprilion/django-cassandra-engine

    def test_sync_cassandra_creates_keyspace_and_tables(self, mock_management):

        alias = get_cassandra_db_alias()
        call_command('sync_cassandra', database=alias)
        replication_factor = 1

        all_models = list(
            chain.from_iterable(
                self.connection.introspection.cql_models.values()))

        mock_management.create_keyspace_simple.assert_called_once_with(
            self.keyspace, replication_factor, connections=[alias])
        mock_management.Model = (Model, DjangoCassandraModel)

        calls = []
        for model in all_models:
            calls.append(
                call(model, keyspaces=[self.keyspace], connections=[alias]))

        mock_management.sync_table.assert_has_calls(calls, any_order=True)

Ejemplo n.º 7

Archivo: test_commands.py Proyecto: r4fek/django-cassandra-engine

    def test_flush(self):

        db_alias = get_cassandra_db_alias()
        call_command('flush', database=db_alias, interactive=False)