mirror of
https://github.com/thegeeklab/ansible-later.git
synced 2024-11-30 00:30:35 +00:00
164 lines
4.3 KiB
Python
164 lines
4.3 KiB
Python
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
|
|
|
|
.. code-block:: python
|
|
|
|
print "hello"
|
|
out = 'there'
|
|
|
|
foo = 'bar'
|
|
|
|
This is just some normal text!
|
|
""",
|
|
expected=[
|
|
C(FileBlock,
|
|
path='file.txt',
|
|
content='.. code-block:: python\n\nprint "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')
|