Changeset - fcc676c6bf3b
Parent rev.
Child rev.
[Not reviewed]
beta
0 1 0
Marcin Kuzminski - 14 years ago 2011-09-29 23:14:31
marcin@python-works.com
updated test_hg_operation script
1 file changed with 66 insertions and 24 deletions:
rhodecode/tests/test_hg_operations.py
66
24
0 comments (0 inline, 0 general)
rhodecode/tests/test_hg_operations.py
➞
Show inline comments
 
@@ -2,26 +2,29 @@
 
"""
 
    rhodecode.tests.test_hg_operations
 
    ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 

			




	
	
	
		
 
    Test suite for making push/pull operations

			




	
	
	
		
 

			




	
	
	
		
 
    :created_on: Dec 30, 2010

			




	
	
	
		
 
    :copyright: (c) 2010 by marcink.

			




	
	
	
		
 
    :license: LICENSE_NAME, see LICENSE_FILE for more details.

			




	
	
	
		
 
"""

			




	
	
	
		
 

			




	
	
	
		
 
import os

			




	
	
	
		
 
import time

			




	
	
	
		
 
import sys

			




	
	
	
		
 
import shutil

			




	
	
	
		
 
import logging

			




	
	
	
		
 

			




	
	
	
		
 
from os.path import join as jn

			




	
	
	
		
 
from os.path import dirname as dn

			




	
	
	
		
 

			




	
	
	
		
 
from tempfile import _RandomNameSequence

			




	
	
	
		
 
from subprocess import Popen, PIPE

			




	
	
	
		
 

			




	
	
	
		
 
from paste.deploy import appconfig

			




	
	
	
		
 
from pylons import config

			




	
	
	
		
 
from sqlalchemy import engine_from_config

			




	
	
	
		
 

			




	
	
	
		
 
from rhodecode.lib.utils import add_cache

			




	
	
	
		
 
from rhodecode.model import init_model

			




	
	
		
 
@@ -32,51 +35,66 @@ from rhodecode.lib.auth import get_crypt

			




	
	
	
		
 
from rhodecode.tests import TESTS_TMP_PATH, NEW_HG_REPO, HG_REPO

			




	
	
	
		
 
from rhodecode.config.environment import load_environment

			




	
	
	
		
 

			




	
	
	
		
 
rel_path = dn(dn(dn(os.path.abspath(__file__))))

			




	
	
	
		
 
conf = appconfig('config:development.ini', relative_to=rel_path)

			




	
	
	
		
 
load_environment(conf.global_conf, conf.local_conf)

			




	
	
	
		
 

			




	
	
	
		
 
add_cache(conf)

			




	
	
	
		
 

			




	
	
	
		
 
USER = 'test_admin'

			




	
	
	
		
 
PASS = 'test12'

			




	
	
	
		
 
HOST = '127.0.0.1:5000'

			




	
	
	
		
 
DEBUG = True

			




	
	
	
		
 
DEBUG = bool(int(sys.argv[1]))

			




	
	
	
		
 
print 'DEBUG:',DEBUG

			




	
	
	
		
 
log = logging.getLogger(__name__)

			




	
	
	
		
 

			




	
	
	
		
 

			




	
	
	
		
 
class Command(object):

			




	
	
	
		
 

			




	
	
	
		
 
    def __init__(self, cwd):

			




	
	
	
		
 
        self.cwd = cwd

			




	
	
	
		
 

			




	
	
	
		
 
    def execute(self, cmd, *args):

			




	
	
	
		
 
        """Runs command on the system with given ``args``.

			




	
	
	
		
 
        """

			




	
	
	
		
 

			




	
	
	
		
 
        command = cmd + ' ' + ' '.join(args)

			




	
	
	
		
 
        log.debug('Executing %s' % command)

			




	
	
	
		
 
        if DEBUG:

			




	
	
	
		
 
            print command

			




	
	
	
		
 
        p = Popen(command, shell=True, stdout=PIPE, stderr=PIPE, cwd=self.cwd)

			




	
	
	
		
 
        stdout, stderr = p.communicate()

			




	
	
	
		
 
        if DEBUG:

			




	
	
	
		
 
            print stdout, stderr

			




	
	
	
		
 
        return stdout, stderr

			




	
	
	
		
 

			




	
	
	
		
 

			




	
	
	
		
 
