From 5bfd05d26c59dd98c67fa6d261b804d547262223 Mon Sep 17 00:00:00 2001
From: sneakers-the-rat <JLSaunders987@gmail.com>
Date: Thu, 26 Oct 2023 18:50:17 -0700
Subject: [PATCH 1/2] Fix https://github.com/dandi/dandi-cli/issues/1339

Add in missing -dt term so ETA counts down correctly :)
---
 dandi/download.py            |  2 +-
 dandi/tests/test_download.py | 43 +++++++++++++++++++++++++++++++++++-
 2 files changed, 43 insertions(+), 2 deletions(-)

diff --git a/dandi/download.py b/dandi/download.py
index aad1af22e..73635da9d 100644
--- a/dandi/download.py
+++ b/dandi/download.py
@@ -444,7 +444,7 @@ def agg_done(self, done_sizes: Iterator[int]) -> List[str]:
             and self.items_summary.size != 0
         ):
             dt = time.time() - self.items_summary.t0
-            more_time = dt / r if r != 1 else 0
+            more_time = (dt / r) - dt if r != 1 else 0
             more_time_str = humanize.naturaldelta(more_time)
             if not self.it.finished:
                 more_time_str += "<"
diff --git a/dandi/tests/test_download.py b/dandi/tests/test_download.py
index 46f2e491a..93a70fe32 100644
--- a/dandi/tests/test_download.py
+++ b/dandi/tests/test_download.py
@@ -5,6 +5,7 @@
 import re
 from shutil import rmtree
 from typing import Callable, List, Tuple
+import time
 
 import numpy as np
 import pytest
@@ -17,7 +18,7 @@
 from .test_helpers import assert_dirtrees_eq
 from ..consts import DRAFT, dandiset_metadata_file
 from ..dandiarchive import DandisetURL
-from ..download import Downloader, ProgressCombiner, download
+from ..download import Downloader, ProgressCombiner, download, PYOUTHelper
 from ..exceptions import NotFoundError
 from ..utils import list_paths
 
@@ -899,3 +900,43 @@ def test_download_empty_dandiset_and_nonexistent_multiasset(
         tmp_path / empty_dandiset.dandiset_id,
         tmp_path / empty_dandiset.dandiset_id / "dandiset.yaml",
     ]
+
+
+def test_pyouthelper_time_remaining_1339():
+    """
+    Ensure that time remaining goes down!
+    https://github.com/dandi/dandi-cli/issues/1339
+
+    Since time.now() is called inside the agg methods, we
+    simulate time elapsing by changing t0
+
+    Assume downloading files totaling 100 arbitrary units at rates to trigger
+    a few hours, then countdown to 0.
+    check that ETA goes down like we want it to
+    """
+
+    helper = PYOUTHelper()
+    helper.items_summary.size = 100
+
+    # first check hours
+    # 5 hours remaining = 50% done in 60*60*5
+    helper.items_summary.t0 = time.time() - (60 * 60 * 5)
+    done = helper.agg_done(iter([50]))
+    assert done[-1] == "ETA: 5 hours<"
+
+    # 5 minutes remaining = 50% done in 60 * 5
+    helper.items_summary.t0 = time.time() - (5 * 60)
+    done = helper.agg_done(iter([50]))
+    assert done[-1] == "ETA: 5 minutes<"
+
+    # countdown the last 10 seconds!
+    for i in range(11):
+        helper.items_summary.t0 = time.time() - (100 - (10 - i))
+        done = helper.agg_done(iter([100 - (10 - i)]))
+        if i == 9:
+            assert done[-1] == f"ETA: a second<"
+        elif i == 10:
+            # once done, dont print ETA
+            assert len(done) == 2
+        else:
+            assert done[-1] == f"ETA: {10-i} seconds<"

From 086f1030600a6d951a4327776546bb29777f012c Mon Sep 17 00:00:00 2001
From: sneakers-the-rat <JLSaunders987@gmail.com>
Date: Fri, 27 Oct 2023 01:14:43 -0700
Subject: [PATCH 2/2] oop sorry linting

---
 dandi/tests/test_download.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dandi/tests/test_download.py b/dandi/tests/test_download.py
index 93a70fe32..b4b8651d1 100644
--- a/dandi/tests/test_download.py
+++ b/dandi/tests/test_download.py
@@ -934,7 +934,7 @@ def test_pyouthelper_time_remaining_1339():
         helper.items_summary.t0 = time.time() - (100 - (10 - i))
         done = helper.agg_done(iter([100 - (10 - i)]))
         if i == 9:
-            assert done[-1] == f"ETA: a second<"
+            assert done[-1] == "ETA: a second<"
         elif i == 10:
             # once done, dont print ETA
             assert len(done) == 2