import time
import pytest
import progressbar2
import original_examples

# Import hack to allow for parallel Tox
try:
    import examples
except ImportError:
    import sys
    sys.path.append('..')
    import examples
    sys.path.remove('..')


def test_examples(monkeypatch):
    for example in examples.examples:
        try:
            example()
        except ValueError:
            pass


@pytest.mark.filterwarnings('ignore:.*maxval.*:DeprecationWarning')
@pytest.mark.parametrize('example', original_examples.examples)
def test_original_examples(example, monkeypatch):
    monkeypatch.setattr(progressbar2.ProgressBar,
                        '_MINIMUM_UPDATE_INTERVAL', 1)
    monkeypatch.setattr(time, 'sleep', lambda t: None)
    example()


@pytest.mark.parametrize('example', examples.examples)
def test_examples_nullbar(monkeypatch, example):
    # Patch progressbar to use null bar instead of regular progress bar
    monkeypatch.setattr(progressbar2, 'ProgressBar', progressbar2.NullBar)
    assert progressbar2.ProgressBar._MINIMUM_UPDATE_INTERVAL < 0.0001
    example()


def test_reuse():
    import progressbar2

    bar = progressbar2.ProgressBar()
    bar.start()
    for i in range(10):
        bar.update(i)
    bar.finish()

    bar.start(init=True)
    for i in range(10):
        bar.update(i)
    bar.finish()

    bar.start(init=False)
    for i in range(10):
        bar.update(i)
    bar.finish()


def test_dirty():
    import progressbar2

    bar = progressbar2.ProgressBar()
    bar.start()
    for i in range(10):
        bar.update(i)
    bar.finish(dirty=True)