sapling/eden/integration/unlink_test.py
Andres Suarez fbdb46f5cb Tidy up license headers
Reviewed By: chadaustin

Differential Revision: D17872966

fbshipit-source-id: cd60a364a2146f0dadbeca693b1d4a5d7c97ff63
2019-10-11 05:28:23 -07:00

117 lines
3.7 KiB
Python

#!/usr/bin/env python3
# Copyright (c) Facebook, Inc. and its affiliates.
#
# This software may be used and distributed according to the terms of the
# GNU General Public License version 2.
import errno
import os
import shutil
from .lib import testcase
@testcase.eden_repo_test
class UnlinkTest(testcase.EdenRepoTest):
def populate_repo(self) -> None:
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_unlink(self) -> None:
filename = os.path.join(self.mount, "hello")
# This file is part of the git repo
with open(filename, "r") as f:
self.assertEqual("hola\n", f.read())
# Removing should succeed
os.unlink(filename)
with self.assertRaises(OSError) as context:
os.lstat(filename)
self.assertEqual(
context.exception.errno,
errno.ENOENT,
msg="lstat on a removed file raises ENOENT",
)
def test_unlink_bogus_file(self) -> None:
with self.assertRaises(OSError) as context:
os.unlink(os.path.join(self.mount, "this-file-does-not-exist"))
self.assertEqual(
context.exception.errno,
errno.ENOENT,
msg="unlink raises ENOENT for nonexistent file",
)
def test_unlink_dir(self) -> None:
adir = os.path.join(self.mount, "adir")
with self.assertRaises(OSError) as context:
os.unlink(adir)
self.assertEqual(
context.exception.errno, errno.EISDIR, msg="unlink on a dir raises EISDIR"
)
def test_unlink_empty_dir(self) -> None:
adir = os.path.join(self.mount, "an-empty-dir")
os.mkdir(adir)
with self.assertRaises(OSError) as context:
os.unlink(adir)
self.assertEqual(
context.exception.errno,
errno.EISDIR,
msg="unlink on an empty dir raises EISDIR",
)
def test_rmdir_file(self) -> None:
filename = os.path.join(self.mount, "hello")
with self.assertRaises(OSError) as context:
os.rmdir(filename)
self.assertEqual(
context.exception.errno, errno.ENOTDIR, msg="rmdir on a file raises ENOTDIR"
)
def test_rmdir(self) -> None:
adir = os.path.join(self.mount, "adir")
with self.assertRaises(OSError) as context:
os.rmdir(adir)
self.assertEqual(
context.exception.errno,
errno.ENOTEMPTY,
msg="rmdir on a non-empty dir raises ENOTEMPTY",
)
shutil.rmtree(adir)
with self.assertRaises(OSError) as context:
os.lstat(adir)
self.assertEqual(
context.exception.errno,
errno.ENOENT,
msg="lstat on a removed dir raises ENOENT",
)
def test_rmdir_overlay(self) -> None:
# Ensure that removing dirs works with things we make in the overlay
deep_root = os.path.join(self.mount, "buck-out")
deep_name = os.path.join(deep_root, "foo", "bar", "baz")
os.makedirs(deep_name)
with self.assertRaises(OSError) as context:
os.rmdir(deep_root)
self.assertEqual(
context.exception.errno,
errno.ENOTEMPTY,
msg="rmdir on a non-empty dir raises ENOTEMPTY",
)
shutil.rmtree(deep_root)
with self.assertRaises(OSError) as context:
os.lstat(deep_root)
self.assertEqual(
context.exception.errno,
errno.ENOENT,
msg="lstat on a removed dir raises ENOENT",
)