Unverified Commit e0208145 authored by liuzhe-lz's avatar liuzhe-lz Committed by GitHub
Browse files

Fix bugs found in integration test (#3214)

parent a1016a6c
from _common import build_wheel
build_wheel()
from datetime import datetime
from _common import set_variable
time = datetime.now().strftime('%Y%m%d%H%M%S')
set_variable('NNI_RELEASE', '999.' + time)
import sys
from _common import build_wheel, run_command
if len(sys.argv) <= 2:
extra_dep = ''
else:
extra_dep = f'[{sys.argv[2]}]'
wheel = build_wheel()
run_command(f'{sys.executable} -m pip install {wheel}{extra_dep}')
"""
Build docker image, start container, then set its SSH service port to VSO variable "docker_port".
Usage:
python start_docker.py <nni-version> <container-name> <password-in-docker>
"""
import random
import socket
import sys
from _common import build_wheel, run_command, set_variable
# find idle port
port = random.randint(10000, 20000)
while True:
sock = socket.socket()
if sock.connect_ex(('localhost', port)) != 0:
break # failed to connect, so this is idle
sock.close()
port = random.randint(10000, 20000)
version = sys.argv[1]
container = sys.argv[2]
password = sys.argv[3]
run_command(f'docker build --build-arg NNI_RELEASE={version} -t nnidev/nni-nightly .')
run_command(f'docker run -d -t -p {port}:22 --name {container} nnidev/nni-nightly')
run_command(f'docker exec {container} useradd --create-home --password {password} nni')
run_command(['docker', 'exec', container, 'bash', '-c', f'echo "nni:{password}" | chpasswd'])
run_command(f'docker exec {container} service ssh start')
set_variable('docker_port', port)
import sys
from _common import run_command
name = sys.argv[1]
run_command(f'docker container stop {name}')
run_command(f'docker container rm {name}')
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment