def test_trip_at_when_not_lazy_relative(os):
("Node#trip_at(path, lazy=False) returns a list when lazy=False "
"(testing with relative path)")
os.walk.return_value = [
("/foo/bar/somewhere", [], ["file1.py", "file2.py"]),
]
nd = Node("/foo/bar/")
nd.trip_at('somewhere', lazy=False).should.equal([
"/foo/bar/somewhere/file1.py",
"/foo/bar/somewhere/file2.py",
])
def test_trip_at_when_not_lazy_absolute(os):
("Node#trip_at(path, lazy=False) returns a list when lazy=False "
"(testing with absolute path)")
os.walk.return_value = [
("/dummy/", [], ["file1.py", "file2.py"]),
]
nd = Node("/foo/bar/")
nd.trip_at('/dummy/', lazy=False).should.equal([
"/dummy/file1.py",
"/dummy/file2.py",
])
def test_trip_at_when_lazy_relative(os):
("Node#trip_at(path, lazy=True) returns a generator when lazy=True "
"(testing with relative path)")
os.walk.return_value = [
("/dummy/", [], ["file1.py", "file2.py"]),
]
nd = Node("/foo/bar/")
nd.trip_at('/dummy/', lazy=True).should.be.a('types.GeneratorType')
list(nd.trip_at('/dummy/', lazy=True)).should.equal([
"/dummy/file1.py",
"/dummy/file2.py",
])
def test_walk_trips_at_node_path():
("Node#walk() trips at node.path")
nd = Node("/foo/bar/")
nd.trip_at = Mock()
nd.walk()
nd.walk(lazy=True)
nd.trip_at.assert_has_calls([
call('/foo/bar', lazy=False),
call('/foo/bar', lazy=True),
])