def test_additional_data(self):
second_field = 'windDir'
units = 1
first_data = {
self.wind_field: random.uniform(1, 100),
'usUnits': units,
'dateTime': int(time.time() + 0.5)
}
second_data = {
second_field: random.uniform(1, 100),
'usUnits': units,
'dateTime': int(time.time() + 0.5)
}
total_data = dict(second_data)
total_data['windSpeed'] = first_data['windSpeed']
del total_data['usUnits']
del total_data['dateTime']
with mock.patch('user.MQTTSubscribe.weewx.units.to_std_system'
) as mock_to_std_system:
SUT = CollectData(self.wind_fields, units)
mock_to_std_system.return_value = dict(first_data)
SUT.add_data(self.wind_field, first_data)
mock_to_std_system.return_value = dict(second_data)
added_data = SUT.add_data(second_field, second_data)
self.assertDictEqual(added_data, {})
self.assertDictEqual(SUT.data, total_data)
def test_get_data(self):
SUT = CollectData(self.wind_fields)
SUT.add_data(self.data)
collected_data = SUT.get_data()
self.assertDictEqual(collected_data, self.data)
def test_get_data(self):
with mock.patch('user.MQTTSubscribe.weewx.units.to_std_system'
) as mock_to_std_system:
SUT = CollectData(self.wind_fields, self.units)
mock_to_std_system.return_value = dict(self.data)
SUT.add_data(self.wind_field, self.data)
collected_data = SUT.get_data()
self.assertDictEqual(collected_data, self.data)
def test_data_in_fields(self):
data = {
'windSpeed': random.uniform(1, 100),
'usUnits': 1,
'dateTime': int(time.time() + 0.5)
}
SUT = CollectData(self.wind_fields)
added_data = SUT.add_data(data)
self.assertDictEqual(added_data, {})
self.assertDictEqual(SUT.data, data)
def test_data_in_fields(self):
units = 1
data = {
self.wind_field: random.uniform(1, 100),
'dateTime': int(time.time() + 0.5)
}
with mock.patch('user.MQTTSubscribe.weewx.units.to_std_system'
) as mock_to_std_system:
SUT = CollectData(self.wind_fields, units)
mock_to_std_system.return_value = data
added_data = SUT.add_data(self.wind_field, data)
self.assertDictEqual(added_data, {})
self.assertDictEqual(SUT.data, data)
def test_second_data(self):
first_data = {
'windSpeed': random.uniform(1, 100),
'usUnits': 1,
'dateTime': int(time.time() + 0.5)
}
second_data = {
'windSpeed': random.uniform(1, 100),
'usUnits': 1,
'dateTime': int(time.time() + 0.5)
}
SUT = CollectData(self.wind_fields)
SUT.add_data(first_data)
added_data = SUT.add_data(second_data)
self.assertDictEqual(added_data, first_data)
self.assertDictEqual(SUT.data, second_data)
def test_second_data(self):
units = 1
first_data = {
self.wind_field: random.uniform(1, 100),
'usUnits': units,
'dateTime': int(time.time() + 0.5)
}
second_data = {
self.wind_field: random.uniform(1, 100),
'usUnits': units,
'dateTime': int(time.time() + 0.5)
}
with mock.patch('user.MQTTSubscribe.weewx.units.to_std_system'
) as mock_to_std_system:
SUT = CollectData(self.wind_fields, units)
mock_to_std_system.return_value = dict(first_data)
SUT.add_data(self.wind_field, first_data)
mock_to_std_system.return_value = dict(second_data)
added_data = SUT.add_data(self.wind_field, second_data)
self.assertDictEqual(added_data, first_data)
self.assertDictEqual(
SUT.data, {self.wind_field: second_data[self.wind_field]})