Current File : /opt/cloudlinux/venv/lib/python3.11/site-packages/testfixtures/tests//test_sybil.py
from textwrap import dedent
from unittest import TestCase
from testfixtures.mock import Mock
from sybil.document import Document
from testfixtures import compare, Comparison as C, TempDirectory
from testfixtures.sybil import FileParser, FileBlock
class TestFileParser(TestCase):
def check_document(self, text, expected):
d = Document(dedent(text), path='/dev/null')
compare(
list(r.parsed for r in FileParser('td')(d)),
expected=expected
)
def test_multiple_files(self):
self.check_document(
text="""
.. topic:: file.txt
:class: write-file
line 1
line 2
line 3
.. topic:: file2.txt
:class: read-file
line 4
line 5
line 6
""",
expected = [
C(FileBlock,
path='file.txt',
content="line 1\n\nline 2\nline 3\n",
action='write'),
C(FileBlock,
path='file2.txt',
content='line 4\n\nline 5\nline 6\n',
action='read'),
])
def test_ignore_literal_blocking(self):
self.check_document(
text="""
.. topic:: file.txt
:class: write-file
::
line 1
line 2
line 3
""",
expected=[
C(FileBlock,
path='file.txt',
content="line 1\n\nline 2\nline 3\n",
action='write'),
])
def test_file_followed_by_text(self):
self.check_document(
text="""
.. topic:: file.txt
:class: write-file
print("hello")
out = 'there'
foo = 'bar'
This is just some normal text!
""",
expected=[
C(FileBlock,
path='file.txt',
content='print("hello")'
'\nout = \'there\'\n\nfoo = \'bar\'\n',
action='write'),
])
def test_red_herring(self):
self.check_document(
text="""
.. topic:: file.txt
:class: not-a-file
print "hello"
out = 'there'
""",
expected=[]
)
def test_no_class(self):
self.check_document(
text="""
.. topic:: file.txt
print "hello"
out = 'there'
""",
expected=[]
)
def check_evaluate(self, dir, block, expected):
parser = FileParser('td')
compare(expected, actual=parser.evaluate(Mock(
parsed=block,
namespace={'td': dir},
path='/the/file',
line=42,
)))
def test_evaluate_read_same(self):
with TempDirectory() as dir:
dir.write('foo', b'content')
self.check_evaluate(
dir,
FileBlock('foo', 'content', 'read'),
expected=None
)
def test_evaluate_read_difference(self):
with TempDirectory() as dir:
dir.write('foo', b'actual')
self.check_evaluate(
dir,
FileBlock('foo', 'expected', 'read'),
expected=(
"--- File '/the/file', line 42:\n"
"+++ Reading from \"{}/foo\":\n"
"@@ -1 +1 @@\n"
"-expected\n"
"+actual"
).format(dir.path)
)
def test_evaluate_write(self):
with TempDirectory() as dir:
self.check_evaluate(
dir,
FileBlock('foo', 'content', 'write'),
expected=None
)
dir.compare(['foo'])
compare(dir.read('foo', 'ascii'), 'content')
