def test_persist_meta(self, rdt): self.worker.get_dataset = Mock() self.worker.get_dataset.return_value = 'dataset_id' self.worker.persist = Mock() mock_rdt = DotDict() mock_rdt.time = np.arange(10) rdt.load_from_granule = Mock() rdt.load_from_granule.return_value = mock_rdt correct_val = { 'stream_id' : 'stream_id', 'dataset_id' : 'dataset_id', 'persisted_sha1' : 'dataset_id', 'encoding_type' : 'coverage', 'ts_create' : '0' } self.worker.persist_meta('stream_id',{}) self.worker.persist.assert_called_once_with(correct_val)