def test_build_postsubmit(self, mock_build_and_push, mock_clone,
_mock_install, _mock_os, _mock_makedirs): # pylint: disable=no-self-use
parser = release.build_parser()
args = parser.parse_args(["postsubmit", "--src_dir=/top/src_dir"])
release.build_postsubmit(args)
mock_build_and_push.assert_called_once_with(
'/top/src_dir/go', '/top/src_dir/go/src/github.com/tensorflow/k8s',
mock.ANY)
mock_clone.assert_called_once_with('/top/src_dir/git_tensorflow_k8s',
'tensorflow', 'k8s', None, None)
def test_build_pr(self, mock_build_and_push, mock_clone, _mock_install, _mock_os, _mock_makedirs): # pylint: disable=no-self-use
parser = release.build_parser()
args = parser.parse_args(["pr", "--pr=10", "--commit=22",
"--src_dir=/top/src_dir"])
release.build_pr(args)
mock_build_and_push.assert_called_once_with(
'/top/src_dir/go', '/top/src_dir/go/src/github.com/tensorflow/k8s',
mock.ANY)
mock_clone.assert_called_once_with(
"/top/src_dir/git_tensorflow_k8s", "tensorflow", "k8s", "22",
["pull/10/head:pr"])
def test_build_postsubmit(self, mock_build_and_push, mock_clone, # pylint: disable=no-self-use
_mock_install, _mock_os, _mock_makedirs):
# Make sure REPO_OWNER and REPO_NAME aren't changed by the environment
release.REPO_ORG = "tensorflow"
release.REPO_NAME = "k8s"
parser = release.build_parser()
args = parser.parse_args(["postsubmit", "--src_dir=/top/src_dir"])
release.build_postsubmit(args)
mock_build_and_push.assert_called_once_with(
'/top/src_dir/go', '/top/src_dir/go/src/github.com/tensorflow/k8s',
mock.ANY)
mock_clone.assert_called_once_with(
'/top/src_dir/git_tensorflow_k8s', 'tensorflow', 'k8s', None, None)
