def setUp(self):
self.session = MagicMock()
self.name = "user"
self.rmqvhost = "vhost"
self.ftpuser = "******"
self.user = User(self.name, self.rmqvhost, self.ftpuser)
self.frontend_scanid = "frontend_scanid"
self.nb_files = randint(50, 100)
self.scan = Scan(frontend_scanid=self.frontend_scanid,
user_id=self.user.id)
self.scan.user = self.user
def test001___init__(self):
user = User(self.name, self.rmqvhost, self.ftpuser)
self.assertEqual(user.name, self.name)
self.assertEqual(user.rmqvhost, self.rmqvhost)
self.assertEqual(user.ftpuser, self.ftpuser)
print("SQL directory does not exist {0}" "..creating".format(dirname))
os.makedirs(dirname)
os.chmod(dirname, 0o777)
elif not (os.path.isdir(dirname)):
print("Error. SQL directory is a not a dir {0}" "".format(dirname))
raise IrmaDatabaseError("Can not create Brain database dir")
if not os.path.exists(db_name):
# touch like method to create a rw-rw-rw- file for db
open(db_name, 'a').close()
os.chmod(db_name, 0o666)
# Retrieve database informations
url = config.get_sql_url()
engine = create_engine(url, echo=config.sql_debug_enabled())
# and create Database in case
Base.metadata.create_all(engine)
with session_transaction() as session:
try:
user = User.get_by_rmqvhost(session, rmqvhost=rmqvhost)
print("rmqvhost {0} is already assigned to user {1}. "
"Updating with new parameters.".format(user.name, user.rmqvhost))
user = user.load(user.id, session)
user.name = name
user.ftpuser = ftpuser
user.update(['name', 'ftpuser'], session)
except IrmaDatabaseResultNotFound:
user = User(name=name, rmqvhost=rmqvhost, ftpuser=ftpuser)
user.save(session)
