Newer
Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
from unittest import mock
import django.test as dtest
from lofardata.models import DataProduct
from lofardata.view_helpers.dataproductinfo import get_distinct_dataproduct_field
test_data_products = [dict(obs_id='12345', oid_source='SAS', dataproduct_source='lofar',
dataproduct_type='paper',
project='LT10_10',
location='Dwingeloo',
activity='paintball',
surl='srm://lta-head.lofar.psnc.pl:4884/subfolder/a_some_nice.tar',
filesize=40,
antenna_set='HBA Dual Inner',
instrument_filter='110-190 MHz',
dysco_compression=True),
dict(obs_id='123789', oid_source='SAS', dataproduct_source='lofar',
dataproduct_type='paper',
project='LT10_10',
location='Dwingeloo',
activity='gokarts',
surl='srm://lta-head.lofar.psnc.pl:4884/subfolder/some_very_nice.tar',
filesize=40,
antenna_set='HBA Dual Inner',
instrument_filter='110-190 MHz',
dysco_compression=True),
dict(obs_id='123789', oid_source='SAS', dataproduct_source='lofar',
dataproduct_type='paper',
project='LT10_10_Classified',
location='Groningen',
activity='gokarts',
surl='srm://lta-head.lofar.psnc.pl:4884/subfolder/some_not_nice.tar',
filesize=40,
antenna_set='HBA Dual Inner',
instrument_filter='110-190 MHz',
dysco_compression=False),
dict(obs_id='1337', oid_source='SAS', dataproduct_source='lofar',
dataproduct_type='plastic',
project='LT10_10_Classified',
location='Groningen',
activity='paintball',
surl='srm://lta-head.lofar.psnc.pl:4884/subfolder/some_not_very_nice.tar',
filesize=40,
antenna_set='HBA Dual Inner',
instrument_filter='110-190 MHz',
dysco_compression=False),
dict(obs_id='3773', oid_source='SAS', dataproduct_source='lofar',
dataproduct_type='plastic',
project='LT10_10_Classified',
location='Dwingeloo',
activity='gokarts',
surl='srm://lta-head.lofar.psnc.pl:4884/subfolder/some_not_very_super_nice.tar',
filesize=40,
antenna_set=None,
instrument_filter='110-190 MHz',
dysco_compression=False)]
class TestDataProductFieldCache(dtest.TestCase):
def setUp(self) -> None:
for test_dataproduct in test_data_products:
DataProduct.objects.create(**test_dataproduct)
@mock.patch("lofardata.view_helpers.dataproductinfo.cache.get")
@mock.patch("lofardata.view_helpers.dataproductinfo.cache.set")
def test_return_distinct_values(self, cache_set_mock, cache_get_mock):
cache_get_mock.return_value = None
actual = get_distinct_dataproduct_field('activity')
expected = ['paintball', 'gokarts']
self.assertCountEqual(actual, expected, "Actual list of distinct dataproduct types does not contain the same elements as the expected mocked list (ignoring order)")
@mock.patch("lofardata.view_helpers.dataproductinfo.cache.get")
@mock.patch("lofardata.view_helpers.dataproductinfo.cache.set")
def test_get_result_from_cache_when_cached(self, cache_set_mock, cache_get_mock):
mocked_return_value = ["Zwolle", "Steenwijk"]
cache_get_mock.return_value = mocked_return_value
actual = get_distinct_dataproduct_field('location')
self.assertCountEqual(actual, mocked_return_value, "Actual list of distinct dataproduct types does not contain the same elements as the expected mocked list (ignoring order)")
@mock.patch("lofardata.view_helpers.dataproductinfo.cache.get")
@mock.patch("lofardata.view_helpers.dataproductinfo.cache.set")
def test_get_result_from_cache_with_correct_key(self, cache_set_mock, cache_get_mock):
get_distinct_dataproduct_field('some_custom_field')
cache_get_mock.assert_called_with("distinct_dataproduct_some_custom_field")
@mock.patch("lofardata.view_helpers.dataproductinfo.cache.get")
@mock.patch("lofardata.view_helpers.dataproductinfo.cache.set")
Robbie Luijben
committed
def test_put_in_cache_with_correct_key_and_data_and_expiration_when_not_cached(self, cache_set_mock, cache_get_mock):
cache_get_mock.return_value = None
get_distinct_dataproduct_field('location')
Robbie Luijben
committed
cache_set_mock.assert_called_with("distinct_dataproduct_location", ["Dwingeloo", "Groningen"], None)