def test_get_json(self):
"""
:return: JSON object of data object
"""
from businesslogic.data import CollectionData
expected_data = "Test data"
expected_collector = "TestCollector"
expected_collector_params = {"param1": "value1", "param2": "value2"}
expected_sourcepath = "source_path"
expected_sourcehash = "This is not a hash"
actual_data_obj = CollectionData(data=expected_data,
collector_name=expected_collector)
actual_data_obj.collector_parameters = expected_collector_params
actual_data_obj._sourcepath = expected_sourcepath
actual_data_obj._sourcehash = expected_sourcehash
expected_time = datetime.now()
actual_data_obj.currentdatetime = expected_time
actual_json_string = actual_data_obj.get_json()
actual_json = json.loads(actual_json_string)
self.assertIsNotNone(actual_json)
self.assertIsNotNone(actual_json['CollectionTime'])
self.assertEqual(actual_json['CollectionTime'], str(expected_time))
self.assertEqual(actual_json['Data'], expected_data)
self.assertEqual(actual_json['CollectorName'], expected_collector)
self.assertEqual(actual_json['SourcePath'], expected_sourcepath)
self.assertEqual(actual_json['SourceHash'], expected_sourcehash)
self.assertEqual(actual_json['param1'],
expected_collector_params['param1'])
self.assertEqual(actual_json['param2'],
expected_collector_params['param2'])
def test_get_collector_info_as_str_no_collector_name(self):
"""
With no collector provided it should raise a ValueError
"""
from businesslogic.data import CollectionData
from businesslogic.errors import NoCollectorError
actual_parameters = {'Param1': 12, 'Param2': 'Test'}
actual_data = CollectionData(data="Test data")
actual_data.collector_parameters = actual_parameters
with self.assertRaises(NoCollectorError):
actual_data.get_collector_info_as_str()
def test_get_collector_info_as_str(self):
"""
:return: Should return a string with collector informormation.
"""
from businesslogic.data import CollectionData
actual_parameters = {'Param1': 12, 'Param2': 'Test'}
actual_data = CollectionData(data="Test data",
collector_name="TestCollector")
actual_data.collector_parameters = actual_parameters
expected = "Collector: TestCollector\r\nParam1: '12'\r\nParam2: 'Test'"
actual = actual_data.get_collector_info_as_str()
self.assertEqual(actual, expected)