def test_poke(self, mock_hook):
mock_instance = mock_hook.return_value
sensor = WasbPrefixSensor(
task_id='wasb_sensor', dag=self.dag, check_options={'timeout': 2}, **self._config
)
sensor.poke(None)
mock_instance.check_for_prefix.assert_called_once_with('container', 'prefix', timeout=2)
def test_init(self):
sensor = WasbPrefixSensor(task_id='wasb_sensor_1', dag=self.dag, **self._config)
self.assertEqual(sensor.container_name, self._config['container_name'])
self.assertEqual(sensor.prefix, self._config['prefix'])
self.assertEqual(sensor.wasb_conn_id, self._config['wasb_conn_id'])
self.assertEqual(sensor.check_options, {})
self.assertEqual(sensor.timeout, self._config['timeout'])
sensor = WasbPrefixSensor(
task_id='wasb_sensor_2', dag=self.dag, check_options={'timeout': 2}, **self._config
)
self.assertEqual(sensor.check_options, {'timeout': 2})
def test_init(self):
sensor = WasbPrefixSensor(task_id='wasb_sensor_1',
dag=self.dag,
**self._config)
assert sensor.container_name == self._config['container_name']
assert sensor.prefix == self._config['prefix']
assert sensor.wasb_conn_id == self._config['wasb_conn_id']
assert sensor.check_options == {}
assert sensor.timeout == self._config['timeout']
sensor = WasbPrefixSensor(task_id='wasb_sensor_2',
dag=self.dag,
check_options={'timeout': 2},
**self._config)
assert sensor.check_options == {'timeout': 2}
