Method test_monitor_goodput

tests/goodput_utils_test.py:71–76

(self, mock_start_goodput_uploader, mock_stop_goodput_uploader)

  @mock.patch("ml_goodput_measurement.monitoring.GoodputMonitor.stop_goodput_uploader")
  @mock.patch("ml_goodput_measurement.monitoring.GoodputMonitor.start_goodput_uploader")
  def test_monitor_goodput(self, mock_start_goodput_uploader, mock_stop_goodput_uploader):
    mock_start_goodput_uploader.return_value = mock.MagicMock()

    with maybe_monitor_goodput(self.config):
      mock_start_goodput_uploader.assert_called()
    mock_stop_goodput_uploader.assert_called()


if __name__ == "__main__":