Skip to content

Commit

Permalink
Forgot to close/create.
Browse files Browse the repository at this point in the history
  • Loading branch information
btimby committed May 11, 2017
1 parent cdc897f commit 266e0e5
Showing 1 changed file with 21 additions and 0 deletions.
21 changes: 21 additions & 0 deletions tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -280,8 +280,17 @@ def test_wait_timeout(self):


class ShortcutTestCase(unittest.TestCase):
"""
Test module-level shortcut functions.
"""

def setUp(self):
self.queue = tpq.Queue('test')
self.queue.create()

def tearDown(self):
self.queue.clear()
self.queue.close()

def test_get(self):
item_put = {'test': 'test'}
Expand All @@ -305,10 +314,20 @@ def test_clear(self):


class CommandTestCase(unittest.TestCase):
"""
Test Command Line Interface.
"""

def setUp(self):
self.queue = tpq.Queue('test')
self.queue.create()

def tearDown(self):
self.queue.clear()
self.queue.close()

def test_main_get(self):
"""Ensure we can get from a queue using CLI."""
stdout = StringIO()
item_put = {'test': 'test'}
self.queue.put(item_put)
Expand All @@ -327,6 +346,7 @@ def test_main_get(self):
self.assertEqual(item_put, json.loads(stdout.getvalue()))

def test_main_put(self):
"""Ensure we can put to a queue using CLI."""
item_put = {'test': 'test'}
main({
'--debug': False,
Expand All @@ -339,5 +359,6 @@ def test_main_put(self):
with self.queue.get() as item_get:
self.assertEqual(item_put, item_get)


if __name__ == '__main__':
unittest.main()

0 comments on commit 266e0e5

Please sign in to comment.