view tests/mocktime.py @ 49157:5d610521a285 stable

test-dirstate: fix detection of Rust environment variable

The Rust path never actually worked. This change also improves clarity of the
comment. The next change will ensure we print something when this check fails.

Differential Revision: https://phab.mercurial-scm.org/D12601

from __future__ import absolute_import

import os
import time


class mocktime(object):
    def __init__(self, increment):
        self.time = 0
        self.increment = [float(s) for s in increment.split()]
        self.pos = 0

    def __call__(self):
        self.time += self.increment[self.pos % len(self.increment)]
        self.pos += 1
        return self.time


def uisetup(ui):
    time.time = mocktime(os.environ.get('MOCKTIME', '0.1'))