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): 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 populate_initial_data(dm: Dimensigon): from dimensigon.domain.entities import ActionTemplate, Locker, Server, User, Parameter with session_scope(session=dm.get_session()) as session: gates = dm.config.http_conf.get('binds', None) SchemaChanges.set_initial(session) dm.server_id = Server.set_initial(session, gates) Locker.set_initial(session, unlock=True) ActionTemplate.set_initial(session) User.set_initial(session) Parameter.set_initial(session)
def test_get_servers_from_scope_more_than_min_quorum( self, mock_get_now, mock_app): mock_get_now.return_value = dt.datetime(2019, 4, 1, tzinfo=dt.timezone.utc) Server.set_initial() servers = [] for i in range(0, 7): s = Server(f'node{i}', port=5000, created_on=old_age) if i == 0: r = Route(s, cost=0) else: r = Route(s, random.choice(servers), cost=i) db.session.add_all([s, r]) servers.append(s) mock_get_now.return_value = dt.datetime(2019, 4, 2, tzinfo=dt.timezone.utc) mock_app.dm.cluster_manager.get_alive.return_value = [ s.id for s in servers ] s62 = Server(f'node72', port=5000) Route(s62, random.choice(servers), cost=6) db.session.add(s62) quorum = get_servers_from_scope(scope=Scope.CATALOG) self.assertEqual(8, len(quorum)) self.assertNotIn(s62, quorum) self.assertIn(s, quorum) self.assertIn(Server.get_current(), quorum)