def test_05_test_Visit_LogQueue(self):
queue = Queue(6);
for i in range(5):
queue.enqueue(i);
queue.accept(FilePrinter("./ThisIsATestFile.txt",""));
with open('./ThisIsATestFile.txt', 'r') as myfile:
result=myfile.read();
self.assertEqual(result, "\n|0|1|2|3|4|\n");
def testVisitLogQueue(self):
queue = Queue(10)
for i in range(5):
queue.enqueue(i)
queue.accept(ScreenPrinter(""))
self.assertEqual(self.capturedOutput.getvalue(), "\n\n|0|1|2|3|4|\n\n")
def testVisitLogQueue(self):
queue = Queue(10)
for i in range(5):
queue.enqueue(i)
queue.accept(FilePrinter("./test.txt", ""))
with open('./test.txt', 'r') as myfile:
result = myfile.read()
self.assertEqual(result, "\n|0|1|2|3|4|\n")
class FilePrinterTest(unittest.TestCase):
def setUp(self):
self.printer = FilePrinter("testFile.txt", "testName")
self.linkedList = LinkedList()
self.stack = Stack(3)
self.queue = Queue(3)
def tearDown(self):
try:
os.remove(self.printer.file_path)
except IOError:
pass
# Test __init__()
def test_a_Init(self):
self.assertEqual(self.printer.file_path, "testFile.txt")
self.assertEqual(self.printer.name, "testName")
def test_b_VisitLinkedList(self):
self.linkedList.append(1)
self.linkedList.append(2)
self.linkedList.append(3)
self.linkedList.accept(self.printer)
with open(self.printer.file_path, 'r') as testFile:
self.assertEqual(testFile.read(), self.printer.name + "\n(1,2,3)\n")
def test_c_VisitStack(self):
self.stack.push(1)
self.stack.push(2)
self.stack.push(3)
self.stack.accept(self.printer)
filler="\n-------\n"
with open(self.printer.file_path, 'r') as testFile:
self.assertEqual(testFile.read(), self.printer.name +filler+' 3 '+filler+' 2 '+filler+' 1 '+filler)
def test_d_VisitLogQueue(self):
self.queue.enqueue(1)
self.queue.enqueue(2)
self.queue.enqueue(3)
self.queue.accept(self.printer)
with open(self.printer.file_path, 'r') as testFile:
self.assertEqual(testFile.read(), self.printer.name + "\n|1|2|3|\n")
class ScreenPrinterTest(unittest.TestCase):
def setUp(self):
self.screenPrinter = ScreenPrinter("test")
self.linkedList = LinkedList()
self.stack = Stack(3)
self.queue = Queue(3)
# Test __init__()
def test_a_Init(self):
self.assertEqual(self.screenPrinter.name, "test")
def test_b_VisitLinkedList(self):
self.linkedList.append(1)
self.linkedList.append(2)
self.linkedList.append(3)
sys.stdout = io.StringIO()
self.linkedList.accept(ScreenPrinter(""))
self.assertEqual(sys.stdout.getvalue(), "\n\n(1,2,3)\n\n")
sys.stdout = sys.__stdout__
def test_c_VisitStack(self):
self.stack.push(1)
self.stack.push(2)
self.stack.push(3)
sys.stdout = io.StringIO()
self.stack.accept(ScreenPrinter(""))
filler = "\n-------\n"
self.assertEqual(
sys.stdout.getvalue(), '\n' + filler + ' 3 ' + filler +
' 2 ' + filler + ' 1 ' + filler + '\n')
sys.stdout = sys.__stdout__
def test_d_VisitLogQueue(self):
self.queue.enqueue(1)
self.queue.enqueue(2)
self.queue.enqueue(3)
sys.stdout = io.StringIO()
self.queue.accept(ScreenPrinter(""))
self.assertEqual(sys.stdout.getvalue(), "\n\n|1|2|3|\n\n")
sys.stdout = sys.__stdout__