def test_whitelisted_extensions(self): record = copy.deepcopy(base_record) record['ext']['extensions'] = ['ext01', 'ext02'] plugin = azvmextensionevent.AzVMExtensionEvent(whitelisted=['w_ext01']) events = list(plugin.eval(record)) self.assertEqual(len(events), 1) self.assertEqual(events[0]['ext']['record_type'], 'vm_unapproved_extension_event') self.assertEqual(events[0]['com']['cloud_type'], 'azure') self.assertEqual(events[0]['com']['record_type'], 'vm_unapproved_extension_event') self.assertTrue('reference' in events[0]['com']) self.assertIsNotNone(events[0]['com']['description']) self.assertIsNotNone(events[0]['com']['recommendation'])
def test_no_blacklisted_extensions(self): record = copy.deepcopy(base_record) record['ext']['extensions'] = ['ext01', 'ext02'] plugin = azvmextensionevent.AzVMExtensionEvent(blacklisted=['ext03']) events = list(plugin.eval(record)) self.assertEqual(len(events), 0)
def test_ext_bucket_missing(self): record = copy.deepcopy(base_record) record['ext'] = None plugin = azvmextensionevent.AzVMExtensionEvent() 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 = azvmextensionevent.AzVMExtensionEvent() events = list(plugin.eval(record)) self.assertEqual(events, [])
def test_cloud_type_non_azure(self): record = copy.deepcopy(base_record) record['com']['cloud_type'] = 'non_azure' plugin = azvmextensionevent.AzVMExtensionEvent() events = list(plugin.eval(record)) self.assertEqual(events, [])