release_test: Revert

Signed-off-by: Ce Gao <[email protected]>
kubeflow · May 5, 2018 · 7de9d3f · 7de9d3f
1 parent aba69c1
commit 7de9d3f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/py/release_test.py b/py/release_test.py
@@ -14,7 +14,8 @@ class ReleaseTest(unittest.TestCase):
   @mock.patch("py.release.util.clone_repo")
   @mock.patch("py.release.build_and_push")
   def test_build_postsubmit(  # pylint: disable=no-self-use
-      self, mock_build_and_push, mock_clone):
+      self, mock_build_and_push, mock_clone, _mock_install, _mock_os,
+      _mock_makedirs):
     # Make sure REPO_OWNER and REPO_NAME aren't changed by the environment
     release.REPO_ORG = "kubeflow"
     release.REPO_NAME = "tf-operator"
@@ -37,7 +38,10 @@ def test_build_postsubmit(  # pylint: disable=no-self-use
   def test_build_pr(# pylint: disable=no-self-use
       self,
       mock_build_and_push,
-      mock_clone):
+      mock_clone,
+      _mock_install,
+      _mock_os,
+      _mock_makedirs):
     parser = release.build_parser()
     args = parser.parse_args(
       ["pr", "--pr=10", "--commit=22", "--src_dir=/top/src_dir"])