93 lines
4.3 KiB
Python
93 lines
4.3 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
"""Tests for deploy_playstore.py."""
|
||
|
|
import io
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import tempfile
|
||
|
|
import unittest
|
||
|
|
from pathlib import Path
|
||
|
|
from unittest.mock import MagicMock, call, patch
|
||
|
|
|
||
|
|
sys.path.insert(0, str(Path(__file__).parent))
|
||
|
|
|
||
|
|
import deploy_playstore
|
||
|
|
|
||
|
|
|
||
|
|
class TestMainEnvChecks(unittest.TestCase):
|
||
|
|
def test_missing_env_exits(self):
|
||
|
|
with patch.dict(os.environ, {}, clear=True):
|
||
|
|
with self.assertRaises(SystemExit) as ctx:
|
||
|
|
deploy_playstore.main()
|
||
|
|
self.assertEqual(ctx.exception.code, 1)
|
||
|
|
|
||
|
|
def test_missing_aab_exits(self):
|
||
|
|
fake_config = '{"type": "service_account"}'
|
||
|
|
with patch.dict(os.environ, {"PLAY_STORE_CONFIG_JSON": fake_config}):
|
||
|
|
with patch("deploy_playstore.os.path.exists", return_value=False):
|
||
|
|
with self.assertRaises(SystemExit) as ctx:
|
||
|
|
deploy_playstore.main()
|
||
|
|
self.assertEqual(ctx.exception.code, 1)
|
||
|
|
|
||
|
|
|
||
|
|
class TestMainHappyPath(unittest.TestCase):
|
||
|
|
def _run_main(self, fake_config):
|
||
|
|
mock_service = MagicMock()
|
||
|
|
mock_edits = mock_service.edits.return_value
|
||
|
|
mock_edits.insert.return_value.execute.return_value = {"id": "edit-42"}
|
||
|
|
mock_edits.bundles.return_value.upload.return_value.execute.return_value = {
|
||
|
|
"versionCode": 7
|
||
|
|
}
|
||
|
|
mock_edits.tracks.return_value.update.return_value.execute.return_value = {}
|
||
|
|
mock_edits.commit.return_value.execute.return_value = {}
|
||
|
|
|
||
|
|
with patch.dict(os.environ, {"PLAY_STORE_CONFIG_JSON": fake_config}):
|
||
|
|
with patch("deploy_playstore.os.path.exists", return_value=True):
|
||
|
|
with patch("deploy_playstore.service_account.Credentials.from_service_account_info"):
|
||
|
|
with patch("deploy_playstore.google_auth_httplib2.AuthorizedHttp"):
|
||
|
|
with patch("deploy_playstore.build", return_value=mock_service):
|
||
|
|
with patch("deploy_playstore.MediaFileUpload"):
|
||
|
|
deploy_playstore.main()
|
||
|
|
|
||
|
|
return mock_edits
|
||
|
|
|
||
|
|
def test_insert_called_with_num_retries(self):
|
||
|
|
edits = self._run_main('{"type":"service_account"}')
|
||
|
|
edits.insert.return_value.execute.assert_called_once_with(num_retries=3)
|
||
|
|
|
||
|
|
def test_bundle_upload_called_with_num_retries(self):
|
||
|
|
edits = self._run_main('{"type":"service_account"}')
|
||
|
|
edits.bundles.return_value.upload.return_value.execute.assert_called_once_with(num_retries=3)
|
||
|
|
|
||
|
|
def test_tracks_update_called_with_num_retries(self):
|
||
|
|
edits = self._run_main('{"type":"service_account"}')
|
||
|
|
edits.tracks.return_value.update.return_value.execute.assert_called_once_with(num_retries=3)
|
||
|
|
|
||
|
|
def test_commit_called_with_num_retries(self):
|
||
|
|
edits = self._run_main('{"type":"service_account"}')
|
||
|
|
edits.commit.return_value.execute.assert_called_once_with(num_retries=3)
|
||
|
|
|
||
|
|
def test_authorized_http_uses_timeout(self):
|
||
|
|
fake_config = '{"type":"service_account"}'
|
||
|
|
with patch.dict(os.environ, {"PLAY_STORE_CONFIG_JSON": fake_config}):
|
||
|
|
with patch("deploy_playstore.os.path.exists", return_value=True):
|
||
|
|
with patch("deploy_playstore.service_account.Credentials.from_service_account_info"):
|
||
|
|
with patch("deploy_playstore.httplib2.Http") as mock_http_cls:
|
||
|
|
with patch("deploy_playstore.google_auth_httplib2.AuthorizedHttp") as mock_auth:
|
||
|
|
mock_service = MagicMock()
|
||
|
|
mock_edits = mock_service.edits.return_value
|
||
|
|
mock_edits.insert.return_value.execute.return_value = {"id": "e1"}
|
||
|
|
mock_edits.bundles.return_value.upload.return_value.execute.return_value = {
|
||
|
|
"versionCode": 1
|
||
|
|
}
|
||
|
|
mock_edits.tracks.return_value.update.return_value.execute.return_value = {}
|
||
|
|
mock_edits.commit.return_value.execute.return_value = {}
|
||
|
|
with patch("deploy_playstore.build", return_value=mock_service):
|
||
|
|
with patch("deploy_playstore.MediaFileUpload"):
|
||
|
|
deploy_playstore.main()
|
||
|
|
|
||
|
|
mock_http_cls.assert_called_once_with(timeout=deploy_playstore._TIMEOUT)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|