GitPython/test/test_base.py at main · gitpython-developers/GitPython · GitHub

Repository navigation

History

162 lines (135 loc) · 6.57 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

# This module is part of GitPython and is released under the

# 3-Clause BSD License: https://opensource.org/license/bsd-3-clause/

import gc

from io import BytesIO

import os

import os.path as osp

import sys

import tempfile

from unittest import skipIf

from unittest.mock import patch

from gitdb import OInfo

from git import Repo

from git.objects import Blob, Commit, TagObject, Tree

import git.objects.base as base

from git.objects.util import get_object_type_by_name

from git.util import HIDE_WINDOWS_FREEZE_ERRORS, hex_to_bin

from test.lib import TestBase as _TestBase, with_rw_and_rw_remote_repo, with_rw_repo

class TestBase(_TestBase):

def tearDown(self):

gc.collect()

type_tuples = (

("blob", "8741fc1d09d61f02ffd8cded15ff603eff1ec070", "blob.py"),

("tree", "3a6a5e3eeed3723c09f1ef0399f81ed6b8d82e79", "directory"),

("commit", "4251bd59fb8e11e40c40548cba38180a9536118c", None),

("tag", "e56a60e8e9cd333cfba0140a77cd12b0d9398f10", None),

)

def test_base_object(self):

# Test interface of base object classes.

types = (Blob, Tree, Commit, TagObject)

self.assertEqual(len(types), len(self.type_tuples))

s = set()

num_objs = 0

num_index_objs = 0

for obj_type, (typename, hexsha, path) in zip(types, self.type_tuples):

binsha = hex_to_bin(hexsha)

item = None

if path is None:

item = obj_type(self.rorepo, binsha)

else:

item = obj_type(self.rorepo, binsha, 0, path)

# END handle index objects

num_objs += 1

self.assertEqual(item.hexsha, hexsha)

self.assertEqual(item.type, typename)

assert item.size

self.assertEqual(item, item)

self.assertNotEqual(not item, item)

self.assertEqual(str(item), item.hexsha)

assert repr(item)

s.add(item)

if isinstance(item, base.IndexObject):

num_index_objs += 1

if hasattr(item, "path"): # Never runs here.

assert not item.path.startswith("/") # Must be relative.

assert isinstance(item.mode, int)

# END index object check

# Read from stream.

data_stream = item.data_stream

data = data_stream.read()

assert data

with tempfile.NamedTemporaryFile(suffix="test-stream", delete=False) as tmpfile:

self.assertEqual(item, item.stream_data(tmpfile))

tmpfile.seek(0)

self.assertEqual(tmpfile.read(), data)

# Remove the file this way, instead of with a context manager or "finally",

# so it is only removed on success, and we can inspect the file on failure.

os.remove(tmpfile.name)

for stored_type in (typename, typename.encode("ascii")):

with patch.object(self.rorepo.odb, "info", return_value=OInfo(binsha, stored_type, item.size)):

self.assertEqual(obj_type(self.rorepo, binsha).size, item.size)

for wrong_type in types:

if wrong_type is obj_type:

continue

invalid = wrong_type(self.rorepo, binsha)

with self.assertRaisesRegex(ValueError, f"{hexsha}.*{typename}.*{wrong_type.type}"):

invalid.size

self.assertEqual(invalid.data_stream.read(), data)

ostream = BytesIO()

invalid.stream_data(ostream)

self.assertEqual(ostream.getvalue(), data)

# END for each object type to create

# Each has a unique sha.

self.assertEqual(len(s), num_objs)

self.assertEqual(len(s | s), num_objs)

self.assertEqual(num_index_objs, 2)

def test_get_object_type_by_name(self):

for tname in base.Object.TYPES:

assert base.Object in get_object_type_by_name(tname).mro()

# END for each known type

self.assertRaises(ValueError, get_object_type_by_name, b"doesntexist")

def test_object_resolution(self):

# Objects must be resolved to shas so they compare equal.

self.assertEqual(self.rorepo.head.reference.object, self.rorepo.active_branch.object)

@with_rw_repo("HEAD", bare=True)

def test_with_bare_rw_repo(self, bare_rw_repo: Repo):

assert bare_rw_repo.config_reader("repository").getboolean("core", "bare")

assert osp.isfile(osp.join(bare_rw_repo.git_dir, "HEAD"))

assert osp.isdir(bare_rw_repo.working_dir)

assert bare_rw_repo.working_tree_dir is None

@with_rw_repo("0.1.6")

def test_with_rw_repo(self, rw_repo: Repo):

assert not rw_repo.config_reader("repository").getboolean("core", "bare")

assert osp.isdir(rw_repo.working_tree_dir)

assert osp.isdir(osp.join(rw_repo.working_tree_dir, "lib"))

assert osp.isdir(rw_repo.working_dir)

@skipIf(HIDE_WINDOWS_FREEZE_ERRORS, "FIXME: Freezes! sometimes...")

@with_rw_and_rw_remote_repo("0.1.6")

def test_with_rw_remote_and_rw_repo(self, rw_repo, rw_remote_repo):

assert not rw_repo.config_reader("repository").getboolean("core", "bare")

assert rw_remote_repo.config_reader("repository").getboolean("core", "bare")

assert osp.isdir(osp.join(rw_repo.working_tree_dir, "lib"))

@with_rw_repo("0.1.6")

def test_add_unicode(self, rw_repo):

filename = "שלום.txt"

file_path = osp.join(rw_repo.working_dir, filename)

# Verify first that we could encode file name in this environment.

try:

file_path.encode(sys.getfilesystemencoding())

except UnicodeEncodeError as e:

raise RuntimeError("Environment doesn't support unicode filenames") from e

with open(file_path, "wb") as fp:

fp.write(b"something")

if sys.platform == "win32":

# On Windows, there is no way this works, see images on:

# https://github.com/gitpython-developers/GitPython/issues/147#issuecomment-68881897

# Therefore, it must be added using the Python implementation.

rw_repo.index.add([file_path])

# However, when the test winds down, rmtree fails to delete this file, which

# is recognized as ??? only.

else:

# On POSIX, we can just add Unicode files without problems.

rw_repo.git.add(rw_repo.working_dir)

rw_repo.index.commit("message")

GitPython/test/test_base.py at main · gitpython-developers/GitPython · GitHub

Repository navigation

Expand file tree

Breadcrumbs

test_base.py

Latest commit

History

Breadcrumbs

test_base.py

File metadata and controls