class TestJSONArrayFileContext(unittest.TestCase): def setUp(self): self.working_directory = tempfile.mkdtemp() self.file_name = os.path.join(self.working_directory, 'foo') self.context = JSONArrayFileContext(self.file_name) self.context.metadata_file_handle = Mock() def tearDown(self): shutil.rmtree(self.working_directory) def test_write_file_header(self): self.context._write_file_header() self.context.metadata_file_handle.write.assert_called_once_with('[') def test_write_file_footer(self): self.context._write_file_footer() self.context.metadata_file_handle.write.assert_called_once_with(']') def test_add_unit_metadata(self): self.context.add_unit_metadata('foo') self.assertEquals(self.context.metadata_file_handle.write.call_count, 0) self.context.add_unit_metadata('bar') self.context.metadata_file_handle.write.assert_called_once_with(',')
def setUp(self): self.working_directory = tempfile.mkdtemp() self.file_name = os.path.join(self.working_directory, 'foo') self.context = JSONArrayFileContext(self.file_name) self.context.metadata_file_handle = Mock()