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]})