Coverage for chalicelib/core/log_tools/datadog.py: 93%

1from chalicelib.core.log_tools import log_tools 

2from schemas import schemas 

3 

4IN_TY = "datadog" 

5 

6 

7def get_all(tenant_id): 

8 return log_tools.get_all_by_tenant(tenant_id=tenant_id, integration=IN_TY) 

9 

10 

11def get(project_id): 

12 return log_tools.get(project_id=project_id, integration=IN_TY) 

13 

14 

15def update(tenant_id, project_id, changes): 

16 options = {} 

17 if "apiKey" in changes: 17 ↛ 19line 17 didn't jump to line 19 because the condition on line 17 was always true

18 options["apiKey"] = changes["apiKey"] 

19 if "applicationKey" in changes: 19 ↛ 22line 19 didn't jump to line 22 because the condition on line 19 was always true

20 options["applicationKey"] = changes["applicationKey"] 

21 

22 return log_tools.edit(project_id=project_id, integration=IN_TY, changes=options) 

23 

24 

25def add(tenant_id, project_id, api_key, application_key): 

26 options = {"apiKey": api_key, "applicationKey": application_key} 

27 return log_tools.add(project_id=project_id, integration=IN_TY, options=options) 

28 

29 

30def delete(tenant_id, project_id): 

31 return log_tools.delete(project_id=project_id, integration=IN_TY) 

32 

33 

34def add_edit(tenant_id, project_id, data: schemas.IntegrationDatadogSchema): 

35 s = get(project_id) 

36 if s is not None: 

37 return update(tenant_id=tenant_id, project_id=project_id, 

38 changes={"apiKey": data.api_key, 

39 "applicationKey": data.application_key}) 

40 else: 

41 return add(tenant_id=tenant_id, 

42 project_id=project_id, 

43 api_key=data.api_key, 

44 application_key=data.application_key)