Coverage for tests/unit/test_main.py: 100%

47 statements  

« prev     ^ index     » next       coverage.py v7.7.1, created at 2025-03-26 18:37 -0600

1"""Unit tests for main.py""" 

2 

3# pylint: disable=C0116 

4from unittest.mock import patch 

5import pytest 

6from es_fieldusage.exceptions import ValueMismatch 

7 

8 

9def test_init(field_usage_instance): 

10 assert field_usage_instance.usage_stats 

11 assert field_usage_instance.indices == "index1" 

12 

13 

14def test_get(field_usage_instance): 

15 assert "index1" in field_usage_instance.usage_stats 

16 assert field_usage_instance.usage_stats["index1"]["field1"] == 10 

17 

18 

19def test_get_field_mappings(field_usage_instance): 

20 mappings = field_usage_instance.get_field_mappings("index1") 

21 assert "field1" in mappings 

22 

23 

24def test_populate_values(field_usage_instance): 

25 data = {} 

26 result = field_usage_instance.populate_values("index1", data) 

27 assert result["field1"] == 10 

28 

29 

30def test_get_resultset(field_usage_instance): 

31 result = field_usage_instance.get_resultset("index1") 

32 assert result["field1"] == 10 

33 

34 

35def test_merge_results(field_usage_instance): 

36 result = field_usage_instance.merge_results("index1") 

37 assert result["field1"] == 10 

38 

39 

40def test_per_index_report(field_usage_instance): 

41 report = field_usage_instance.per_index_report 

42 assert "index1" in report 

43 assert "accessed" in report["index1"] 

44 assert "field1" in report["index1"]["accessed"] 

45 

46 

47def test_report(field_usage_instance): 

48 report = field_usage_instance.report 

49 assert "accessed" in report 

50 assert "field1" in report["accessed"] 

51 

52 

53def test_result(field_usage_instance): 

54 result = field_usage_instance.result() 

55 assert "field1" in result 

56 

57 

58def test_results_by_index(field_usage_instance): 

59 results = field_usage_instance.results_by_index 

60 assert "index1" in results 

61 assert "field1" in results["index1"] 

62 

63 

64def test_results(field_usage_instance): 

65 results = field_usage_instance.results 

66 assert "field1" in results 

67 

68 

69def test_indices(field_usage_instance): 

70 assert field_usage_instance.indices == "index1" 

71 

72 

73def test_sum_index_stats(field_usage_instance): 

74 field_usage = { 

75 "index1": {"shards": [{"stats": {"fields": {"field1": {"any": 5}}}}]} 

76 } 

77 result = field_usage_instance.sum_index_stats(field_usage, "index1") 

78 assert result["field1"] == 5