def test_poke_non_existing(self):
partitions = [
"{}.{}/{}={}".format(self.database, self.table, self.partition_by,
self.next_day)
]
sensor = NamedHivePartitionSensor(partition_names=partitions,
task_id='test_poke_non_existing',
poke_interval=1,
hook=self.hook,
dag=self.dag)
self.assertFalse(sensor.poke(None))
def test_poke_non_existing(self):
partitions = ["{}.{}/{}={}".format(self.database,
self.table,
self.partition_by,
self.next_day)]
sensor = NamedHivePartitionSensor(partition_names=partitions,
task_id='test_poke_non_existing',
poke_interval=1,
hook=self.hook,
dag=self.dag)
self.assertFalse(sensor.poke(None))
def test_poke_non_existing(self):
self.hook.metastore.__enter__(
).check_for_named_partition.return_value = False
partitions = [
"{}.{}/{}={}".format(self.database, self.table, self.partition_by,
self.next_day)
]
sensor = NamedHivePartitionSensor(partition_names=partitions,
task_id='test_poke_non_existing',
poke_interval=1,
hook=self.hook,
dag=self.dag)
self.assertFalse(sensor.poke(None))
self.hook.metastore.__enter__(
).check_for_named_partition.assert_called_with(
self.database, self.table, "{}={}".format(self.partition_by,
self.next_day))