def setUp(self): """Create and configure a new app instance for each test.""" self.maxDiff = None # create the app with common test config self.set_scoped_session() self.app = create_app('test') if self.db_uris: self.app.config['SQLALCHEMY_DATABASE_URI'] = self.db_uris[0] remove_db_file(self.db_uris[0]) self.app.config['SERVER_NAME'] = 'node1' self.app_context = self.app.app_context() self.client = self.app.test_client() self.app_context.push() self.app2 = create_app('test') if self.db_uris: self.app2.config['SQLALCHEMY_DATABASE_URI'] = self.db_uris[1] remove_db_file(self.db_uris[1]) self.app2.config['SERVER_NAME'] = 'node2' self.app2_context = self.app2.app_context() self.client2 = self.app2.test_client() self._fill_database(self.app) self._fill_database(self.app2) self.generate_auth() super().setUp()
def setUp(self) -> None: self.file1 = os.path.join(DEST_FOLDER, 'server1.tempfile.log') self.file2 = os.path.join(DEST_FOLDER, 'server2.tempfile.log') self.remove_files() self.app1 = create_app(Server1()) self.app2 = create_app(Server2()) self.client1 = self.app1.test_client() self.client2 = self.app2.test_client() self.lines = ['line 1\nline 2\n', 'line 3\n', 'line 4\nline 5\n'] self.i = 0 self.append_data()
def setUp(self): """Create and configure a new app instance for each test.""" # create the app with common test config self.app = create_app('test') self.app.config['SECURIZER'] = True @self.app.route('/', methods=['GET', 'POST']) @securizer def home(): return {'msg': 'default response'} self.app_context = self.app.app_context() self.app_context.push() self.client = self.app.test_client() db.init_app(self.app) db.create_all() self.server = Server('me', port=5000, me=True) db.session.add(self.server) self.dim = generate_dimension('dimension') self.dim.current = True db.session.add(self.dim) self.token = create_access_token('test') self.auth = HTTPBearerAuth(self.token) self.url = 'https://me:5000/'
def setUp(self): """Create and configure a new app instance for each test.""" # create the app with common test config self.app = create_app('test') self.app_context = self.app.app_context() self.app_context.push() db.create_all()
def setUp(self): """Create and configure a new app instance for each test.""" # create the app with common test config self.app = create_app('test') self.app_context = self.app.app_context() self.app_context.push() self.client = self.app.test_client() self.headers = { "Authorization": f"Bearer {create_access_token('00000000-0000-0000-0000-000000000001')}" } db.create_all() set_initial() self.soft = Software(id='aaaaaaaa-1234-5678-1234-56781234aaa1', name='test', version='1', filename='file') self.soft2 = Software(id='aaaaaaaa-1234-5678-1234-56781234aaa2', name='test', version='2', filename='file') self.ssa = SoftwareServerAssociation(software=self.soft, server=Server.get_current(), path='/root') db.session.add_all([self.soft, self.soft2, self.ssa]) db.session.commit()
def setUp(self): self.maxDiff = None self.app_join = create_app('test') self.app_join.config['SERVER_NAME'] = 'join' self.app_join_context = self.app_join.app_context() self.client_join = self.app_join.test_client() with self.app_join.app_context(): db.create_all() set_initial(server=False) self.join_server_id = Server.set_initial() db.session.commit() self.mock_dm = mock.MagicMock() self.mock_dm.flask_app = self.app_join self.mock_dm.engine = db.engine self.mock_dm.catalog_manager.db_update_catalog = update_db_catalog super().setUp() self.app.config['SECURIZER'] = True self.app_join.config['SECURIZER'] = True self.token = create_access_token(JOIN, additional_claims={'applicant': 'me'}) self.setUpPyfakefs() open('/origin_key', 'w').write('keyfile') open('/origin_cert', 'w').write('certfile')
def setUp(self): """Create and configure a new app instance for each test.""" # create the app with common test config self.app = create_app('test') self.app_context = self.app.app_context() self.app_context.push() self.client = self.app.test_client() self.headers = { "Authorization": f"Bearer {create_access_token('00000000-0000-0000-0000-000000000001')}" } db.create_all() # set_initial(server=False) self.srv1 = Server('node1', id='00000000-0000-0000-0000-000000000001', me=True) self.file = File(source_server=self.srv1, target='/etc/ssh/sshd_config', id='00000000-0000-0000-0000-000000000002') self.fsa = FileServerAssociation(file=self.file, destination_server=self.srv1) db.session.add_all([self.srv1, self.file, self.fsa]) db.session.commit()
def setUp(self) -> None: self.maxDiff = None self.app = create_app('test') self.app.config['SECURIZER'] = False # user sqlite in a file as flush does not emits states in an inmemory database # self.app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///'+os.path.join(basedir, 'test.db') self.app_context = self.app.app_context() self.app_context.push() db.create_all()
def setUp(self): self.maxDiff = None self.set_scoped_session() self.app = create_app('test') self.app.config['SERVER_NAME'] = 'me' self.app_context = self.app.app_context() self.client = self.app.test_client() self.app_context.push() db.create_all()
def setUp(self): """Create and configure a new app instance for each test.""" # create the app with common test config self.app = create_app('test') self.app_context = self.app.app_context() self.app_context.push() self.client = self.app.test_client() self.auth = HTTPBearerAuth(create_access_token('00000000-0000-0000-0000-000000000001')) db.create_all()
def setUp(self) -> None: self.maxDiff = None set_test_scoped_session(db) self.app1 = create_app('test') self.app1.config['SERVER_NAME'] = 'node1' self.app_context = self.app1.app_context() self.app_context.push() self.app2 = create_app('test') self.app2.config['SERVER_NAME'] = 'node2' self.app3 = create_app('test') self.app3.config['SERVER_NAME'] = 'node3' self.fill_database('node1') with self.app2.app_context(): self.fill_database('node2') with self.app3.app_context(): self.fill_database('node3')
def setUp(self): """Create and configure a new app instance for each test.""" # create the app with common test config self.app = create_app('test') self.app_context = self.app.app_context() self.app_context.push() self.client = self.app.test_client() db.create_all() set_initial() self.auth = HTTPBearerAuth( create_access_token(User.get_by_name('root').id))
def setUp(self): """Create and configure a new app instance for each test.""" # create the app with common test config self.app = create_app('test') self.app_context = self.app.app_context() self.app_context.push() self.client = self.app.test_client() self.auth = HTTPBearerAuth(create_access_token('00000000-0000-0000-0000-000000000001')) db.create_all() self.src = Server('source', port=5000) self.dst = Server('destination', port=5000)
def setUp(self): """Create and configure a new app instance for each test.""" # create the app with common test config self.app = create_app('test') self.app_context = self.app.app_context() self.app_context.push() self.client = self.app.test_client() self.auth = HTTPBearerAuth(create_access_token('00000000-0000-0000-0000-000000000001')) db.create_all() self.me = Server('me', port=5000, _me=True) self.remote = Server('remote', port=5000) db.session.add_all([self.remote, self.me]) self.maxDiff = None
def setUp(self, mock_now): """Create and configure a new app instance for each test.""" mock_now.return_value = dt.datetime(2019, 4, 1, tzinfo=dt.timezone.utc) # create the app with common test config self.app = create_app('test') self.app_context = self.app.app_context() self.app_context.push() self.client = self.app.test_client() db.create_all() Locker.set_initial() Server.set_initial() db.session.commit()
def setUp(self): """Create and configure a new app instance for each test.""" # create the app with common test config self.app = create_app('test') self.app_context = self.app.app_context() self.app_context.push() self.client = self.app.test_client() self.headers = { "Authorization": f"Bearer {create_access_token('00000000-0000-0000-0000-000000000001')}" } db.create_all() db.session.commit()
def setUp(self) -> None: self.maxDiff = None self.set_scoped_session() self.app = create_app('test') if self.db_uris: self.app.config['SQLALCHEMY_DATABASE_URI'] = self.db_uris[0] remove_db_file(self.db_uris[0]) self.app.config['SERVER_NAME'] = 'node1' self.app_context = self.app.app_context() self.client = self.app.test_client() self.app_context.push() self._fill_database() self.generate_auth() super().setUp()
def setUp(self): """Create and configure a new app instance for each test.""" # create the app with common test config self.app = create_app('test') self.app_context = self.app.app_context() self.app_context.push() self.client = self.app.test_client() self.auth = HTTPBearerAuth( create_access_token('00000000-0000-0000-0000-000000000001')) db.create_all() self.at = ActionTemplate(name='action', version=1, action_type=ActionType.SHELL, code='code to run', expected_stdout='expected output', expected_rc=0, system_kwargs={}) ActionTemplate.set_initial()
def setUp(self): """Create and configure a new app instance for each test.""" # create the app with common test config self.app = create_app('test') self.app_context = self.app.app_context() self.app_context.push() self.client = self.app.test_client() self.headers = { "Authorization": f"Bearer {create_access_token('00000000-0000-0000-0000-000000000001')}" } db.create_all() set_initial() self.n1 = Server("node1", port=8000) Route(self.n1, cost=0) self.n2 = Server("node2", port=8000) Route(self.n2, cost=0) db.session.add_all([self.n1, self.n2]) db.session.commit() self.datamark = Catalog.max_catalog(str)
def setUp(self): """Create and configure a new app instance for each test.""" # create the app with common test config self.app = create_app('test') self.app_context = self.app.app_context() self.app_context.push() self.client = self.app.test_client() self.headers = {"Authorization": f"Bearer {create_access_token('00000000-0000-0000-0000-000000000001')}"} db.create_all() self.at1 = ActionTemplate(id='11111111-2222-3333-4444-555555550001', name='action1', version=1, action_type=ActionType.SHELL, code='code to run', expected_stdout='expected output', expected_stderr='stderr', expected_rc=0, system_kwargs={}) self.at2 = ActionTemplate(id='11111111-2222-3333-4444-555555550002', name='action2', version=1, action_type=ActionType.SHELL, code='code to run', expected_stdout='expected output', expected_stderr='stderr', expected_rc=0, system_kwargs={}) self.o = Orchestration(id='11111111-2222-3333-4444-666666660001', name='orchestration_name', version=1, description='desc')
def setUp(self): self.maxDiff = None """Create and configure a new app instance for each test.""" # create the app with common test config self.app = create_app('test') self.app_context = self.app.app_context() self.app_context.push() self.client = self.app.test_client() self.auth = HTTPBearerAuth( create_access_token('00000000-0000-0000-0000-000000000001')) db.create_all() self.src = Server('source', port=5000, id='00000000-0000-0000-0000-000000000000') self.dst1 = Server('destination1', port=5000, id='00000000-0000-0000-0000-000000000001') self.dst2 = Server('destination2', port=5000, id='00000000-0000-0000-0000-000000000002') self.dst3 = Server('destination3', port=5000, id='00000000-0000-0000-0000-000000000003') db.session.add_all([self.src, self.dst1, self.dst2, self.dst3])
from sqlalchemy import exc, sql from dimensigon import defaults from dimensigon.core import Dimensigon from dimensigon.dshell.output import dprint basedir = os.path.abspath(os.path.dirname(__file__)) PLATFORM = platform.system() from dimensigon.domain.entities import * from dimensigon.web.network import pack_msg2, unpack_msg2 from dimensigon.web import create_app, db, get_root_auth from dimensigon.utils.helpers import generate_symmetric_key, generate_dimension, get_now app: Flask = create_app(os.getenv('FLASK_CONFIG') or 'default') def new(dm: Dimensigon, name: str): dm.create_flask_instance() with dm.flask_app.app_context(): from cryptography import x509 from cryptography.hazmat.backends import default_backend from cryptography.hazmat.primitives import hashes, serialization from cryptography.x509.oid import NameOID import datetime Server.set_initial() count = Dimension.query.count() if count > 0: exit("Only one dimension can be created")
def create_flask_instance(self): if self.flask_app is None: self.flask_app = create_app(self.config.flask_conf) self.flask_app.dm = self
def setUp(self) -> None: self.app = create_app('test')