def test__get_data(self, build_granule_mock, TaxyTool_mock,
RecordDictionaryTool_mock):
config = {
'constraints': {
'count': 6
},
'max_records': 4,
'data_producer_id': sentinel.dprod_id,
'taxonomy': sentinel.taxy_tool
}
TaxyTool_mock.load.return_value = sentinel.ttool_ret_val
RecordDictionaryTool_mock.side_effect = lambda **kwargs: {}
for x in FibonacciDataHandler._get_data(config):
RecordDictionaryTool_mock.assert_called_with(
taxonomy=sentinel.ttool_ret_val)
build_granule_mock.assert_called_with(
taxonomy=sentinel.ttool_ret_val,
data_producer_id=sentinel.dprod_id,
record_dictionary=ANY)
cargs = build_granule_mock.call_args
self.assertIn('data', cargs[1]['record_dictionary'])
self.assertIsInstance(cargs[1]['record_dictionary']['data'],
numpy.ndarray)
self.assertEquals(build_granule_mock.call_count, 2)
self.assertEquals(RecordDictionaryTool_mock.call_count, 2)
TaxyTool_mock.load.assert_called_once_with(sentinel.taxy_tool)
def test__get_data(self, build_granule_mock, ParamDict_mock, RecordDictionaryTool_mock):
config={'constraints':{'count':6},'max_records':4,'data_producer_id':sentinel.dprod_id,'param_dictionary':sentinel.pdict}
ParamDict_mock.load.return_value = sentinel.pdict_return_value
RecordDictionaryTool_mock.side_effect = lambda **kwargs: {}
for x in FibonacciDataHandler._get_data(config):
RecordDictionaryTool_mock.assert_called_with(param_dictionary=sentinel.pdict_return_value)
build_granule_mock.assert_called_with(data_producer_id=sentinel.dprod_id, record_dictionary=ANY, param_dictionary=sentinel.pdict_return_value)
cargs=build_granule_mock.call_args
self.assertIn('data', cargs[1]['record_dictionary'])
self.assertIsInstance(cargs[1]['record_dictionary']['data'], numpy.ndarray)
self.assertEquals(build_granule_mock.call_count, 2)
self.assertEquals(RecordDictionaryTool_mock.call_count, 2)
ParamDict_mock.load.assert_called_once_with(sentinel.pdict)
def test__get_data(self, RecordDictionaryTool_mock):
config = {'constraints': {'count': 6},
'max_records': 4,
'data_producer_id': sentinel.dprod_id,
'param_dictionary': sentinel.pdict,
'stream_def': sentinel.stream_def_id
}
my_dict = {}
def setitem(name, val):
my_dict[name] = val
retval = MagicMock(spec=RecordDictionaryTool)
retval.__setitem__ = Mock(side_effect=setitem)
retval.to_granule.return_value = MagicMock(spec=Granule)
RecordDictionaryTool_mock.return_value = retval
for x in FibonacciDataHandler._get_data(config):
self.assertTrue(isinstance(x, Granule))
retval.to_granule.assert_any_call()
def test__get_data(self, RecordDictionaryTool_mock):
config = {'constraints': {'count': 6},
'max_records': 4,
'data_producer_id': sentinel.dprod_id,
'param_dictionary': sentinel.pdict,
'stream_def': sentinel.stream_def_id
}
my_dict = {}
def setitem(name, val):
my_dict[name] = val
retval = MagicMock(spec=RecordDictionaryTool)
retval.__setitem__ = Mock(side_effect=setitem)
retval.to_granule.return_value = MagicMock(spec=Granule)
RecordDictionaryTool_mock.return_value = retval
for x in FibonacciDataHandler._get_data(config):
self.assertTrue(isinstance(x, Granule))
retval.to_granule.assert_any_call()
def test__get_data(self, build_granule_mock, ParamDict_mock, RecordDictionaryTool_mock):
config = {
"constraints": {"count": 6},
"max_records": 4,
"data_producer_id": sentinel.dprod_id,
"param_dictionary": sentinel.pdict,
}
ParamDict_mock.load.return_value = sentinel.pdict_return_value
RecordDictionaryTool_mock.side_effect = lambda **kwargs: {}
for x in FibonacciDataHandler._get_data(config):
RecordDictionaryTool_mock.assert_called_with(param_dictionary=sentinel.pdict_return_value)
build_granule_mock.assert_called_with(
data_producer_id=sentinel.dprod_id, record_dictionary=ANY, param_dictionary=sentinel.pdict_return_value
)
cargs = build_granule_mock.call_args
self.assertIn("data", cargs[1]["record_dictionary"])
self.assertIsInstance(cargs[1]["record_dictionary"]["data"], numpy.ndarray)
self.assertEquals(build_granule_mock.call_count, 2)
self.assertEquals(RecordDictionaryTool_mock.call_count, 2)
ParamDict_mock.load.assert_called_once_with(sentinel.pdict)
def test__constraints_for_new_request(self):
ret = FibonacciDataHandler._constraints_for_new_request({})
self.assertTrue("count" in ret)
i = ret["count"]
self.assertIsInstance(i, int)
self.assertTrue(5 <= i <= 20)
def test__init_acquisition_cycle(self, _init_acquisition_cycle_mock):
config = {}
FibonacciDataHandler._init_acquisition_cycle(config)
_init_acquisition_cycle_mock.assert_called_once_with(config)
def test__constraints_for_new_request(self):
ret = FibonacciDataHandler._constraints_for_new_request({})
self.assertTrue('count' in ret)
i = ret['count']
self.assertIsInstance(i, int)
self.assertTrue(5 <= i <= 20)
def test__init_acquisition_cycle(self, _init_acquisition_cycle_mock):
config = {}
FibonacciDataHandler._init_acquisition_cycle(config)
_init_acquisition_cycle_mock.assert_called_once_with(config)
def test__new_data_constraints(self):
ret = FibonacciDataHandler._new_data_constraints({})
self.assertTrue('count' in ret)
i=ret['count']
self.assertIsInstance(i, int)
self.assertTrue(5 <= i <= 20)
