def test_write_writes_new_content_to_path(self):
s = Source()
tf = tempfile.NamedTemporaryFile()
s.get_updated_contents = lambda: 'abc\ndef'
s.path = tf.name
s.write()
with open(tf.name) as f:
self.assertEqual(f.read(), s.get_updated_contents())
def test_adds_final_newline_if_necessary(self):
s = Source()
s.update('new stuff')
self.assertEqual(s.get_updated_contents(), 'new stuff\n')
def test_update_with_empty_contents(self):
s = Source()
s.update('new stuff\n')
self.assertEqual(s.get_updated_contents(), 'new stuff\n')
def test_lines(self):
s = Source()
s.contents = 'abc\ndef'
self.assertEqual(s.lines, ['abc', 'def'])
