From e216e0a98a6409e00d1ee0625a11f7eecfe0ed01 Mon Sep 17 00:00:00 2001
From: Oliver Marks <oly@digitaloctave.com>
Date: Sun, 13 Jan 2019 21:28:34 +0000
Subject: [PATCH] Fix up tests, and make div valid.

---
 eorg/generate.py        | 3 ++-
 tests/test_documents.py | 4 ++--
 tests/test_html.py      | 4 ++--
 3 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/eorg/generate.py b/eorg/generate.py
index 703b9f9..3dab2e6 100644
--- a/eorg/generate.py
+++ b/eorg/generate.py
@@ -155,7 +155,8 @@ def handle_token(doc, item, root=False):
 
 
 def html(doc):
-    response = StringIO('<div class="org-body">')
+    response = StringIO()
+    response.write('<div class="org-body">')
     for item in doc:
         response.write(handle_token(doc, item, True))
     response.write('</div>')
diff --git a/tests/test_documents.py b/tests/test_documents.py
index 92226a3..830b62e 100755
--- a/tests/test_documents.py
+++ b/tests/test_documents.py
@@ -24,7 +24,7 @@ def test_html_output():
         print(htmlbody)
         assert (
             htmlbody
-            == """<h2> Header 1</h2>
+            == """<div class="org-body"><h2> Header 1</h2>
 <h3> Sub Header 1</h3>
 <p class="flow-text">body <code>text</code>
  over multiple <b>lines</b>
@@ -34,5 +34,5 @@ def test_html_output():
 </pre></div>
 </td></tr></table><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre>1</pre></div></td><td class="code"><div class="highlight"><pre><span></span><span class="p">(</span><span class="nv">test</span> <span class="nv">code</span><span class="p">)</span>
 </pre></div>
-</td></tr></table>"""
+</td></tr></table></div>"""
         )
diff --git a/tests/test_html.py b/tests/test_html.py
index 038afc7..cd83ebc 100644
--- a/tests/test_html.py
+++ b/tests/test_html.py
@@ -7,7 +7,7 @@ from eorg.generate import html
 
 
 def test_bullet_block():
-    expected = """<ul class="browser-default"><li class="collection-item">Bullet 1</li><li class="collection-item">Bullet 2</li></ul>"""
+    expected = """<div class="org-body"><ul class="browser-default"><li class="collection-item">Bullet 1</li><li class="collection-item">Bullet 2</li></ul></div>"""
     result = html(parse(snippets.bullet_plus_snippet).doc)
     assert result.read() == expected
 
@@ -33,4 +33,4 @@ def test_render_results():
     assert tags[1].value[1].value == expected[1].value
 
     htmlbody = html(tags).read()
-    assert htmlbody == '<img style="margin:auto;" src="test.png" alt="" />'
+    assert htmlbody == '<div class="org-body"><img style="margin:auto;" src="test.png" alt="" /></div>'