def test_wrapp(func):

			




	
	
	
		
 
    

			




	
	
	
		
 
    def __wrapp(*args,**kwargs):

			




	
	
	
		
 
        print '###%s###' %func.__name__

			




	
	
	
		
 
        try:

			




	
	
	
		
 
            res = func(*args,**kwargs)

			




	
	
	
		
 
        except:

			




	
	
	
		
 
            print '--%s failed--' % func.__name__

			




	
	
	
		
 
            return

			




	
	
	
		
 
        print 'ok'

			




	
	
	
		
 
        return res

			




	
	
	
		
 
    return __wrapp

			




	
	
	
		
 

			




	
	
	
		
 
def get_session():

			




	
	
	
		
 
    engine = engine_from_config(conf, 'sqlalchemy.db1.')

			




	
	
	
		
 
    init_model(engine)

			




	
	
	
		
 
    sa = meta.Session()

			




	
	
	
		
 
    sa = meta.Session

			




	
	
	
		
 
    return sa

			




	
	
	
		
 

			




	
	
	
		
 

			




	
	
	
		
 
def create_test_user(force=True):

			




	
	
	
		
 
    print 'creating test user'

			




	
	
	
		
 
    sa = get_session()

			




	
	
	
		
 

			




	
	
	
		
 
    user = sa.query(User).filter(User.username == USER).scalar()

			




	
	
	
		
 

			




	
	
	
		
 
    if force and user is not None:

			




	
	
	
		
 
        print 'removing current user'

			




	
	
	
		
 
        for repo in sa.query(Repository).filter(Repository.user == user).all():

			




	
	
		
 
@@ -120,138 +138,158 @@ def create_test_repo(force=True):

			




	
	
	
		
 
                     'clone_uri':'' }

			




	
	
	
		
 
        rm = RepoModel(sa)

			




	
	
	
		
 
        rm.base_path = '/home/hg'

			




	
	
	
		
 
        rm.create(form_data, user)

			




	
	
	
		
 

			




	
	
	
		
 

			




	
	
	
		
 
def set_anonymous_access(enable=True):

			




	
	
	
		
 
    sa = get_session()

			




	
	
	
		
 
    user = sa.query(User).filter(User.username == 'default').one()

			




	
	
	
		
 
    user.active = enable

			




	
	
	
		
 
    sa.add(user)

			




	
	
	
		
 
    sa.commit()

			




	
	
	
		
 
    sa.remove()

			




	
	
	
		
 
    

			




	
	
	
		
 
    print 'anonymous access is now:',enable

			




	
	
	
		
 

			




	
	
	
		
 

			




	
	
	
		
 
def get_anonymous_access():

			




	
	
	
		
 
    sa = get_session()

			




	
	
	
		
 
    return sa.query(User).filter(User.username == 'default').one().active

			




	
	
	
		
 
    obj1 = sa.query(User).filter(User.username == 'default').one()

			




	
	
	
		
 
    sa.expire(obj1)

			




	
	
	
		
 
    return obj1.active

			




	
	
	
		
 

			




	
	
	
		
 

			




	
	
	
		
 
#==============================================================================

			




	
	
	
		
 
# TESTS

			




	
	
	
		
 
#==============================================================================

			




	
	
	
		
 
def test_clone(no_errors=False):

			




	
	
	
		
 
@test_wrapp

			




	
	
	
		
 
def test_clone_with_credentials(no_errors=False):

			




	
	
	
		
 
    cwd = path = jn(TESTS_TMP_PATH, HG_REPO)

			




	
	
	
		
 

			




	
	
	
		
 
    try:

			




	
	
	
		
 
        shutil.rmtree(path, ignore_errors=True)

			




	
	
	
		
 
        os.makedirs(path)

			




	
	
	
		
 
        #print 'made dirs %s' % jn(path)

			




	
	
	
		
 
    except OSError:

			




	
	
	
		
 
        raise

			




	
	
	
		
 

			




	
	
	
		
 

			




	
	
	
		
 
    print 'checking if anonymous access is enabled'

			




	
	
	
		
 
    anonymous_access = get_anonymous_access()

			




	
	
	
		
 
    if anonymous_access:

			




	
	
	
		
 
        print 'enabled, disabling it '

			




	
	
	
		
 
        set_anonymous_access(enable=False)

			




	
	
	
		
 
        time.sleep(1)

			




	
	
	
		
 
        

			




	
	
	
		
 
    clone_url = 'http://%(user)s:%(pass)s@%(host)s/%(cloned_repo)s %(dest)s' % \

			




	
	
	
		
 
                  {'user':USER,

			




	
	
	
		
 
                   'pass':PASS,

			




	
	
	
		
 
                   'host':HOST,

			




	
	
	
		
 
                   'cloned_repo':HG_REPO,

			




	
	
	
		
 
                   'dest':path}

			




	
	
	
		
 

			




	
	
	
		
 
    stdout, stderr = Command(cwd).execute('hg clone', clone_url)

			




	
	
	
		
 

			




	
	
	
		
 
    if no_errors is False:

			




	
	
	
		
 
        assert """adding file changes""" in stdout, 'no messages about cloning'

			




	
	
	
		
 
        assert """abort""" not in stderr , 'got error from clone'

			




	
	
	
		
 

			




	
	
	
		
 

			




	
	
	
		
 

			




	
	
	
		
 
