def test_get_distributor_type_none(self): step = PublishStep('foo_step') self.assertEquals(None, step.get_distributor_type())
def test_get_distributor_type_from_parent(self): step = PublishStep('foo_step') step.conduit = 'foo' step.parent = Mock() step.parent.get_plugin_type.return_value = 'foo' self.assertEquals('foo', step.get_distributor_type())
def test_get_distributor_type(self): step = PublishStep('foo_step') step.plugin_type = 'foo' self.assertEquals('foo', step.get_distributor_type())