def test__publish_data_no_granules(self, log_mock): publisher = Mock() granule1 = Mock() granule2 = Mock() granule3 = Mock() data_generator = [granule1, granule2, granule3] BaseDataHandler._publish_data(publisher=publisher, data_generator=data_generator) self.assertEqual(publisher.publish.call_count, 0) self.assertEqual(log_mock.warn.call_count, 3)
def test__publish_data_with_granules(self): publisher = Mock() granule1 = Mock(spec=Granule) granule2 = Mock(spec=Granule) granule3 = Mock(spec=Granule) data_generator = [granule1, granule2, granule3] BaseDataHandler._publish_data(publisher=publisher, data_generator=data_generator) expected = [call(granule1), call(granule2), call(granule3)] self.assertEqual(publisher.publish.call_args_list, expected)
def test__publish_data_no_granules(self, log_mock): publisher = Mock() granule1 = Mock() granule2 = Mock() granule3 = Mock() data_generator = [granule1, granule2, granule3] BaseDataHandler._publish_data(publisher=publisher, data_generator=data_generator) self.assertEqual(publisher.publish.call_count, 0) self.assertEqual(log_mock.warn.call_count, 3)
def test__publish_data_with_granules(self): publisher = Mock() granule1 = Mock(spec=Granule) granule2 = Mock(spec=Granule) granule3 = Mock(spec=Granule) data_generator = [granule1, granule2, granule3] BaseDataHandler._publish_data(publisher=publisher, data_generator=data_generator) expected = [call(granule1), call(granule2), call(granule3)] self.assertEqual(publisher.publish.call_args_list, expected)