1 files changed, 8 insertions, 4 deletions
diff --git a/src/mongo/db/s/sharding_data_transform_metrics_test_fixture.h b/src/mongo/db/s/sharding_data_transform_metrics_test_fixture.h
index a898e9fca73..cc66423e231 100644
--- a/src/mongo/db/s/sharding_data_transform_metrics_test_fixture.h
+++ b/src/mongo/db/s/sharding_data_transform_metrics_test_fixture.h
@@ -63,11 +63,11 @@ public:
         return _uuid;
     }
 
-    virtual int64_t getHighEstimateRemainingTimeMillis() const override {
+    virtual boost::optional<Milliseconds> getHighEstimateRemainingTimeMillis() const override {
         return _timeRemainingHigh;
     }
 
-    virtual int64_t getLowEstimateRemainingTimeMillis() const override {
+    virtual boost::optional<Milliseconds> getLowEstimateRemainingTimeMillis() const override {
         return _timeRemainingLow;
     }
 
@@ -82,8 +82,8 @@ public:
 private:
     UUID _uuid;
     Date_t _startTime;
-    int64_t _timeRemainingHigh;
-    int64_t _timeRemainingLow;
+    Milliseconds _timeRemainingHigh;
+    Milliseconds _timeRemainingLow;
     ShardingDataTransformMetrics::Role _role;
 };
 
@@ -186,6 +186,10 @@ protected:
         return &clock.value();
     }
 
+    ShardingDataTransformCumulativeMetrics* getCumulativeMetrics() {
+        return _cumulativeMetrics.get();
+    }
+
     using SpecialIndexBehaviorMap = stdx::unordered_map<int, std::function<void()>>;
     const SpecialIndexBehaviorMap kNoSpecialBehavior{};
     SpecialIndexBehaviorMap registerAtIndex(int index, const ObserverMock* mock) {