from __future__ import absolute_import, print_function

import os
import sys

# make it runnable directly without run-tests.py
sys.path[0:0] = [os.path.join(os.path.dirname(__file__), '..')]

if not sys.platform.startswith('linux'):
    sys.stderr.write('skipped: linux required\n')
    sys.exit(80)

from hgext3rd import patchrmdir
from mercurial import util

patchrmdir.uisetup(None)

testtmp = os.environ['TESTTMP']

d1 = os.path.join(testtmp, 'd1')
d2 = os.path.join(testtmp, 'd1', 'd2')

os.mkdir(d1)
os.mkdir(d2)

def tryfunc(func):
    try:
        func()
    except Exception as ex:
        # normalize the error message across platforms
        if 'Non-empty directory' not in str(ex):
            ex = '*'
        else:
            ex = 'Non-empty directory'
        print('  error: %s' % ex)
    else:
        print('  success')

print('rmdir d1 - should fail with ENOTEMPTY')
tryfunc(lambda: os.rmdir(d1))

print('rmdir d1/d2 - should succeed')
tryfunc(lambda: os.rmdir(d2))

open(d2, 'w').close()

print('rmdir d1 - should fail with ENOTEMPTY')
tryfunc(lambda: os.rmdir(d1))

os.unlink(d2)

print('rmdir d1 - should succeed')
tryfunc(lambda: os.rmdir(d1))

print('rmdir d1 - should fail with ENOENT')
tryfunc(lambda: os.rmdir(d1))

os.mkdir(d1)
os.mkdir(d2)

print('removedirs d2 (and d1) - should succeed')
tryfunc(lambda: util.removedirs(d2))

print('removedirs d1 - should fail with ENOENT')
tryfunc(lambda: util.removedirs(d1))