def main(argv=sys.argv):
if len(argv) < 2:
usage(argv)
config_uri = argv[1]
#options = parse_vars(argv[2:])
setup_logging(config_uri)
#settings = get_appsettings(config_uri, options=options)
settings = get_appsettings(config_uri)
engine = engine_from_config(settings, 'sqlalchemy.')
DBSession.configure(bind=engine)
Base.metadata.create_all(engine)
with transaction.manager:
# Create the groups for two types of users (for now)
auth_group = Group(u'auth', u'Regular user account.')
admin_group = Group(u'admin', u'Administrative user account.')
# Add them to the db session
DBSession.add(auth_group)
DBSession.add(admin_group)
transaction.commit()
# Create the admin user so we can always access the system as admin.
su = User(first_name=u'Ixia', last_name=u'User', username=u'admin',
email=u'*****@*****.**', mobile=random.randint(2140000000, 2149999999))
su._set_password('admin')
su.groups.append(admin_group)
DBSession.add(su)
transaction.commit()
# Configure global system parameters.
config = GlobalConfigs(bps=u'192.168.0.132', metasploit=u'192.168.0.150', splunk=u'192.168.0.133',
ips=u'', ngfw=u'192.168.0.134', dlp=u'192.168.0.140',
windows=u'192.168.0.132', kali=u'192.168.0.170', atip=u'192.168.0.171', version=u'01.00.00')
DBSession.add(config)
transaction.commit()
def test_passing_view(self):
from cyberrange.models.models import (
Base,
MyModel,
)
Base.metadata.create_all(self.engine)
with transaction.manager:
model = MyModel(name='one', value=55)
DBSession.add(model)
from cyberrange.views.views import my_view
request = testing.DummyRequest()
info = my_view(request)
self.assertEqual(info['one'].name, 'one')
self.assertEqual(info['project'], 'CyberRange')
def setUp(self):
self.config = testing.setUp()
from sqlalchemy import create_engine
self.engine = create_engine('sqlite://')
DBSession.configure(bind=self.engine)
def tearDown(self):
DBSession.remove()
testing.tearDown()