mirror of
https://github.com/facebook/sapling.git
synced 2024-10-11 09:17:30 +03:00
e07ef44b1c
Summary: This was an error that an end-user ran into. Previously, we did not fail gracefully and the user was faced with an intimidating stacktrace. Reviewed By: simpkins Differential Revision: D6195529 fbshipit-source-id: bde3c2a3e6f49457a4c6ac5c87103cf52cd227c2
214 lines
7.7 KiB
Python
214 lines
7.7 KiB
Python
#!/usr/bin/env python3
|
|
#
|
|
# Copyright (c) 2016-present, Facebook, Inc.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the BSD-style license found in the
|
|
# LICENSE file in the root directory of this source tree. An additional grant
|
|
# of patent rights can be found in the PATENTS file in the same directory.
|
|
|
|
import json
|
|
import os
|
|
from .lib import edenclient, gitrepo, hgrepo, testcase
|
|
|
|
|
|
@testcase.eden_repo_test
|
|
class RemountTest:
|
|
def populate_repo(self):
|
|
self.repo.write_file('hello', 'hola\n')
|
|
self.repo.write_file('adir/file', 'foo!\n')
|
|
self.repo.symlink('slink', 'hello')
|
|
self.repo.commit('Initial commit.')
|
|
|
|
def test_remount_basic(self):
|
|
# Mount multiple clients
|
|
for i in range(5):
|
|
self.eden.clone(self.repo_name, self.mount + '-' + str(i))
|
|
|
|
self.eden.shutdown()
|
|
self.eden.start()
|
|
|
|
# Verify that clients are remounted on startup
|
|
for i in range(5):
|
|
entries = sorted(os.listdir(self.mount + '-' + str(i)))
|
|
self.assertEqual(['.eden', 'adir', 'hello', 'slink'], entries)
|
|
|
|
# Verify that default repo created by EdenRepoTestBase is remounted
|
|
entries = sorted(os.listdir(self.mount))
|
|
self.assertEqual(['.eden', 'adir', 'hello', 'slink'], entries)
|
|
|
|
def test_git_and_hg(self):
|
|
# Create git and hg repositories for mounting
|
|
repo_names = {'git': 'git_repo', 'hg': 'hg_repo'}
|
|
git_repo = self.create_repo(repo_names['git'], gitrepo.GitRepository)
|
|
hg_repo = self.create_repo(repo_names['hg'], hgrepo.HgRepository)
|
|
|
|
git_repo.write_file('hello', 'hola\n')
|
|
git_repo.commit('Initial commit.')
|
|
|
|
hg_repo.write_file('hello', 'hola\n')
|
|
hg_repo.commit('Initial commit.')
|
|
|
|
self.eden.add_repository(repo_names['git'], git_repo.path)
|
|
self.eden.add_repository(repo_names['hg'], hg_repo.path)
|
|
|
|
# Mount git and hg clients
|
|
for name in repo_names.values():
|
|
for i in range(3):
|
|
self.eden.clone(name, os.path.join(self.mounts_dir,
|
|
name + '-' + str(i)))
|
|
|
|
self.eden.shutdown()
|
|
self.eden.start()
|
|
|
|
# Verify that clients are remounted on startup
|
|
for name in repo_names.values():
|
|
for i in range(3):
|
|
mount_path = os.path.join(self.mounts_dir, name + '-' + str(i))
|
|
entries = sorted(os.listdir(mount_path))
|
|
self.assertEqual(['.eden', 'hello'], entries)
|
|
|
|
hello = os.path.join(mount_path, 'hello')
|
|
with open(hello, 'r') as f:
|
|
self.assertEqual('hola\n', f.read())
|
|
|
|
def test_partial_unmount(self):
|
|
# Mount multiple clients
|
|
for i in range(5):
|
|
self.eden.clone(self.repo_name, self.mount + '-' + str(i))
|
|
|
|
# Unmount a client
|
|
self.eden.unmount(self.mount)
|
|
self.assertFalse(os.path.exists(self.mount))
|
|
|
|
self.eden.shutdown()
|
|
self.eden.start()
|
|
|
|
# Verify that clients that were still mounted at shutdown are remounted
|
|
for i in range(5):
|
|
entries = sorted(os.listdir(self.mount + '-' + str(i)))
|
|
self.assertEqual(['.eden', 'adir', 'hello', 'slink'], entries)
|
|
|
|
# Verify that unmounted client is not remounted
|
|
self.assertFalse(os.path.exists(self.mount))
|
|
|
|
def test_restart_twice(self):
|
|
# Mount multiple clients
|
|
for i in range(5):
|
|
self.eden.clone(self.repo_name, self.mount + '-' + str(i))
|
|
|
|
self.eden.shutdown()
|
|
self.eden.start()
|
|
|
|
# Unmount clients
|
|
self.eden.unmount(self.mount)
|
|
self.eden.unmount(self.mount + '-3')
|
|
|
|
self.eden.shutdown()
|
|
self.eden.start()
|
|
|
|
# Verify that clients that were still mounted at shutdown are remounted
|
|
for i in range(5):
|
|
if i == 3:
|
|
continue
|
|
entries = sorted(os.listdir(self.mount + '-' + str(i)))
|
|
self.assertEqual(['.eden', 'adir', 'hello', 'slink'], entries)
|
|
|
|
# Verify that unmounted clients are not remounted
|
|
self.assertFalse(os.path.exists(self.mount))
|
|
self.assertFalse(os.path.exists(self.mount + "3"))
|
|
|
|
def test_try_remount_existing_mount(self):
|
|
'''Verify trying to mount an existing mount prints a sensible error.'''
|
|
mount_destination = self.mount + '-0'
|
|
self.eden.clone(self.repo_name, mount_destination)
|
|
with self.assertRaises(edenclient.EdenCommandError) as context:
|
|
self.eden.run_cmd('mount', mount_destination)
|
|
self.assertEqual(
|
|
(b'ERROR: Mount point in use! %s is already mounted by Eden.\n'
|
|
% mount_destination.encode()),
|
|
context.exception.stderr)
|
|
self.assertEqual(1, context.exception.returncode)
|
|
|
|
def test_empty_config_json(self):
|
|
for i in range(5):
|
|
self.eden.clone(self.repo_name, self.mount + '-' + str(i))
|
|
|
|
# Clear the contents of config.json file
|
|
open(os.path.join(self.eden_dir, 'config.json'), 'w').close()
|
|
|
|
self.eden.shutdown()
|
|
self.eden.start()
|
|
|
|
# Verify that no clients are remounted. No errors should be thrown here
|
|
for i in range(5):
|
|
entries = sorted(os.listdir(self.mount + '-' + str(i)))
|
|
self.assertEqual([], entries)
|
|
|
|
entries = sorted(os.listdir(self.mount))
|
|
self.assertEqual([], entries)
|
|
|
|
def test_deleted_config_json(self):
|
|
for i in range(5):
|
|
self.eden.clone(self.repo_name, self.mount + '-' + str(i))
|
|
|
|
# Delete the config.json file
|
|
os.remove(os.path.join(self.eden_dir, 'config.json'))
|
|
|
|
self.eden.shutdown()
|
|
self.eden.start()
|
|
|
|
# Verify that no clients are remounted. No errors should be thrown here
|
|
for i in range(5):
|
|
entries = sorted(os.listdir(self.mount + '-' + str(i)))
|
|
self.assertEqual([], entries)
|
|
|
|
entries = sorted(os.listdir(self.mount))
|
|
self.assertEqual([], entries)
|
|
|
|
def test_incorrect_config_json(self):
|
|
for i in range(5):
|
|
self.eden.clone(self.repo_name, self.mount + '-' + str(i))
|
|
|
|
# Reload config.json file with incorrect data
|
|
config_data = {self.mounts_dir + '/incorrectdir1': 'jskdnbailreylbflhv',
|
|
self.mounts_dir + '/incorrectdir2': 'ndjnhaleruybfialus'}
|
|
|
|
with open(os.path.join(self.eden_dir, 'config.json'), 'w') as f:
|
|
json.dump(config_data, f, indent=2, sort_keys=True)
|
|
f.write('\n')
|
|
|
|
self.eden.shutdown()
|
|
self.eden.start()
|
|
|
|
# Verify that no clients are remounted. No errors should be thrown here
|
|
for i in range(5):
|
|
entries = sorted(os.listdir(self.mount + '-' + str(i)))
|
|
self.assertEqual([], entries)
|
|
|
|
entries = sorted(os.listdir(self.mount))
|
|
self.assertEqual([], entries)
|
|
|
|
# Incorrect mount paths from config.json should not have been created
|
|
for incorrect_mount in config_data:
|
|
self.assertFalse(os.path.isdir(incorrect_mount))
|
|
|
|
def test_bad_config_json(self):
|
|
for i in range(5):
|
|
self.eden.clone(self.repo_name, self.mount + '-' + str(i))
|
|
|
|
# Reload config.json file with random data
|
|
with open(os.path.join(self.eden_dir, 'config.json'), 'w') as f:
|
|
f.write('njfaeriurbvailuwrawikc\n')
|
|
|
|
self.eden.shutdown()
|
|
self.eden.start()
|
|
|
|
# Verify that no clients are remounted. No errors should be thrown here
|
|
for i in range(5):
|
|
entries = sorted(os.listdir(self.mount + '-' + str(i)))
|
|
self.assertEqual([], entries)
|
|
|
|
entries = sorted(os.listdir(self.mount))
|
|
self.assertEqual([], entries)
|