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())