Coverage for chalicelib/core/log_tools/newrelic.py: 91%

1from chalicelib.core.log_tools import log_tools 

2from schemas import schemas 

3 

4IN_TY = "newrelic" 

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 "region" in changes: 17 ↛ 19line 17 didn't jump to line 19 because the condition on line 17 was always true

18 options["region"] = changes["region"] 

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

20 options["applicationId"] = changes["applicationId"] 

21 if "xQueryKey" in changes: 21 ↛ 24line 21 didn't jump to line 24 because the condition on line 21 was always true

22 options["xQueryKey"] = changes["xQueryKey"] 

23 

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

25 

26 

27def add(tenant_id, project_id, application_id, x_query_key, region): 

28 # region=False => US; region=True => EU 

29 options = {"applicationId": application_id, "xQueryKey": x_query_key, "region": region} 

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

31 

32 

33def delete(tenant_id, project_id): 

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

35 

36 

37def add_edit(tenant_id, project_id, data: schemas.IntegrationNewrelicSchema): 

38 s = get(project_id) 

39 if s is not None: 

40 return update(tenant_id=tenant_id, project_id=project_id, 

41 changes={"applicationId": data.application_id, 

42 "xQueryKey": data.x_query_key, 

43 "region": data.region}) 

44 else: 

45 return add(tenant_id=tenant_id, 

46 project_id=project_id, 

47 application_id=data.application_id, 

48 x_query_key=data.x_query_key, 

49 region=data.region)