def test_literal(self): source = TokenSource([(b'abc {7}', b'foo bar'), b')']) tokens = iter(source) self.assertEqual(next(tokens), b'abc') self.assertEqual(next(tokens), b'{7}') self.assertEqual(source.current_literal, b'foo bar') self.assertEqual(next(tokens), b')') self.assertRaises(StopIteration, lambda: next(tokens))
def fake_get_line(): count = six.next(counter) if count == 0: return '* 1 EXISTS' elif count == 1: return '* 0 EXPUNGE' else: raise socket.timeout
def fake_get_response(): count = six.next(counter) if count == 0: return '* 99 EXISTS' client._imap.tagged_commands[sentinel.tag] = ('OK', ['Idle done'])
def fake_get_line(): count = six.next(counter) if count == 0: return b'* 99 EXISTS' else: raise socket.timeout