def test_finalize_error_on_footer(self, mock_logger): path = os.path.join(self.metadata_file_dir, 'test.xml') context = MetadataFileContext(path) context._write_file_footer = Mock(side_effect=Exception('foo')) context._open_metadata_file_handle() context._write_file_header() context.finalize() self.assertTrue(mock_logger.called)
def test_finalize_error_on_close(self, mock_logger): path = os.path.join(self.metadata_file_dir, 'test.xml') context = MetadataFileContext(path) context._close_metadata_file_handle = Mock(side_effect=Exception('foo')) context._open_metadata_file_handle() context._write_file_header() context.finalize() self.assertTrue(mock_logger.called)
def test_finalize_with_valid_checksum_type(self): path = os.path.join(self.metadata_file_dir, 'test.xml') checksum_type = 'sha1' context = MetadataFileContext(path, checksum_type) context._open_metadata_file_handle() context._write_file_header() context.finalize() expected_metadata_file_name = context.checksum + '-' + 'test.xml' expected_metadata_file_path = os.path.join(self.metadata_file_dir, expected_metadata_file_name) self.assertEquals(expected_metadata_file_path, context.metadata_file_path)
def test_open_handle_gzip(self): path = os.path.join(self.metadata_file_dir, 'test.xml.gz') context = MetadataFileContext(path) context._open_metadata_file_handle() self.assertTrue(os.path.exists(path)) context._write_file_header() context._close_metadata_file_handle() try: h = gzip.open(path) except Exception, e: self.fail(e.message)
def test_initialize_double_call(self): path = os.path.join(self.metadata_file_dir, 'test.xml') context = MetadataFileContext(path) context.initialize() context._write_file_header = Mock() context.initialize() self.assertEquals(0, context._write_file_header.call_count) context.finalize()
def test_open_handle_gzip(self): path = os.path.join(self.metadata_file_dir, 'test.xml.gz') context = MetadataFileContext(path) context._open_metadata_file_handle() self.assertTrue(os.path.exists(path)) context._write_file_header() context._close_metadata_file_handle() try: h = gzip.open(path) except Exception, e: self.fail(e.message)
def test_initialize_double_call(self): path = os.path.join(self.metadata_file_dir, 'test.xml') context = MetadataFileContext(path) context.initialize() context._write_file_header = Mock() context.initialize() self.assertEquals(0, context._write_file_header.call_count) context.finalize()
def test_initialize(self): path = os.path.join(self.metadata_file_dir, 'foo', 'header.xml') context = MetadataFileContext(path) context._write_file_header = Mock() context.initialize() context._write_file_header.assert_called_once_with() self.assertTrue(os.path.exists(path)) with open(path) as h: content = h.read() expected_content = '' self.assertEqual(content, expected_content)
def test_initialize(self): path = os.path.join(self.metadata_file_dir, 'foo', 'header.xml') context = MetadataFileContext(path) context._write_file_header = Mock() context.initialize() context._write_file_header.assert_called_once_with() self.assertTrue(os.path.exists(path)) with open(path) as h: content = h.read() expected_content = '' self.assertEqual(content, expected_content)