def test_force_publish_module(self, mock_shell, mock_isfile, mock_listdir): module_path = '/tmp/module/pkg' output_dir = '/tmp/output' files = [ 'path_1%s' % builder.ARCHIVE_SUFFIX, 'path_2%s' % builder.ARCHIVE_SUFFIX, 'path_3', # not a module 'path_4%s' % builder.ARCHIVE_SUFFIX, # already exists ] mock_listdir.return_value = files mock_isfile.side_effect = [False, False, True] # test builder.publish_module(module_path, output_dir, True) # validation mock_shell.assert_any_call('mkdir -p %s' % output_dir) mock_shell.assert_any_call('cp %s %s' % (os.path.join(module_path, files[0]), output_dir)) mock_shell.assert_any_call('cp %s %s' % (os.path.join(module_path, files[1]), output_dir)) mock_shell.assert_any_call('cp %s %s' % (os.path.join(module_path, files[3]), output_dir)) self.assertEqual(mock_shell.call_count, 4)
def test_publish_module(self, mock_shell, mock_listdir): module_path = '/tmp/module/pkg' output_dir = '/tmp/output' files = [ 'path_1%s' % builder.ARCHIVE_SUFFIX, 'path_2%s' % builder.ARCHIVE_SUFFIX, 'path_3', # not a module ] mock_listdir.return_value = files # test builder.publish_module(module_path, output_dir) # validation mock_shell.assert_any_call('mkdir -p %s' % output_dir) mock_shell.assert_any_call('cp %s %s' % (os.path.join(module_path, files[0]), output_dir)) mock_shell.assert_any_call('cp %s %s' % (os.path.join(module_path, files[1]), output_dir)) self.assertEqual(mock_shell.call_count, 3)