def test_clone_anonymous_ok():

			




	
	
	
		
 
@test_wrapp

			




	
	
	
		
 
def test_clone_anonymous():

			




	
	
	
		
 
    cwd = path = jn(TESTS_TMP_PATH, HG_REPO)

			




	
	
	
		
 

			




	
	
	
		
 
    try:

			




	
	
	
		
 
        shutil.rmtree(path, ignore_errors=True)

			




	
	
	
		
 
        os.makedirs(path)

			




	
	
	
		
 
        #print 'made dirs %s' % jn(path)

			




	
	
	
		
 
    except OSError:

			




	
	
	
		
 
        raise

			




	
	
	
		
 

			




	
	
	
		
 

			




	
	
	
		
 
    print 'checking if anonymous access is enabled'

			




	
	
	
		
 
    anonymous_access = get_anonymous_access()

			




	
	
	
		
 
    if not anonymous_access:

			




	
	
	
		
 
        print 'not enabled, enabling it '

			




	
	
	
		
 
        set_anonymous_access(enable=True)

			




	
	
	
		
 

			




	
	
	
		
 
        time.sleep(1)

			




	
	
	
		
 
        

			




	
	
	
		
 
    clone_url = 'http://%(host)s/%(cloned_repo)s %(dest)s' % \

			




	
	
	
		
 
                  {'user':USER,

			




	
	
	
		
 
                   'pass':PASS,

			




	
	
	
		
 
                   'host':HOST,

			




	
	
	
		
 
                   'cloned_repo':HG_REPO,

			




	
	
	
		
 
                   'dest':path}

			




	
	
	
		
 

			




	
	
	
		
 
    stdout, stderr = Command(cwd).execute('hg clone', clone_url)

			




	
	
	
		
 
    print stdout, stderr

			




	
	
	
		
 

			




	
	
	
		
 

			




	
	
	
		
 
    assert """adding file changes""" in stdout, 'no messages about cloning'

			




	
	
	
		
 
    assert """abort""" not in stderr , 'got error from clone'

			




	
	
	
		
 

			




	
	
	
		
 
    #disable if it was enabled

			




	
	
	
		
 
    if not anonymous_access:

			




	
	
	
		
 
        print 'disabling anonymous access'

			




	
	
	
		
 
        set_anonymous_access(enable=False)

			




	
	
	
		
 

			




	
	
	
		
 

			




	
	
	
		
 
@test_wrapp

			




	
	
	
		
 
def test_clone_wrong_credentials():

			




	
	
	
		
 
    cwd = path = jn(TESTS_TMP_PATH, HG_REPO)

			




	
	
	
		
 

			




	
	
	
		
 
    try:

			




	
	
	
		
 
        shutil.rmtree(path, ignore_errors=True)

			




	
	
	
		
 
        os.makedirs(path)

			




	
	
	
		
 
        #print 'made dirs %s' % jn(path)

			




	
	
	
		
 
    except OSError:

			




	
	
	
		
 
        raise

			




	
	
	
		
 

			




	
	
	
		
 

			




	
	
	
		
 
    print 'checking if anonymous access is enabled'

			




	
	
	
		
 
    anonymous_access = get_anonymous_access()

			




	
	
	
		
 
    if anonymous_access:

			




	
	
	
		
 
        print 'enabled, disabling it '

			




	
	
	
		
 
        set_anonymous_access(enable=False)

			




	
	
	
		
 
        

			




	
	
	
		
 
    clone_url = 'http://%(user)s:%(pass)s@%(host)s/%(cloned_repo)s %(dest)s' % \

			




	
	
	
		
 
                  {'user':USER + 'error',

			




	
	
	
		
 
                   'pass':PASS,

			




	
	
	
		
 
                   'host':HOST,

			




	
	
	
		
 
                   'cloned_repo':HG_REPO,

			




	
	
	
		
 
                   'dest':path}

			




	
	
	
		
 

			




	
	
	
		
 
    stdout, stderr = Command(cwd).execute('hg clone', clone_url)

			




	
	
	
		
 

			




	
	
	
		
 
    assert """abort: authorization failed""" in stderr , 'no error from wrong credentials'

			




	
	
	
		
 
    if not """abort: authorization failed"""  in stderr:

			




	
	
	
		
 
        raise Exception('Failure')    

			




	
	
	
		
 

			




	
	
	
		
 

			




	
	
	
		
 
