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')