def test_record_exists_with_keyspace_from_session(self):
hook = CassandraHook("cassandra_default_with_schema")
session = hook.get_conn()
cqls = [
"DROP TABLE IF EXISTS t",
"CREATE TABLE t (pk1 text, pk2 text, c text, PRIMARY KEY (pk1, pk2))",
"INSERT INTO t (pk1, pk2, c) VALUES ('foo', 'bar', 'baz')",
]
for cql in cqls:
session.execute(cql)
self.assertTrue(hook.record_exists("t", {"pk1": "foo", "pk2": "bar"}))
self.assertFalse(hook.record_exists("t", {"pk1": "foo", "pk2": "baz"}))
session.shutdown()
hook.shutdown_cluster()
def poke(self, context):
self.log.info('Sensor check existence of record: %s', self.keys)
hook = CassandraHook(self.cassandra_conn_id)
return hook.record_exists(self.table, self.keys)