@test_wrapp

			




	
	
	
		
 
def test_pull():

			




	
	
	
		
 
    pass

			




	
	
	
		
 

			




	
	
	
		
 
@test_wrapp

			




	
	
	
		
 
def test_push_modify_file(f_name='setup.py'):

			




	
	
	
		
 
    cwd = path = jn(TESTS_TMP_PATH, HG_REPO)

			




	
	
	
		
 
    modified_file = jn(TESTS_TMP_PATH, HG_REPO, f_name)

			




	
	
	
		
 
    for i in xrange(5):

			




	
	
	
		
 
        cmd = """echo 'added_line%s' >> %s""" % (i, modified_file)

			




	
	
	
		
 
        Command(cwd).execute(cmd)

			




	
	
	
		
 

			




	
	
	
		
 
        cmd = """hg ci -m 'changed file %s' %s """ % (i, modified_file)

			




	
	
	
		
 
        Command(cwd).execute(cmd)

			




	
	
	
		
 

			




	
	
	
		
 
    Command(cwd).execute('hg push %s' % jn(TESTS_TMP_PATH, HG_REPO))

			




	
	
	
		
 

			




	
	
	
		
 
@test_wrapp

			




	
	
	
		
 
def test_push_new_file(commits=15, with_clone=True):

			




	
	
	
		
 

			




	
	
	
		
 
    if with_clone:

			




	
	
	
		
 
        test_clone(no_errors=True)

			




	
	
	
		
 
        test_clone_with_credentials(no_errors=True)

			




	
	
	
		
 

			




	
	
	
		
 
    cwd = path = jn(TESTS_TMP_PATH, HG_REPO)

			




	
	
	
		
 
    added_file = jn(path, '%ssetupążźć.py' % _RandomNameSequence().next())

			




	
	
	
		
 

			




	
	
	
		
 
    Command(cwd).execute('touch %s' % added_file)

			




	
	
	
		
 

			




	
	
	
		
 
    Command(cwd).execute('hg add %s' % added_file)

			




	
	
	
		
 

			




	
	
	
		
 
    for i in xrange(commits):

			




	
	
	
		
 
        cmd = """echo 'added_line%s' >> %s""" % (i, added_file)

			




	
	
	
		
 
        Command(cwd).execute(cmd)

			




	
	
	
		
 

			




	
	
		
 
@@ -260,43 +298,45 @@ def test_push_new_file(commits=15, with_

			




	
	
	
		
 
                                added_file)

			




	
	
	
		
 
        Command(cwd).execute(cmd)

			




	
	
	
		
 

			




	
	
	
		
 
    push_url = 'http://%(user)s:%(pass)s@%(host)s/%(cloned_repo)s' % \

			




	
	
	
		
 
                  {'user':USER,

			




	
	
	
		
 
                   'pass':PASS,

			




	
	
	
		
 
                   'host':HOST,

			




	
	
	
		
 
                   'cloned_repo':HG_REPO,

			




	
	
	
		
 
                   'dest':jn(TESTS_TMP_PATH, HG_REPO)}

			




	
	
	
		
 

			




	
	
	
		
 
    Command(cwd).execute('hg push --verbose --debug %s' % push_url)

			




	
	
	
		
 

			




	
	
	
		
 
