def install_on_docker_compose(self):
path = ComposeConfigManager.get_config_file_path()
path = '/'.join(path.split('/')[:-1])
# Fetch docker-compose
Transport().download(
url=
'https://github.com/polyaxon/polyaxon-compose/archive/master.tar.gz',
filename=path + '/file',
untar=True,
delete_tar=True,
extract_path=path)
# Move necessary info
shutil.copy(path + '/polyaxon-compose-master/docker-compose.yml',
path + '/docker-compose.yml')
shutil.copy(path + '/polyaxon-compose-master/components.env',
path + '/components.env')
shutil.copy(path + '/polyaxon-compose-master/base.env',
path + '/base.env')
shutil.rmtree(path + '/polyaxon-compose-master/')
# Generate env from config
ComposeConfigManager.set_config(self.compose.generate_env(self.config))
Printer.print_success('Docker Compose deployment is initialised.')
if self.dry_run:
Printer.print_success('Polyaxon generated deployment env.')
return
self.docker.execute(['volume', 'create', '--name=polyaxon-postgres'])
Printer.print_success('Docker volume created.')
self.compose.execute(['-f', path + '/docker-compose.yml', 'up', '-d'])
Printer.print_success('Deployment is running in the background.')
Printer.print_success(
'You can configure your CLI by running: '
'polyaxon config set --host=localhost --port=8000.')
def setUp(self):
self.host = 'localhost'
self.http_port = 8000
self.ws_port = 1337
self.api_config = ApiConfig(host=self.host,
http_port=self.http_port,
ws_port=self.ws_port,
version='v1',
token='token',
reraise=True,
use_https=False,
is_managed=False,
interval=0,
timeout=0.01,
schema_response=True)
self.transport = Transport(config=self.api_config)
settings.TMP_AUTH_TOKEN_PATH = '{}/{}'.format(tempfile.mkdtemp(),
'.plx')
settings.USER_CLIENT_CONFIG_PATH = '{}/{}'.format(
tempfile.mkdtemp(), '.plx')
class TestHttpTransport(BaseTestCaseTransport):
# pylint:disable=protected-access
def setUp(self):
super(TestHttpTransport, self).setUp()
self.transport = Transport()
def test_create_progress_callback(self):
encoder = mock.MagicMock()
encoder.configure_mock(len=10)
_, progress_bar = self.transport.create_progress_callback(encoder)
assert isinstance(progress_bar, Bar)
def test_format_sizeof(self):
assert self.transport.format_sizeof(10) == '10.0B'
assert self.transport.format_sizeof(10000) == '9.8KiB'
assert self.transport.format_sizeof(100000) == '97.7KiB'
assert self.transport.format_sizeof(10000000) == '9.5MiB'
assert self.transport.format_sizeof(10000000000) == '9.3GiB'
def test_session(self):
assert hasattr(self.transport, '_session') is False
assert isinstance(self.transport.session, requests.Session)
assert isinstance(self.transport._session, requests.Session)
def transport(self):
if not self._transport:
self._transport = Transport(config=self.api_config)
return self._transport
def setUp(self):
super(TestHttpTransport, self).setUp()
self.transport = Transport()