def test_com_bucket_missing(self):
record = copy.deepcopy(base_record)
record['ext']['all_data_disks_encrypted'] = False
record['com'] = None
plugin = azvmdatadiskencryptionevent.AzVMDataDiskEncryptionEvent()
events = list(plugin.eval(record))
self.assertEqual(events, [])
def test_cloud_type_non_azure(self):
record = copy.deepcopy(base_record)
record['ext']['all_data_disks_encrypted'] = False
record['com']['cloud_type'] = 'non_azure'
plugin = azvmdatadiskencryptionevent.AzVMDataDiskEncryptionEvent()
events = list(plugin.eval(record))
self.assertEqual(events, [])
def test_single_encrypted_data_disk(self):
record = copy.deepcopy(base_record)
record['raw'] = copy.deepcopy(base_raw_bucket)
record['ext']['all_data_disks_encrypted'] = False
encrypted_data_disk = copy.deepcopy(base_unencrypted_data_disk)
encrypted_data_disk['encryption_settings'][0]['enabled'] = True
record['raw']['instance_view']['disks'].append(encrypted_data_disk)
plugin = azvmdatadiskencryptionevent.AzVMDataDiskEncryptionEvent()
events = list(plugin.eval(record))
self.assertEqual(len(events), 0)
def test_single_unencrypted_data_disk(self):
record = copy.deepcopy(base_record)
record['raw'] = copy.deepcopy(base_raw_bucket)
record['ext']['all_data_disks_encrypted'] = False
record['raw']['instance_view']['disks'].append(
copy.deepcopy(base_unencrypted_data_disk))
plugin = azvmdatadiskencryptionevent.AzVMDataDiskEncryptionEvent()
events = list(plugin.eval(record))
self.assertEqual(len(events), 1)
self.assertEqual(events[0]['ext']['record_type'],
'vm_data_disk_encryption_event')
self.assertEqual(events[0]['com']['record_type'],
'vm_data_disk_encryption_event')
def test_multiple_unencrypted_data_disk(self):
record = copy.deepcopy(base_record)
record['raw'] = copy.deepcopy(base_raw_bucket)
record['ext']['all_data_disks_encrypted'] = False
unencrypted_data_disks = random.randint(1, 5)
for x in range(unencrypted_data_disks):
unencrypted_data_disk = copy.deepcopy(base_unencrypted_data_disk)
unencrypted_data_disk['name'] = 'data_disk' + str(x)
record['raw']['instance_view']['disks'].append(
unencrypted_data_disk)
plugin = azvmdatadiskencryptionevent.AzVMDataDiskEncryptionEvent()
events = list(plugin.eval(record))
self.assertEqual(len(events), unencrypted_data_disks)
def test_all_data_disks_encrypted(self):
record = copy.deepcopy(base_record)
plugin = azvmdatadiskencryptionevent.AzVMDataDiskEncryptionEvent()
events = list(plugin.eval(record))
self.assertEqual(events, [])
def test_record_type_non_vm_instance_view(self):
record = copy.deepcopy(base_record)
record['ext']['record_type'] = 'non_vm_instance_view'
plugin = azvmdatadiskencryptionevent.AzVMDataDiskEncryptionEvent()
events = list(plugin.eval(record))
self.assertEqual(events, [])
def test_record_type_non_compute(self):
record = copy.deepcopy(base_record)
record['com']['record_type'] = 'non_compute'
plugin = azvmdatadiskencryptionevent.AzVMDataDiskEncryptionEvent()
events = list(plugin.eval(record))
self.assertEqual(events, [])
