1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
import imp
import os
import sys
import types
from pytest import raises
from invoke import Config
from invoke.loader import Loader, FilesystemLoader as FSLoader
from invoke.exceptions import CollectionNotFound
from _util import support
class _BasicLoader(Loader):
"""
Tests top level Loader behavior with basic finder stub.
Used when we want to make sure we're testing Loader.load and not e.g.
FilesystemLoader's specific implementation.
"""
def find(self, name):
self.fd, self.path, self.desc = t = imp.find_module(name, [support])
return t
class Loader_:
def exhibits_default_config_object(self):
loader = _BasicLoader()
assert isinstance(loader.config, Config)
assert loader.config.tasks.collection_name == "tasks"
def returns_module_and_location(self):
mod, path = _BasicLoader().load("namespacing")
assert isinstance(mod, types.ModuleType)
assert path == support
def may_configure_config_via_constructor(self):
config = Config({"tasks": {"collection_name": "mytasks"}})
loader = _BasicLoader(config=config)
assert loader.config.tasks.collection_name == "mytasks"
def adds_module_parent_dir_to_sys_path(self):
# Crummy doesn't-explode test.
_BasicLoader().load("namespacing")
def doesnt_dupliate_parent_dir_addition(self):
_BasicLoader().load("namespacing")
_BasicLoader().load("namespacing")
# If the bug is present, this will be 2 at least (and often more, since
# other tests will pollute it (!).
assert sys.path.count(support) == 1
def closes_opened_file_object(self):
loader = _BasicLoader()
loader.load("foo")
assert loader.fd.closed
def can_load_package(self):
loader = _BasicLoader()
# make sure it doesn't explode
loader.load("package")
def load_name_defaults_to_config_tasks_collection_name(self):
"load() name defaults to config.tasks.collection_name"
class MockLoader(_BasicLoader):
def find(self, name):
# Sanity
assert name == "simple_ns_list"
return super(MockLoader, self).find(name)
config = Config({"tasks": {"collection_name": "simple_ns_list"}})
loader = MockLoader(config=config)
# More sanity: expect simple_ns_list.py (not tasks.py)
mod, path = loader.load()
assert mod.__file__ == os.path.join(support, "simple_ns_list.py")
class FilesystemLoader_:
def setup(self):
self.loader = FSLoader(start=support)
def discovery_start_point_defaults_to_cwd(self):
assert FSLoader().start == os.getcwd()
def exposes_start_point_as_attribute(self):
assert FSLoader().start == os.getcwd()
def start_point_is_configurable_via_kwarg(self):
start = "/tmp/"
assert FSLoader(start=start).start == start
def start_point_is_configurable_via_config(self):
config = Config({"tasks": {"search_root": "nowhere"}})
assert FSLoader(config=config).start == "nowhere"
def raises_CollectionNotFound_if_not_found(self):
with raises(CollectionNotFound):
self.loader.load("nope")
def raises_ImportError_if_found_collection_cannot_be_imported(self):
# Instead of masking with a CollectionNotFound
with raises(ImportError):
self.loader.load("oops")
def searches_towards_root_of_filesystem(self):
# Loaded while root is in same dir as .py
directly = self.loader.load("foo")
# Loaded while root is multiple dirs deeper than the .py
deep = os.path.join(support, "ignoreme", "ignoremetoo")
indirectly = FSLoader(start=deep).load("foo")
assert directly == indirectly
|
