def setUp(self): setup_config('data/config_openstack.py') self.platform = "origin_1" mocked_image = Mock(id=1, status='active', get=Mock(return_value='snapshot'), min_disk=20, min_ram=2, instance_type_flavorid=1) type(mocked_image).name = PropertyMock(return_value='test_origin_1') with patch('core.network.Network', Mock()), patch( 'core.connection.Virsh', Mock()), patch.multiple( 'core.utils.openstack_utils', nova_client=Mock(), neutron_client=Mock(return_value=Mock()), glance_client=Mock(return_value=Mock())), patch( 'vmpool.platforms.OpenstackPlatforms.images', Mock(return_value=[mocked_image])): from flask import Flask self.app = Flask(__name__) from vmpool.virtual_machines_pool import VirtualMachinesPool self.app.pool = VirtualMachinesPool(self.app) self.ctx = self.app.test_request_context() self.ctx.push()
def setUp(self): setup_config('data/config.py') self.address = ("localhost", 9001) with patch( 'core.connection.Virsh', Mock(), ), patch('core.network.Network', Mock()), patch('core.db.Database', DatabaseMock()): from vmmaster.server import VMMasterServer self.vmmaster = VMMasterServer(reactor, self.address[1]) self.pool = self.vmmaster.app.pool server_is_up(self.address) self.desired_caps = { 'desiredCapabilities': { 'platform': self.pool.platforms.platforms.keys()[0] } } from core.config import config config.SESSION_TIMEOUT = 1 self.ctx = self.vmmaster.app.app_context() self.ctx.push()
def setUpClass(cls): from core.config import setup_config setup_config('data/config.py') cls.app = Flask(__name__) cls.app.database = DatabaseMock() cls.app.sessions = Mock()
def setUpClass(cls): setup_config('data/config.py') cls.db = Database(config.DATABASE) Base.metadata.drop_all(cls.db.engine) Base.metadata.create_all(cls.db.engine) cls.size = 20
def setUp(self): from core.config import setup_config setup_config('data/config.py') with patch('core.network.Network', Mock(name='Network')), patch( 'core.connection.Virsh', Mock(name='Virsh')), patch( 'core.utils.init.home_dir', Mock(return_value=fake_home_dir())), patch( 'core.logger.setup_logging', Mock(return_value=Mock())), patch( 'core.db.Database', DatabaseMock()): from vmmaster.server import create_app self.app = create_app() self.vmmaster_client = self.app.test_client() self.platforms = self.app.pool.platforms.platforms self.platform = sorted(self.app.pool.platforms.platforms.keys())[0] self.desired_caps = { 'desiredCapabilities': { 'platform': self.platform } } self.ctx = self.app.app_context() self.ctx.push()
def setUp(self): from core.config import setup_config setup_config('data/config_openstack.py') with patch('core.utils.init.home_dir', Mock(return_value=fake_home_dir())), patch( 'core.db.Database', DatabaseMock()), patch( 'vmpool.virtual_machines_pool.VirtualMachinesPool', Mock()): from vmpool.server import create_app from vmpool.virtual_machines_pool import VirtualMachinesPool self.app = create_app() self.app.pool = VirtualMachinesPool( self.app, 'unnamed', platforms_class=Mock(), preloader_class=Mock(), artifact_collector_class=Mock(), endpoint_preparer_class=Mock(), endpoint_remover_class=Mock()) self.vmmaster_client = self.app.test_client() self.platform = 'origin_1' self.ctx = self.app.app_context() self.ctx.push()
def setUp(self): from flask import Flask from core.config import setup_config setup_config('data/config_openstack.py') self.app = Flask(__name__) self.app.database = DatabaseMock() self.pool = Mock(id=1, provider=Provider(name='fake', url='no//url')) self.mocked_origin = Mock(short_name="platform_1", id=1, status="active", get=Mock(return_value="snapshot"), min_disk=20, min_ram=2, instance_type_flavorid=1) self.ctx = self.app.app_context() self.ctx.push() with patch("core.utils.openstack_utils.nova_client", Mock(return_value=Mock())), patch("flask.current_app", self.app): from core.db.models import OpenstackClone self.clone = OpenstackClone(self.mocked_origin, "preloaded", self.pool)
def setUpClass(cls): cls.app = Flask(__name__) cls.app.running = True cls.app.database = None from core.config import setup_config setup_config('data/config_openstack.py')
def setUp(self): setup_config('data/config.py') self.platform = "test_origin_1" with patch('core.connection.Virsh', Mock()), patch('core.network.Network', Mock()): from vmpool.virtual_machines_pool import VirtualMachinesPool self.pool = VirtualMachinesPool()
def setUpClass(cls): setup_config('data/config_openstack.py') cls.app = Flask(__name__) cls.app.database = DatabaseMock() cls.app.sessions = Mock() cls.session_id = 1 cls.artifact_dir = os.sep.join([config.SCREENSHOTS_DIR, str(cls.session_id)]) if not os.path.exists(cls.artifact_dir): os.mkdir(cls.artifact_dir)
def setUpClass(cls): setup_config('data/config.py') from flask import Flask cls.app = Flask(__name__) cls.app.sessions = None from core import db cls.app.database = db.Database(config.DATABASE)
def setUp(self): setup_config('data/config.py') from flask import Flask self.app = Flask(__name__) self.app.sessions = Mock() self.app.sessions.app = self.app from core.sessions import SessionWorker self.worker = SessionWorker(self.app.sessions)
def setUp(self): with patch("docker.DockerClient", Mock( containers=Mock( create=Mock(), get=Mock() ) )): from core.config import setup_config from core.clients.docker_client import DockerManageClient setup_config('data/config_openstack.py') self._docker_client = DockerManageClient()
def setUp(self): setup_config('data/config_with_preloaded.py') self.address = ("localhost", 9001) self.desired_caps = {'desiredCapabilities': {'platform': 'origin_1'}} mocked_image = Mock(id=1, status='active', get=Mock(return_value='snapshot'), min_disk=20, min_ram=2, instance_type_flavorid=1) type(mocked_image).name = PropertyMock(return_value='test_origin_1') with patch( 'core.connection.Virsh', Mock(), ), patch('core.network.Network', Mock()), patch( 'core.db.Database', Mock(add=Mock(side_effect=set_primary_key))), patch.multiple( 'core.utils.openstack_utils', neutron_client=Mock(return_value=Mock()), glance_client=Mock(return_value=Mock()), nova_client=Mock(return_value=Mock())), patch.multiple( 'vmpool.clone.OpenstackClone', _wait_for_activated_service=custom_wait, get_network_id=Mock(return_value=1), get_network_name=Mock( return_value='Local-Net')), patch.multiple( 'vmpool.platforms.OpenstackPlatforms', images=Mock(return_value=[mocked_image]), flavor_params=Mock(return_value={ 'vcpus': 1, 'ram': 2 }), limits=Mock( return_value={ 'maxTotalCores': 10, 'maxTotalInstances': 10, 'maxTotalRAMSize': 100, 'totalCoresUsed': 0, 'totalInstancesUsed': 0, 'totalRAMUsed': 0 })): from vmmaster.server import VMMasterServer self.vmmaster = VMMasterServer(reactor, self.address[1]) self.ctx = self.vmmaster.app.app_context() self.ctx.push() self.pool = self.vmmaster.app.pool server_is_up(self.address)
def setUpClass(cls): cls.app = Flask(__name__) cls.app.database = None from core.config import setup_config setup_config('data/config.py') cls.method = "GET" from base64 import urlsafe_b64encode from core.auth.custom_auth import anonymous cls.headers = dict(Authorization="Basic " + urlsafe_b64encode( str(anonymous.username) + ":" + str(anonymous.password)))
def setUp(self): setup_config('data/config.py') self.platform = "test_origin_1" self.app = Flask(__name__) self.ctx = self.app.app_context() self.ctx.push() with patch('core.connection.Virsh', Mock()), patch('core.network.Network', Mock()): from vmpool.virtual_machines_pool import VirtualMachinesPool self.pool = VirtualMachinesPool(self.app)
def setUpClass(cls): setup_config('data/config.py') from flask import Flask cls.app = Flask(__name__) cls.app.sessions = None from core import db cls.app.database = db.Database(config.DATABASE) with patch('core.utils.init.home_dir', Mock(return_value=config.BASEDIR)): from vmmaster import cleanup cls.cleanup = cleanup
def setUp(self): setup_config('data/config_openstack.py') self.platform = "origin_1" self.address = ("localhost", 9001) self.mocked_image = Mock(id=1, status='active', get=Mock(return_value='snapshot'), min_disk=20, min_ram=2, instance_type_flavorid=1) type(self.mocked_image).name = PropertyMock( return_value='test_origin_1') with patch( 'core.connection.Virsh', Mock(), ), patch('core.network.Network', Mock()), patch.multiple( 'core.utils.openstack_utils', neutron_client=Mock(return_value=Mock()), nova_client=Mock(return_value=Mock()), glance_client=Mock(return_value=Mock())), patch.multiple( 'vmpool.platforms.OpenstackPlatforms', images=Mock(return_value=[self.mocked_image]), flavor_params=Mock(return_value={ 'vcpus': 1, 'ram': 2 }), limits=Mock( return_value={ 'maxTotalCores': 10, 'maxTotalInstances': 10, 'maxTotalRAMSize': 100, 'totalCoresUsed': 0, 'totalInstancesUsed': 0, 'totalRAMUsed': 0 })): from flask import Flask self.app = Flask(__name__) from vmpool.virtual_machines_pool import VirtualMachinesPool self.app.pool = VirtualMachinesPool(self.app) from vmpool.platforms import Platforms self.app.platforms = Platforms() self.ctx = self.app.test_request_context() self.ctx.push()
def setUp(self): setup_config('data/config.py') self.host = "localhost" self.port = 9001 self.address = (self.host, self.port) self.vmmaster = vmmaster_server_mock(self.port) server_is_up(self.address) self.free_port = get_free_port() self.ctx = self.vmmaster.app.app_context() self.ctx.push() from core.sessions import Session self.session = Session() self.session.endpoint_ip = "localhost"
def setUp(self): setup_config('data/config.py') self.address = ("localhost", 9001) super(TestServer, self).setUp() self.desired_caps = { 'desiredCapabilities': { 'platform': 'test_origin_1' } } self.ctx = self.vmmaster.app.app_context() self.ctx.push()
def setUp(self): setup_config('data/config.py') self.address = ("localhost", 9001) super(TestSessionSteps, self).setUp() self.desired_caps = { 'desiredCapabilities': { 'platform': self.vmmaster.app.pool.platforms.platforms.keys()[0] } } self.ctx = self.vmmaster.app.app_context() self.ctx.push()
def setUp(self): from core.db import Database setup_config('data/config.py') self.platform_name = "origin_1" self.app = Flask(__name__) self.ctx = self.app.app_context() self.ctx.push() self.app.database = Database() self.app.sessions = Mock() self.mocked_image = Mock( id=1, status='active', get=Mock(return_value='snapshot'), min_disk=20, min_ram=2, instance_type_flavorid=1, ) type(self.mocked_image).name = PropertyMock( return_value='test_origin_1') with patch.multiple( 'vmpool.platforms.OpenstackPlatforms', images=Mock(return_value=[self.mocked_image]), flavor_params=Mock(return_value={ 'vcpus': 1, 'ram': 2 }), limits=Mock( return_value={ 'maxTotalCores': 10, 'maxTotalInstances': 10, 'maxTotalRAMSize': 100, 'totalCoresUsed': 0, 'totalInstancesUsed': 0, 'totalRAMUsed': 0 }), ): from vmpool.virtual_machines_pool import VirtualMachinesPool self.pool = VirtualMachinesPool(self.app, preloader_class=Mock(), artifact_collector_class=Mock(), endpoint_preparer_class=Mock()) self.ctx = self.app.app_context() self.ctx.push() self.pool.register()
def setUp(self): setup_config('data/config_openstack.py') self.host = "localhost" self.port = config.PORT self.address = (self.host, self.port) self.vmmaster = vmmaster_server_mock(self.port) self.assertTrue(server_is_up(self.address)) self.free_port = get_free_port() self.ctx = self.vmmaster.app.app_context() self.ctx.push() from core.db.models import Session, Provider, Endpoint self.session = Session('some_platform') provider = Provider(name='noname', url='nourl') endpoint = Endpoint(Mock(), '', provider) endpoint.ip = 'localhost' self.session.endpoint = endpoint
def setUpClass(cls): setup_config("data/config_openstack.py") body = { "sessionId": None, "desiredCapabilities": { "platform": "some_platform", "browserName": "firefox", "version": "", "javascriptEnabled": True } } session_request_body = json.dumps(body) session_request_headers = { 'content-length': '%s' % len(session_request_body), 'accept-encoding': 'identity', 'Connection': 'close', 'accept': 'application/json', 'user-agent': 'Python-urllib/2.7', 'host': '127.0.0.1:9000', 'content-type': 'application/json;charset=UTF-8', } cls.request = Mock() cls.request.method = "POST" cls.request.path = "/wd/hub/session" cls.request.headers = dict() cls.request.headers.update(session_request_headers) cls.request.data = session_request_body cls.webdriver_server = ServerMock(cls.host, get_free_port()) cls.webdriver_server.start() cls.vmmaster_agent = ServerMock(cls.host, get_free_port()) cls.vmmaster_agent.start() cls.vnc_server = ServerMock(cls.host, get_free_port()) cls.vnc_server.start() cls.app = Flask(__name__) cls.app.database = None cls.app.sessions = None cls.app.database_task_queue = Mock() cls.app.pool = Mock()
def setUp(self): from core.config import setup_config setup_config('data/config_openstack.py') with patch('core.utils.init.home_dir', Mock(return_value=fake_home_dir())), patch( 'core.video.start_vnc_proxy', Mock(return_value=(99999, 99999))), patch( 'core.utils.kill_process', Mock()), patch('core.db.Database', DatabaseMock()): from vmmaster.server import create_app self.app = create_app() self.vmmaster_client = self.app.test_client() self.platform = 'origin_1' self.desired_caps = { 'desiredCapabilities': { 'platform': self.platform } } self.ctx = self.app.app_context() self.ctx.push()
def setUp(self): setup_config('data/config_openstack.py') self.mocked_origin = Mock(short_name="platform_1") self.pool = Mock(pool=dict(), using=dict())
def setUp(self): from core.config import setup_config setup_config('data/config_openstack.py')
def setUp(self): with patch("docker.DockerClient", Mock()): from core.config import setup_config from core.clients.docker_client import DockerManageClient setup_config('data/config.py') self.docker_client = DockerManageClient()
def setUpClass(cls): setup_config('data/config.py') cls.app = Flask(__name__) cls.app.database = DatabaseMock() cls.app.sessions = Mock()
import logging from datetime import datetime, timedelta from sqlalchemy import create_engine from sqlalchemy.orm import sessionmaker from sqlalchemy.exc import ArgumentError from core.config import config, setup_config from core.db.models import Session, User from core.utils import change_user_vmmaster from core.utils.init import home_dir from shutil import rmtree from errno import ENOENT setup_config('%s/config.py' % home_dir()) log = logging.getLogger(__name__) engine = create_engine(config.DATABASE) session_factory = sessionmaker(bind=engine) def transaction(func): def wrapper(*args, **kwargs): dbsession = session_factory() try: return func(dbsession=dbsession, *args, **kwargs) except: dbsession.rollback() raise finally: