test_issue28.py

import unittest
import sandbox
import os.path
import os
from io import StringIO
import sys
from assignment import Assignment
from command import Command


OUTFILE = sandbox.dir('issue28.txt')


class test_issue28(unittest.TestCase):

    def tearDown(self):
        if os.path.isfile(OUTFILE):
            os.remove(OUTFILE)

    def test_missingFile_keepGoing(self):
        out = self._get_stdout_as_StringIO()
        try:
            collector = Collector(OUTFILE)
            Assignment(sandbox.dir('issue28.json')).accept(collector.visit)
            self.assertTrue(len(collector.get_lines()) == 5)
            output = out.getvalue().strip()
            self.assertRegex(output, 'Not found: .*student2/file1.txt',
                    msg=output)
        finally:
            self._restore_stdout()

    def _get_stdout_as_StringIO(self):
        self._saved_stdout = sys.stdout
        out = StringIO()
        sys.stdout = out
        return out

    def _restore_stdout(self):
        sys.stdout = self._saved_stdout


class Collector(object):
    def __init__(self, file_):
        self._file = file_
        self.ls = Command('ls "{ins}" >> ' + file_)

    def visit(self, submission_directory, files_to_collect):
        for file in files_to_collect:
            self.ls(file)

    def get_lines(self):
        lines = []
        with open(self._file) as outfile:
            lines = outfile.readlines()
        return lines



if __name__ == '__main__':
    unittest.main()