def test_get_first_element_resource(self):
client_mock = Mock()
client_mock.activity_logs.list.return_value = self.events
manager = Mock()
manager.type = 'vm'
manager.get_client.return_value = client_mock
resource = tools.get_resource({})
base = AutoTagDate(data={'tag': 'test'}, manager=manager)
base._prepare_processing()
result = base._get_first_event(resource)
client_mock.activity_logs.list.assert_called_once()
self.assertEqual(result, self.events[-1])
def test_get_first_element_resource_group(self):
events = copy.copy(self.events)
for e in events:
e.operation_name.value = 'Microsoft.Resources/subscriptions/resourcegroups/write'
client_mock = Mock()
client_mock.activity_logs.list.return_value = events
manager = Mock()
manager.type = 'resourcegroup'
manager.get_client.return_value = client_mock
resource_group = tools.get_resource_group_resource({})
base = AutoTagDate(data={'tag': 'test'}, manager=manager)
base._prepare_processing()
result = base._get_first_event(resource_group)
client_mock.activity_logs.list.assert_called_once()
self.assertEqual(result, self.events[-1])
def _get_action(self, data):
return AutoTagDate(data=data, manager=Mock())
