Proposal for solving issue #379

lib/yaml/emitter.py

@@ -202,7 +202,7 @@ def expect_document_start(self, first=False):
                     self.write_indent()
             self.state = self.expect_document_root
         elif isinstance(self.event, StreamEndEvent):
-            if self.open_ended:
+            if self.open_ended and self.event.explicit_end:
                 self.write_indicator('...', True)
                 self.write_indent()
             self.write_stream_end()

lib/yaml/events.py

@@ -40,7 +40,8 @@ def __init__(self, start_mark=None, end_mark=None, encoding=None):
         self.encoding = encoding
 
 class StreamEndEvent(Event):
-    pass
+    def __init__(self, explicit_end=None):
+        self.explicit_end = explicit_end
 
 class DocumentStartEvent(Event):
     def __init__(self, start_mark=None, end_mark=None,

lib/yaml/serializer.py

@@ -37,7 +37,7 @@ def close(self):
         if self.closed is None:
             raise SerializerError("serializer is not opened")
         elif not self.closed:
-            self.emit(StreamEndEvent())
+            self.emit(StreamEndEvent(explicit_end=self.use_explicit_end))
             self.closed = True
 
     #def __del__(self):

tests/legacy_tests/test_emitter.py

@@ -98,6 +98,14 @@ def test_emitter_events(events_filename, verbose=False):
     new_events = list(yaml.parse(output))
     _compare_events(events, new_events)
 
+def test_marks_on_primitives():
+    assert yaml.dump(42, explicit_start=False) == '42'
+    assert yaml.dump(42, explicit_start=True) == '--- 42'
+    assert yaml.dump(42, explicit_end=False) == '42'
+    assert yaml.dump(42, explicit_end=True) == '42\n...\n'
+    assert yaml.dump(42, explicit_start=False, explicit_end=False) == '42'
+    assert yaml.dump(42, explicit_start=True, explicit_end=True) == '--- 42\n...\n'
+
 if __name__ == '__main__':
     import test_appliance
     test_appliance.run(globals())