def test_write_fieldserver_post_data_with_offset(self, round_time): post_handler = PostHandler() post_handler.date_time_str = "2014-09-06 16:45:00" round_time.return_value = pytz.utc.localize(datetime.datetime(2014, 9, 5)) data = {"cc14153300_Offsets_0010-0019": "44.000000 ,44.5 ,45 ,45.5 ,46 ,46.5 ,47 ,47.5 ,48 ,48.5"} destination_stream = io.StringIO() post_handler.write_fieldserver_post_data(data, destination_stream) destination_stream.seek(0) written_data = destination_stream.read().strip().split("\n") assert len(written_data) == 10 assert json.loads(written_data[0]) == {"site_id": "cc14153300", "offset": 10, "value": "44.000000", "timestamp": "2014-09-05 00:00:00", "date_added": post_handler.date_time_str} assert json.loads(written_data[1]) == {"site_id": "cc14153300", "offset": 11, "value": "44.5", "timestamp": "2014-09-05 00:00:00", "date_added": post_handler.date_time_str}
def test_handle_fieldserver_post(self, open_mock, path_join): data_mock = MagicMock() post_handler = PostHandler() post_handler.write_fieldserver_post_data = Mock() post_handler.fieldserver_raw_folder = Mock() post_handler.handle_fieldserver_post(data_mock) path_join.assert_called_with(post_handler.fieldserver_raw_folder, post_handler.date_str) open_mock.assert_called_with(path_join.return_value, "a") post_handler.write_fieldserver_post_data.assert_called_with(data_mock, open_mock.return_value.__enter__.return_value)