本文整理汇总了Python中cassandra.cluster.Cluster.connection_factory方法的典型用法代码示例。如果您正苦于以下问题:Python Cluster.connection_factory方法的具体用法?Python Cluster.connection_factory怎么用?Python Cluster.connection_factory使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类cassandra.cluster.Cluster
的用法示例。
在下文中一共展示了Cluster.connection_factory方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: test_prepare_on_ignored_hosts
# 需要导入模块: from cassandra.cluster import Cluster [as 别名]
# 或者: from cassandra.cluster.Cluster import connection_factory [as 别名]
def test_prepare_on_ignored_hosts(self):
cluster = Cluster(protocol_version=PROTOCOL_VERSION,
load_balancing_policy=self.ignore_node_3_policy)
session = cluster.connect()
cluster.reprepare_on_up, cluster.prepare_on_all_hosts = True, False
hosts = cluster.metadata.all_hosts()
session.execute("CREATE KEYSPACE clustertests "
"WITH replication = "
"{'class': 'SimpleStrategy', 'replication_factor': '1'}")
session.execute("CREATE TABLE clustertests.tab (a text, PRIMARY KEY (a))")
# assign to an unused variable so cluster._prepared_statements retains
# reference
_ = session.prepare("INSERT INTO clustertests.tab (a) VALUES ('a')") # noqa
cluster.connection_factory = Mock(wraps=cluster.connection_factory)
unignored_address = '127.0.0.1'
unignored_host = next(h for h in hosts if h.address == unignored_address)
ignored_host = next(h for h in hosts if h.address in self.ignored_addresses)
unignored_host.is_up = ignored_host.is_up = False
cluster.on_up(unignored_host)
cluster.on_up(ignored_host)
# the length of mock_calls will vary, but all should use the unignored
# address
for c in cluster.connection_factory.mock_calls:
self.assertEqual(call(unignored_address), c)
cluster.shutdown()