From ec6f2ed9765a5b5377bd7911dab7b20d61b052d7 Mon Sep 17 00:00:00 2001
From: Florian Einfalt <florian.einfalt@me.com>
Date: Sat, 18 Feb 2017 14:35:13 +0000
Subject: [PATCH] Update tests for progress bar context

---
 nukecontexts/__init__.py |  2 +-
 tests/test_contexts.py   | 25 +++++++++++++++++--------
 2 files changed, 18 insertions(+), 9 deletions(-)

diff --git a/nukecontexts/__init__.py b/nukecontexts/__init__.py
index 78edaac..ebed89d 100644
--- a/nukecontexts/__init__.py
+++ b/nukecontexts/__init__.py
@@ -8,7 +8,7 @@
 TESTING = False
 try:
     TESTING = os.environ['NON_PRODUCTION_CONTEXT']
-    logger = None
+    logger = logging.getLogger()
     sentry = None
 except:
     if platform.system() == 'Darwin':
diff --git a/tests/test_contexts.py b/tests/test_contexts.py
index 71eba5e..e49d022 100644
--- a/tests/test_contexts.py
+++ b/tests/test_contexts.py
@@ -5,7 +5,7 @@
 def test_enabled(node):
     node['disable'].setValue(True)
     assert node['disable'].value() == True
-    with ctx.enabled(node, verbose=True):
+    with ctx.enabled(node):
         assert node['disable'].value() == False
     assert node['disable'].value() == True
     node['disable'].setValue(False)
@@ -13,7 +13,7 @@ def test_enabled(node):
 
 def test_disabled(node):
     assert node['disable'].value() == False
-    with ctx.disabled(node, verbose=True):
+    with ctx.disabled(node):
         assert node['disable'].value() == True
     assert node['disable'].value() == False
 
@@ -21,15 +21,15 @@ def test_disabled(node):
 def test_set_attr_errors(node):
     # Test invalid node error
     with pytest.raises(ctx.NukeContextError):
-        with ctx.set_attr(None, 'invalid_attr', True, verbose=True):
+        with ctx.set_attr(None, 'invalid_attr', True):
             print 'should not print'
     # Test invalid attribute error
     with pytest.raises(ctx.NukeContextError):
-        with ctx.set_attr(node, 'invalid_attr', True, verbose=True):
+        with ctx.set_attr(node, 'invalid_attr', True):
             print 'should not print'
     # Test invalid value error
     with pytest.raises(ctx.NukeContextError):
-        with ctx.set_attr(node, 'file_type', 2.0, verbose=True):
+        with ctx.set_attr(node, 'file_type', 2.0):
             print 'should not print'
 
 
@@ -41,17 +41,26 @@ def test_multiple(node):
     node['disable'].setValue(True)
     node['file_type'].setValue('exr')
 
-    assert node['disable'].value() == True
+    assert node['disable'].value()
     assert node['file_type'].value() == 'exr'
     assert node['channels'].value() == 'rgb'
 
     with ctx.multiple_contexts([ctx1, ctx2, ctx3]):
-        assert node['disable'].value() == False
+        assert not node['disable'].value()
         assert node['file_type'].value() == 'jpeg'
         assert node['channels'].value() == 'rgba'
 
-    assert node['disable'].value() == True
+    assert node['disable'].value()
     assert node['file_type'].value() == 'exr'
     assert node['channels'].value() == 'rgb'
 
     node['disable'].setValue(False)
+
+
+def test_progress(node):
+    ctx1 = ctx.enabled(node)
+    iterable = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
+
+    with ctx1, ctx.Progress(iterable) as progress:
+        for idx, item in enumerate(progress):
+            assert item == idx + 1