@test_wrapp

			




	
	
	
		
 
def test_push_wrong_credentials():

			




	
	
	
		
 
    cwd = path = jn(TESTS_TMP_PATH, HG_REPO)

			




	
	
	
		
 
    clone_url = 'http://%(user)s:%(pass)s@%(host)s/%(cloned_repo)s' % \

			




	
	
	
		
 
                  {'user':USER + 'xxx',

			




	
	
	
		
 
                   'pass':PASS,

			




	
	
	
		
 
                   'host':HOST,

			




	
	
	
		
 
                   'cloned_repo':HG_REPO,

			




	
	
	
		
 
                   'dest':jn(TESTS_TMP_PATH, HG_REPO)}

			




	
	
	
		
 

			




	
	
	
		
 
    modified_file = jn(TESTS_TMP_PATH, HG_REPO, 'setup.py')

			




	
	
	
		
 
    for i in xrange(5):

			




	
	
	
		
 
        cmd = """echo 'added_line%s' >> %s""" % (i, modified_file)

			




	
	
	
		
 
        Command(cwd).execute(cmd)

			




	
	
	
		
 

			




	
	
	
		
 
        cmd = """hg ci -m 'commited %s' %s """ % (i, modified_file)

			




	
	
	
		
 
        Command(cwd).execute(cmd)

			




	
	
	
		
 

			




	
	
	
		
 
    Command(cwd).execute('hg push %s' % clone_url)

			




	
	
	
		
 

			




	
	
	
		
 
@test_wrapp

			




	
	
	
		
 
def test_push_wrong_path():

			




	
	
	
		
 
    cwd = path = jn(TESTS_TMP_PATH, HG_REPO)

			




	
	
	
		
 
    added_file = jn(path, 'somefile.py')

			




	
	
	
		
 

			




	
	
	
		
 
    try:

			




	
	
	
		
 
        shutil.rmtree(path, ignore_errors=True)

			




	
	
	
		
 
        os.makedirs(path)

			




	
	
	
		
 
        print 'made dirs %s' % jn(path)

			




	
	
	
		
 
    except OSError:

			




	
	
	
		
 
        raise

			




	
	
	
		
 

			




	
	
	
		
 
    Command(cwd).execute("""echo '' > %s""" % added_file)

			




	
	
		
 
@@ -309,29 +349,31 @@ def test_push_wrong_path():

			




	
	
	
		
 

			




	
	
	
		
 
        cmd = """hg ci -m 'commited new %s' %s """ % (i, added_file)

			




	
	
	
		
 
        Command(cwd).execute(cmd)

			




	
	
	
		
 

			




	
	
	
		
 
    clone_url = 'http://%(user)s:%(pass)s@%(host)s/%(cloned_repo)s' % \

			




	
	
	
		
 
                  {'user':USER,

			




	
	
	
		
 
                   'pass':PASS,

			




	
	
	
		
 
                   'host':HOST,

			




	
	
	
		
 
                   'cloned_repo':HG_REPO + '_error',

			




	
	
	
		
 
                   'dest':jn(TESTS_TMP_PATH, HG_REPO)}

			




	
	
	
		
 

			




	
	
	
		
 
    stdout, stderr = Command(cwd).execute('hg push %s' % clone_url)

			




	
	
	
		
 
    assert """abort: HTTP Error 403: Forbidden"""  in stderr

			




	
	
	
		
 
    if not """abort: HTTP Error 403: Forbidden"""  in stderr:

			




	
	
	
		
 
        raise Exception('Failure')

			




	
	
	
		
 

			




	
	
	
		
 

			




	
	
	
		
 
if __name__ == '__main__':

			




	
	
	
		
 
    create_test_user(force=False)

			




	
	
	
		
 
    create_test_repo()

			




	
	
	
		
 
    #test_push_modify_file()

			




	
	
	
		
 
    #test_clone()

			




	
	
	
		
 
    #test_clone_anonymous_ok()

			




	
	
	
		
 
    

			




	
	
	
		
 
#    test_push_modify_file()

			




	
	
	
		
 
    test_clone_with_credentials()

			




	
	
	
		
 
    test_clone_wrong_credentials()

			




	
	
	
		
 

			




	
	
	
		
 
    #test_clone_wrong_credentials()

			




	
	
	
		
 

			




	
	
	
		
 
    test_pull()

			




	
	
	
		
 
    test_push_new_file(commits=2, with_clone=True)

			




	
	
	
		
 

			




	
	
	
		
 
    #test_push_wrong_path()

			




	
	
	
		
 
    #test_push_wrong_credentials()

			




	
	
	
		
 
#

			




	
	
	
		
 
    test_push_wrong_path()

			




	
	
	
		
 
    

			




	
	
	
		
 
    test_clone_anonymous()

			




	
	
	
		
 
    test_push_wrong_credentials()

			




	
	
		
 
\ No newline at end of file

			




        

    



    


    



    



      

      





    
    0 comments (0 inline, 0 general)
    





    


  

  







  


    




    








    
        
    
    
        This site is powered by
            Kallithea 0.7.0,
        which is
        © 2010–2025 by various authors & licensed under GPLv3.