55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
import unittest
|
|
|
|
import mitogen.core
|
|
|
|
from mitogen.core import next
|
|
|
|
|
|
class IterSplitTest(unittest.TestCase):
|
|
func = staticmethod(mitogen.core.iter_split)
|
|
|
|
def test_empty_buffer(self):
|
|
lst = []
|
|
trailer, cont = self.func(buf='', delim='\n', func=lst.append)
|
|
self.assertTrue(cont)
|
|
self.assertEqual('', trailer)
|
|
self.assertEqual([], lst)
|
|
|
|
def test_empty_line(self):
|
|
lst = []
|
|
trailer, cont = self.func(buf='\n', delim='\n', func=lst.append)
|
|
self.assertTrue(cont)
|
|
self.assertEqual('', trailer)
|
|
self.assertEqual([''], lst)
|
|
|
|
def test_one_line(self):
|
|
buf = 'xxxx\n'
|
|
lst = []
|
|
trailer, cont = self.func(buf=buf, delim='\n', func=lst.append)
|
|
self.assertTrue(cont)
|
|
self.assertEqual('', trailer)
|
|
self.assertEqual(lst, ['xxxx'])
|
|
|
|
def test_one_incomplete(self):
|
|
buf = 'xxxx\nyy'
|
|
lst = []
|
|
trailer, cont = self.func(buf=buf, delim='\n', func=lst.append)
|
|
self.assertTrue(cont)
|
|
self.assertEqual('yy', trailer)
|
|
self.assertEqual(lst, ['xxxx'])
|
|
|
|
def test_returns_false_immediately(self):
|
|
buf = 'xxxx\nyy'
|
|
func = lambda buf: False
|
|
trailer, cont = self.func(buf=buf, delim='\n', func=func)
|
|
self.assertFalse(cont)
|
|
self.assertEqual('yy', trailer)
|
|
|
|
def test_returns_false_second_call(self):
|
|
buf = 'xxxx\nyy\nzz'
|
|
it = iter([True, False])
|
|
func = lambda buf: next(it)
|
|
trailer, cont = self.func(buf=buf, delim='\n', func=func)
|
|
self.assertFalse(cont)
|
|
self.assertEqual('zz', trailer)
|