From 3c4da757a970603c965f33970053295a54b57a02 Mon Sep 17 00:00:00 2001 From: Leandro Lucarella Date: Tue, 13 Mar 2007 16:00:13 +0000 Subject: [PATCH] Mergear archivos a comparar de prueba y comando. --- TODO.txt | 1 - doc/testdata.py | 3 ++- sercom/tester.py | 44 ++++++++++++++++++++++++++++---------------- 3 files changed, 30 insertions(+), 18 deletions(-) diff --git a/TODO.txt b/TODO.txt index 4274d25..121ff89 100644 --- a/TODO.txt +++ b/TODO.txt @@ -1,5 +1,4 @@ - Backend (luca) - * Mergear archivos a comparar de prueba y comando. * Hacer el finalizador de entregas (selección de entregas a corregir y round robbin para repartir tps). * Hacer un control del límite de tiempo que un subproceso puede tomar más diff --git a/doc/testdata.py b/doc/testdata.py index 6009f5f..594d351 100644 --- a/doc/testdata.py +++ b/doc/testdata.py @@ -38,7 +38,8 @@ c = Curso(anio=2007, cuatrimestre=1, numero=1, descripcion=u'Martes', cp1 = e1.add_caso_de_prueba(nombre=u'Sin parámetros', retorno=0, descripcion=u'Un caso', comando='./tito') cp2 = e1.add_caso_de_prueba(nombre=u'2 parámetross', retorno=1, - comando='./tito --test -c "con espacios"', terminar_si_falla=False) + comando='./tito --test -c "con espacios"', terminar_si_falla=False, + archivos_a_guardar=('__stdout__',)) # Ejercicios ej1 = c.ejercicios[0] diff --git a/sercom/tester.py b/sercom/tester.py index 51c861e..0326068 100644 --- a/sercom/tester.py +++ b/sercom/tester.py @@ -103,6 +103,24 @@ def unzip(bytes, default_dst='.', specific_dst=dict()): # {{{ zfile.close() #}}} +class Multizip(object): #{{{ + def __init__(self, *zip_streams): + self.zips = [ZipFile(StringIO(z), 'r') for z in zip_streams + if z is not None] + self.names = set() + for z in self.zips: + self.names |= set(z.namelist()) + def read(self, name): + for z in self.zips: + try: + return z.read(name) + except KeyError: + pass + raise KeyError(name) + def namelist(self): + return self.names +#}}} + class SecureProcess(object): #{{{ default = dict( max_tiempo_cpu = 120, @@ -316,12 +334,8 @@ def ejecutar_comando_fuente(self, path, entrega): #{{{ else: options['preexec_fn'].close_stdin = True a_guardar = set(self.archivos_a_guardar) - if self.archivos_a_comparar: - zip_a_comparar = ZipFile(StringIO(self.archivos_a_comparar), 'r') - a_comparar = set(zip_a_comparar.namelist()) - else: - zip_a_comparar = None - a_comparar = frozenset() + zip_a_comparar = Multizip(self.archivos_a_comparar) + a_comparar = set(zip_a_comparar.namelist()) a_usar = frozenset(a_guardar | a_comparar) if self.STDOUTERR in a_usar: options['stdout'] = file('%s.%s.stdouterr' % (basetmp, @@ -489,6 +503,7 @@ def ejecutar_comando_prueba(self, path, prueba): #{{{ # y setup/clean de test. log.debug(_(u'ComandoPrueba.ejecutar(path=%s, prueba=%s)'), path, prueba.shortrepr()) + caso_de_prueba = prueba.caso_de_prueba comando_ejecutado = prueba.add_comando_ejecutado(self) # TODO debería rodear solo la ejecución del comando basetmp = '/tmp/sercom.tester.prueba' # FIXME TODO /var/run/sercom? #{{{ Código que solo va en ComandoPrueba (setup de directorio) @@ -505,7 +520,7 @@ def ejecutar_comando_prueba(self, path, prueba): #{{{ os.seteuid(user_info.uid) log.debug(_(u'Usuario y grupo efectivos cambiados a %s:%s (%s:%s)'), user_info.user, user_info.group, user_info.uid, user_info.gid) - unzip(prueba.caso_de_prueba.archivos_entrada, path, # TODO try/except + unzip(caso_de_prueba.archivos_entrada, path, # TODO try/except {self.STDIN: '%s.%s.stdin' % (basetmp, comando_ejecutado.id)}) #}}} unzip(self.archivos_entrada, path, # TODO try/except @@ -521,13 +536,10 @@ def ejecutar_comando_prueba(self, path, prueba): #{{{ else: options['preexec_fn'].close_stdin = True a_guardar = set(self.archivos_a_guardar) - a_guardar |= set(prueba.caso_de_prueba.archivos_a_guardar) # FIXME Esto es propio de ComandoPrueba - if self.archivos_a_comparar: - zip_a_comparar = ZipFile(StringIO(self.archivos_a_comparar), 'r') - a_comparar = set(zip_a_comparar.namelist()) - else: - zip_a_comparar = None - a_comparar = frozenset() + a_guardar |= set(caso_de_prueba.archivos_a_guardar) # FIXME Esto es propio de ComandoPrueba + zip_a_comparar = Multizip(self.archivos_a_comparar, + caso_de_prueba.archivos_a_comparar) # FIXME Esto es propio de ComandoPrueba + a_comparar = set(zip_a_comparar.namelist()) a_usar = frozenset(a_guardar | a_comparar) if self.STDOUTERR in a_usar: options['stdout'] = file('%s.%s.stdouterr' % (basetmp, @@ -544,7 +556,7 @@ def ejecutar_comando_prueba(self, path, prueba): #{{{ comando_ejecutado.id), 'w') else: options['preexec_fn'].close_stderr = True - comando = self.comando + ' ' + prueba.caso_de_prueba.comando # FIXME Esto es propio de ComandoPrueba + comando = self.comando + ' ' + caso_de_prueba.comando # FIXME Esto es propio de ComandoPrueba log.debug(_(u'Ejecutando como root: %s'), comando) os.seteuid(0) # Dios! (para chroot) os.setegid(0) @@ -564,7 +576,7 @@ def ejecutar_comando_prueba(self, path, prueba): #{{{ comando_ejecutado.fin_tareas = datetime.now() # TODO debería rodear solo la ejecución del comando retorno = self.retorno if retorno == self.RET_PRUEBA: # FIXME Esto es propio de ComandoPrueba - retorno = prueba.caso_de_prueba.retorno # FIXME Esto es propio de ComandoPrueba + retorno = caso_de_prueba.retorno # FIXME Esto es propio de ComandoPrueba if retorno != self.RET_ANY: if retorno == self.RET_FAIL: if proc.returncode == 0: -- 2.43.0