def setUp(self):
self.working_dir = tempfile.mkdtemp(prefix='working_')
self.published_dir = tempfile.mkdtemp(prefix='published_')
self.master_dir = os.path.join(self.published_dir, 'master')
self.repo_id = 'publish-test-repo'
self.repo = Repository(self.repo_id, working_dir=self.working_dir)
self.conduit = Mock()
self.conduit = RepoPublishConduit(self.repo_id, 'test_distributor_id')
self.conduit.get_repo_scratchpad = Mock(return_value={})
self.config = PluginCallConfiguration(None, None)
self.publisher = publish_step.PublishStep(
"base-step", repo=self.repo, publish_conduit=self.conduit,
config=self.config, distributor_type='test_distributor_type')
def test_publish(self):
# just test that process_lifecycle got called, that is where the functionality lives now
step = publish_step.PublishStep("foo")
step.process_lifecycle = Mock()
step.publish()
self.assertTrue(step.process_lifecycle.called)
def test_clear_directory(self, mock_clear):
step = publish_step.PublishStep("foo")
step._clear_directory(self.working_dir, ['two'])
mock_clear.assert_called_once_with(self.working_dir, ['two'])
def test_create_symlink(self, mock_symlink):
step = publish_step.PublishStep("foo")
step._create_symlink('foo', 'bar')
mock_symlink.assert_called_once_with('foo', 'bar')
def test_get_distributor_type_from_parent(self):
step = publish_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_none(self):
step = publish_step.PublishStep('foo_step')
self.assertEquals(None, step.get_distributor_type())
def test_get_distributor_type(self):
step = publish_step.PublishStep('foo_step')
step.plugin_type = 'foo'
self.assertEquals('foo', step.get_distributor_type())
def test_clear_children(self):
step = publish_step.PublishStep("foo")
step.children = ['bar']
step.clear_children()
self.assertEquals(0, len(step.children))
