Remove single value use case

Author: Bill Prin

bigquery/tests/test_async_query.py

@@ -16,7 +16,7 @@
 import unittest
 
 from bigquery.samples.async_query import run, main
-from tests import CloudBaseTest, mock_raw_input_list, BUCKET_NAME_ENV, \
+from tests import CloudBaseTest, mock_raw_input, BUCKET_NAME_ENV, \
     PROJECT_ID_ENV
 
 class TestAsyncQuery(CloudBaseTest):
@@ -37,7 +37,7 @@ def test_async_query_runner(self):
         test_project_id = os.environ.get(PROJECT_ID_ENV)
         answers = [test_bucket_name, test_project_id, 'n',
                    '1', '1']
-        with mock_raw_input_list(answers):
+        with mock_raw_input(answers):
             main()
 
 

tests/__init__.py

@@ -27,7 +27,7 @@
     os.path.abspath(os.path.dirname(__file__)), 'resources')
 
 
-class mock_raw_input_list(object):
+class mock_raw_input(object):
 
     def __init__(self, list_):
         self.i = 0
@@ -46,14 +46,6 @@ def __exit__(self, exc_type, exc_value, traceback):
         __builtin__.raw_input = self.raw_input_cache
 
 
-@contextmanager
-def mock_raw_input(mock):
-    original_raw_input = __builtin__.raw_input
-    __builtin__.raw_input = mock
-    yield
-    __builtin__.raw_input = original_raw_input
-
-
 class CloudBaseTest(unittest.TestCase):
 
     def setUp(self):