def test_insert(self): log = BuildLog(self.env, build=1, step='test', generator='distutils', filename='1.log.gz') full_file = log.get_log_file('1.log.gz') if os.path.exists(full_file): os.remove(full_file) log.messages = [(BuildLog.INFO, 'running tests'), (BuildLog.ERROR, 'tests failed')] log.insert() self.assertNotEqual(None, log.id) db = self.env.get_db_cnx() cursor = db.cursor() cursor.execute( "SELECT build,step,generator,filename FROM bitten_log " "WHERE id=%s", (log.id, )) self.assertEqual((1, 'test', 'distutils', '1.log.gz'), cursor.fetchone()) lines = gzip.open(full_file, "rb").readlines() self.assertEqual('running tests\n', lines[0]) self.assertEqual('tests failed\n', lines[1]) if os.path.exists(full_file): os.remove(full_file)
def add_log_table(env, db): """Add a table for storing the builds logs.""" from bitten.model import BuildLog, BuildStep cursor = db.cursor() connector, _ = DatabaseManager(env)._get_connector() for table in BuildLog._schema: for stmt in connector.to_sql(table): cursor.execute(stmt) cursor.execute("SELECT build,name,log FROM bitten_step " "WHERE log IS NOT NULL") for build, step, log in cursor: build_log = BuildLog(env, build, step) build_log.messages = [(BuildLog.INFO, msg) for msg in log.splitlines()] build_log.insert(db) cursor.execute("CREATE TEMPORARY TABLE old_step AS SELECT * FROM bitten_step") cursor.execute("DROP TABLE bitten_step") for table in BuildStep._schema: for stmt in connector.to_sql(table): cursor.execute(stmt) cursor.execute("INSERT INTO bitten_step (build,name,description,status," "started,stopped) SELECT build,name,description,status," "started,stopped FROM old_step")
def add_log_table(env, db): """Add a table for storing the builds logs.""" from bitten.model import BuildLog, BuildStep cursor = db.cursor() connector, _ = DatabaseManager(env)._get_connector() for table in BuildLog._schema: for stmt in connector.to_sql(table): cursor.execute(stmt) cursor.execute("SELECT build,name,log FROM bitten_step " "WHERE log IS NOT NULL") for build, step, log in cursor: build_log = BuildLog(env, build, step) build_log.messages = [(BuildLog.INFO, msg) for msg in log.splitlines()] build_log.insert(db) cursor.execute( "CREATE TEMPORARY TABLE old_step AS SELECT * FROM bitten_step") cursor.execute("DROP TABLE bitten_step") for table in BuildStep._schema: for stmt in connector.to_sql(table): cursor.execute(stmt) cursor.execute("INSERT INTO bitten_step (build,name,description,status," "started,stopped) SELECT build,name,description,status," "started,stopped FROM old_step")
def test_insert_empty(self): log = BuildLog(self.env, build=1, step='test', generator='distutils', filename="1.log") full_file = log.get_log_file('1.log') if os.path.exists(full_file): os.remove(full_file) log.messages = [] log.insert() self.assertNotEqual(None, log.id) db = self.env.get_db_cnx() cursor = db.cursor() cursor.execute("SELECT build,step,generator,filename FROM bitten_log " "WHERE id=%s", (log.id,)) self.assertEqual((1, 'test', 'distutils', '1.log'), cursor.fetchone()) file_exists = os.path.exists(full_file) if file_exists: os.remove(full_file) assert not file_exists
def test_insert_and_delete_files(self): # create - files should be created automatically build_log = BuildLog(self.env, build=1, step='test', generator='make') build_log.messages = [(BuildLog.INFO, 'running')] build_log.insert() # fetch it fresh - check object and files build_log = BuildLog.fetch(self.env, id=build_log.id) self.assertEquals(build_log.filename, "%s.log" % build_log.id) log_file = build_log.get_log_file(build_log.filename) levels_file = log_file+'.levels' self.failUnless(os.path.exists(log_file), 'log_file does not exist') self.failUnless(os.path.exists(levels_file), 'levels_file does not exist') self.assertEquals(build_log.messages, [(BuildLog.INFO, 'running')]) # delete - object and file should be gone build_log.delete() self.assertEquals(None, BuildLog.fetch(self.env, id=build_log.id)) self.failIf(os.path.exists(log_file), 'log_file exists after delete()') self.failIf(os.path.exists(levels_file), 'levels_file exists after delete()')
def test_insert(self): log = BuildLog(self.env, build=1, step='test', generator='distutils', filename='1.log') full_file = log.get_log_file('1.log') if os.path.exists(full_file): os.remove(full_file) log.messages = [ (BuildLog.INFO, 'running tests'), (BuildLog.ERROR, 'tests failed') ] log.insert() self.assertNotEqual(None, log.id) db = self.env.get_db_cnx() cursor = db.cursor() cursor.execute("SELECT build,step,generator,filename FROM bitten_log " "WHERE id=%s", (log.id,)) self.assertEqual((1, 'test', 'distutils', '1.log'), cursor.fetchone()) lines = open(full_file, "rb").readlines() self.assertEqual('running tests\n', lines[0]) self.assertEqual('tests failed\n', lines[1]) if os.path.exists(full_file): os.remove(full_file)
step.status = BuildStep.SUCCESS step.errors += [error.gettext() for error in elem.children('error')] # TODO: step.update(db=db) step.delete(db=db) step.insert(db=db) # Collect log messages from the request body for idx, log_elem in enumerate(elem.children('log')): build_log = BuildLog(self.env, build=build.id, step=stepname, generator=log_elem.attr.get('generator'), orderno=idx) for message_elem in log_elem.children('message'): build_log.messages.append((message_elem.attr['level'], message_elem.gettext())) build_log.insert(db=db) # Collect report data from the request body for report_elem in elem.children('report'): report = Report(self.env, build=build.id, step=stepname, category=report_elem.attr.get('category'), generator=report_elem.attr.get('generator')) for item_elem in report_elem.children(): item = {'type': item_elem.name} item.update(item_elem.attr) for child_elem in item_elem.children(): item[child_elem.name] = child_elem.gettext() report.items.append(item) report.insert(db=db) # If this was the last step in the recipe we mark the build as
# TODO: step.update(db=db) step.delete(db=db) step.insert(db=db) # Collect log messages from the request body for idx, log_elem in enumerate(elem.children('log')): build_log = BuildLog(self.env, build=build.id, step=stepname, generator=log_elem.attr.get('generator'), orderno=idx) for message_elem in log_elem.children('message'): build_log.messages.append( (message_elem.attr['level'], message_elem.gettext())) build_log.insert(db=db) # Collect report data from the request body for report_elem in elem.children('report'): report = Report(self.env, build=build.id, step=stepname, category=report_elem.attr.get('category'), generator=report_elem.attr.get('generator')) for item_elem in report_elem.children(): item = {'type': item_elem.name} item.update(item_elem.attr) for child_elem in item_elem.children(): item[child_elem.name] = child_elem.gettext() report.items.append(item) report.insert(db=db)