def test_make_spec(self): t_config = TConfig() t_config.DATA_DIR = os.getcwd() pyu = PyUpdater(t_config) pyu.setup() spec_cmd = [ 'make-spec', 'app.py', '-F', '--app-name', 'MyApp', '--app-version', '0.1.0' ] spec_file_name = get_system() + '.spec' build_cmd = [ 'build', '--app-name', 'MyApp', '--app-version', '0.1.0', spec_file_name ] build_cmd = [str(b) for b in build_cmd] parser = get_parser() with open('app.py', 'w') as f: f.write('print "Hello, World!"') args, pyu_args = parser.parse_known_args(spec_cmd) b = Builder(args, pyu_args) b.make_spec() assert os.path.exists(spec_file_name) args, pyu_args = parser.parse_known_args(build_cmd) b = Builder(args, pyu_args) b.build() with ChDir(new_folder): assert len(os.listdir(os.getcwd())) == 1
def test_bad_pub_key(self): t_config = TConfig() # If changed ensure it's a valid key format t_config.PUBLIC_KEY = '25RSdhJ+xCsxxTjY5jffilatipp29tnKp/D5BelSMJM' t_config.DATA_DIR = os.getcwd() client = Client(t_config, refresh=True, test=True) assert client.update_check(client.app_name, '0.0.0') is None
def test_original_init(): config = TConfig() updater = PyiUpdater(config) client = Client(updater, test=True) assert client.app_name == u'jms' assert client.update_urls[0] == (u'https://s3-us-west-1.amazon' 'aws.com/pyi-test/')
def test_set_uploader_bad_settings(): config = TConfig() config.ACCESS_KEY_ID = None config.SECRET_ACCESS_KEY = u'Not an actual secret' config.BUCKET_NAME = u'Bucket Name' s_nst = PyiUpdater(config) uploader = Uploader(s_nst) uploader.set_uploader(u's3')
def test_init(self, db): data_dir = os.getcwd() t_config = TConfig() t_config.DATA_DIR = data_dir config = TransistionDict() config.from_object(t_config) p = PackageHandler(config, db) assert p.files_dir == os.path.join(data_dir, s_dir, 'files') assert p.deploy_dir == os.path.join(data_dir, s_dir, 'deploy')
def test_process_packages(self): data_dir = os.getcwd() t_config = TConfig() t_config.DATA_DIR = data_dir t_config.UPDATE_PATCHES = False config = Config() config.from_object(t_config) p = PackageHandler(config) p.process_packages()
def test_init(self): data_dir = os.getcwd() t_config = TConfig() t_config.DATA_DIR = data_dir config = Config() config.from_object(t_config) p = PackageHandler(config) assert p.files_dir == os.path.join(data_dir, user_data_dir, "files") assert p.deploy_dir == os.path.join(data_dir, user_data_dir, "deploy")
def test_http(self): t_config = TConfig() t_config.DATA_DIR = os.getcwd() client = Client(t_config, refresh=True, test=True) update = client.update_check(client.app_name, '0.0.1') assert update is not None assert update.app_name == 'Acme' assert update.download() is True assert update.is_downloaded() is True
def test_no_patch_support(self, db): data_dir = os.getcwd() t_config = TConfig() t_config.DATA_DIR = data_dir t_config.UPDATE_PATCHES = False config = TransistionDict() config.from_object(t_config) p = PackageHandler(config, db) p.process_packages()
def setup_func(): global test_data_dir config = TConfig() updater = PyiUpdater(config) ph = PackageHandler(updater) kh = KeyHandler(updater) kh.test = True ph.setup() kh.make_keys()
def test_manifest_filesystem(self): t_config = TConfig() t_config.DATA_DIR = os.getcwd() client = Client(t_config, refresh=True, test=True) filesystem_data = client._get_manifest_from_disk() assert filesystem_data is not None filesystem_data = filesystem_data.decode() filesystem_data = json.loads(filesystem_data) del filesystem_data["signature"] assert client.json_data == filesystem_data
def test_directory_creation(self): data_dir = os.getcwd() pyu_data_dir = os.path.join(data_dir, 'pyu-data') t_config = TConfig() t_config.DATA_DIR = data_dir pyu = PyUpdater(t_config) pyu.setup() assert os.path.exists(pyu_data_dir) assert os.path.exists(os.path.join(pyu_data_dir, 'deploy')) assert os.path.exists(os.path.join(pyu_data_dir, 'files')) assert os.path.exists(os.path.join(pyu_data_dir, 'new'))
def test_callback(self): def cb(status): print(status) def cb2(status): raise IndexError t_config = TConfig() t_config.PUBLIC_KEY = '25RSdhJ+xCsxxTjY5jffilatipp29tnKp/D5BelSMJM' t_config.DATA_DIR = os.getcwd() client = Client(t_config, refresh=True, test=True, progress_hooks=[cb]) client.add_progress_hook(cb2) assert client.update_check(client.app_name, '0.0.0') is None
def test_setup(): global test_data_dir config = TConfig() updater = PyiUpdater(config) ph = PackageHandler(updater) key_dir = os.path.join(ph.data_dir, u'keys') kh = KeyHandler(updater) kh.test = True assert os.path.exists(os.path.abspath(key_dir)) assert kh.private_key_name == u'jms.pem' assert kh.public_key_name == u'jms.pub'
def test_background_http(self): t_config = TConfig() t_config.DATA_DIR = os.getcwd() client = Client(t_config, refresh=True, test=True) update = client.update_check(client.app_name, '0.0.1') assert update is not None assert update.app_name == 'Acme' update.download(background=True) count = 0 while count < 61: if update.is_downloaded() is True: break time.sleep(1) count += 1 assert update.is_downloaded() is True
def test_build_fail_no_spec_or_src(self): with pytest.raises(SystemExit): t_config = TConfig() t_config.DATA_DIR = os.getcwd() pyu = PyUpdater(t_config) pyu.setup() build_cmd = [ 'build', '--app-name', 'MyApp', '--clean' '--app-version', '0.1.0', '-F' ] parser = get_parser() args, pyu_args = parser.parse_known_args(build_cmd) b = Builder(args, pyu_args) b.build()
def test_multiple_background_calls(self, client): t_config = TConfig() t_config.DATA_DIR = os.getcwd() t_config.VERIFY_SERVER_CERT = False client = Client(t_config, refresh=True, test=True) update = client.update_check(client.app_name, '0.0.1') assert update is not None assert update.app_name == 'Acme' update.download(background=True) count = 0 assert update.download(background=True) is None assert update.download() is None while count < 61: if update.is_downloaded() is True: break time.sleep(1) count += 1 assert update.is_downloaded() is True
def test_build(self): t_config = TConfig() t_config.DATA_DIR = os.getcwd() pyu = PyUpdater(t_config) pyu.setup() build_cmd = [ 'build', '--app-name', 'MyApp', '--app-version', '0.1.0', 'app.py' ] build_cmd = [str(b) for b in build_cmd] parser = get_parser() with open('app.py', 'w') as f: f.write('print "Hello, World!"') args, pyu_args = parser.parse_known_args(build_cmd) b = Builder(args, pyu_args) b.build() with ChDir(new_folder): assert len(os.listdir(os.getcwd())) == 1
def test_execution(): global test_data_dir config = TConfig() updater = PyiUpdater(config) ph = PackageHandler(updater) kh = KeyHandler(updater) ph.setup() kh.test = True kh.make_keys() # Make zipfile with ChDir(test_data_dir): os.mkdir(u'test-app') with ChDir(u'test-app'): with open(u'app.txt', u'w') as f: f.write(u'I am so happy' * 1000) shutil.make_archive(u'Test App-mac-0.2.0', u'zip', u'test-app') shutil.move(u'Test App-mac-0.2.0.zip', u'new') ph.process_packages() kh.sign_update()
def test_build_fail_script_syntax_error(self): with pytest.raises(SystemExit): t_config = TConfig() t_config.DATA_DIR = os.getcwd() pyu = PyUpdater(t_config) pyu.setup() spec_cmd = ['make-spec', 'app.py', '-F'] spec_file_name = get_system() + '.spec' build_cmd = [ 'build', '--app-name', 'MyApp', '--clean' '--app-version', '0.1.0', spec_file_name ] parser = get_parser() with open('app.py', 'w') as f: # Missing closing quote f.write('print "Hello, World!') args, pyu_args = parser.parse_known_args(spec_cmd) b = Builder(args, pyu_args) b.make_spec() assert os.path.exists(spec_file_name) args, pyu_args = parser.parse_known_args(build_cmd) b = Builder(args, pyu_args)
import os import shutil import sys from nose import with_setup sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) from pyi_updater import PyiUpdater from pyi_updater.client import Client from tconfig import TConfig client = Client(TConfig(), test=True) def tear_down(): if os.path.exists(client.data_dir): shutil.rmtree(client.data_dir, ignore_errors=True) def test_data_dir(): client = Client(TConfig(), test=True) assert os.path.exists(client.data_dir) is True def test_original_init(): config = TConfig() updater = PyiUpdater(config) client = Client(updater, test=True) assert client.app_name == u'jms' assert client.update_urls[0] == (u'https://s3-us-west-1.amazon'
def test_download(): client = Client(TConfig(), test=True) assert client.app_name == u'jms' assert client.update_check(client.app_name, '0.0.1') is True assert client.download() is True assert client.install() is True
def test_check_version(): config = TConfig() client = Client(config, test=True) assert client.update_check(client.app_name, '0.0.2') is True assert client.update_check(client.app_name, '6.0.0') is False
def test_data_dir(): client = Client(TConfig(), test=True) assert os.path.exists(client.data_dir) is True
def pyu(): t_config = TConfig() t_config.DATA_DIR = os.getcwd() pyu = PyUpdater(t_config) return pyu
def test_url_str_attr(self): t_config = TConfig() t_config.DATA_DIR = os.getcwd() t_config.UPDATE_URLS = 'http://acme.com/update' client = Client(t_config, test=True) assert isinstance(client.update_urls, list)
def client(): t_config = TConfig() t_config.DATA_DIR = os.getcwd() client = Client(t_config, refresh=True, test=True) client.FROZEN = True return client
def test_new_init(): config = TConfig() client = Client(config, test=True) assert client.app_name == u'jms' assert client.update_urls[0] == (u'https://s3-us-west-1.amazon' 'aws.com/pyi-test/')
# distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. # -------------------------------------------------------------------------- from __future__ import unicode_literals import pytest import requests from pyupdater.uploader import BaseUploader, Uploader from pyupdater.utils.exceptions import UploaderError, UploaderPluginError from tconfig import TConfig my_config = TConfig() @pytest.mark.usefixtures('cleandir') class TestUtils(object): def test_plugin_baseclass_not_implemented(self): with pytest.raises(NotImplementedError): b = BaseUploader() b.init() with pytest.raises(NotImplementedError): b = BaseUploader() b.connect() with pytest.raises(NotImplementedError):
def test_bad_pub_key(): config = TConfig() config.PUBLIC_KEY = 'bad key' client = Client(config, test=True) assert client.update_check(u'jms', '0.0.0') is False