pyyaml / tests / test_reader.py

import test_appliance
from yaml.reader import Reader, ReaderError

import codecs

class TestReaderErrors(test_appliance.TestAppliance):

    def _testReaderUnicodeErrors(self, test_name, stream_filename):
        for encoding in ['utf-8', 'utf-16-le', 'utf-16-be']:
            try:
                data = unicode(file(stream_filename, 'rb').read(), encoding)
                break
            except:
                pass
        else:
            return
        #self._load(data)
        self.failUnlessRaises(ReaderError,
                lambda: self._load(data))
        #self._load(codecs.open(stream_filename, encoding=encoding))
        self.failUnlessRaises(ReaderError,
                lambda: self._load(codecs.open(stream_filename, encoding=encoding)))

    def _testReaderStringErrors(self, test_name, stream_filename):
        data = file(stream_filename, 'rb').read()
        #self._load(data)
        self.failUnlessRaises(ReaderError, lambda: self._load(data))

    def _testReaderFileErrors(self, test_name, stream_filename):
        data = file(stream_filename, 'rb')
        #self._load(data)
        self.failUnlessRaises(ReaderError, lambda: self._load(data))

    def _load(self, data):
        stream = Reader(data)
        while stream.peek() != u'\0':
            stream.forward()

TestReaderErrors.add_tests('testReaderUnicodeErrors', '.stream-error')
TestReaderErrors.add_tests('testReaderStringErrors', '.stream-error')
TestReaderErrors.add_tests('testReaderFileErrors', '.stream-error')
Tip: Filter by directory path e.g. /media app.js to search for public/media/app.js.
Tip: Use camelCasing e.g. ProjME to search for ProjectModifiedEvent.java.
Tip: Filter by extension type e.g. /repo .js to search for all .js files in the /repo directory.
Tip: Separate your search with spaces e.g. /ssh pom.xml to search for src/ssh/pom.xml.
Tip: Use ↑ and ↓ arrow keys to navigate and return to view the file.
Tip: You can also navigate files with Ctrl+j (next) and Ctrl+k (previous) and view the file with Ctrl+o.
Tip: You can also navigate files with Alt+j (next) and Alt+k (previous) and view the file with Alt+o.