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
115
116
117
|
from xdg import BaseDirectory
import os
from os import environ
import unittest
import tempfile, shutil
import stat
try:
reload
except NameError:
from importlib import reload
class BaseDirectoryTest(unittest.TestCase):
def setUp(self):
self.environ_save = environ.copy()
def tearDown(self):
# Restore environment variables
environ.clear()
for k, v in self.environ_save.items():
environ[k] = v
def test_save_config_path(self):
tmpdir = tempfile.mkdtemp()
try:
environ['XDG_CONFIG_HOME'] = tmpdir
reload(BaseDirectory)
configpath = BaseDirectory.save_config_path("foo")
self.assertEqual(configpath, os.path.join(tmpdir, "foo"))
finally:
shutil.rmtree(tmpdir)
def test_save_data_path(self):
tmpdir = tempfile.mkdtemp()
try:
environ['XDG_DATA_HOME'] = tmpdir
reload(BaseDirectory)
datapath = BaseDirectory.save_data_path("foo")
self.assertEqual(datapath, os.path.join(tmpdir, "foo"))
finally:
shutil.rmtree(tmpdir)
def test_save_cache_path(self):
tmpdir = tempfile.mkdtemp()
try:
environ['XDG_CACHE_HOME'] = tmpdir
reload(BaseDirectory)
datapath = BaseDirectory.save_cache_path("foo")
self.assertEqual(datapath, os.path.join(tmpdir, "foo"))
finally:
shutil.rmtree(tmpdir)
def test_save_state_path(self):
tmpdir = tempfile.mkdtemp()
try:
environ['XDG_STATE_HOME'] = tmpdir
reload(BaseDirectory)
statepath = BaseDirectory.save_state_path("foo")
self.assertEqual(statepath, os.path.join(tmpdir, "foo"))
finally:
shutil.rmtree(tmpdir)
def test_load_first_config(self):
tmpdir = tempfile.mkdtemp()
tmpdir2 = tempfile.mkdtemp()
tmpdir3 = tempfile.mkdtemp()
path = os.path.join(tmpdir3, "wpokewefketnrhruit")
os.mkdir(path)
try:
environ['XDG_CONFIG_HOME'] = tmpdir
environ['XDG_CONFIG_DIRS'] = ":".join([tmpdir2, tmpdir3])
reload(BaseDirectory)
configdir = BaseDirectory.load_first_config("wpokewefketnrhruit")
self.assertEqual(configdir, path)
finally:
shutil.rmtree(tmpdir)
shutil.rmtree(tmpdir2)
shutil.rmtree(tmpdir3)
def test_load_config_paths(self):
tmpdir = tempfile.mkdtemp()
path = os.path.join(tmpdir, "wpokewefketnrhruit")
os.mkdir(path)
tmpdir2 = tempfile.mkdtemp()
path2 = os.path.join(tmpdir2, "wpokewefketnrhruit")
os.mkdir(path2)
try:
environ['XDG_CONFIG_HOME'] = tmpdir
environ['XDG_CONFIG_DIRS'] = tmpdir2 + ":/etc/xdg"
reload(BaseDirectory)
configdirs = BaseDirectory.load_config_paths("wpokewefketnrhruit")
self.assertEqual(list(configdirs), [path, path2])
finally:
shutil.rmtree(tmpdir)
shutil.rmtree(tmpdir2)
def test_runtime_dir(self):
rd = '/pyxdg-example/run/user/fred'
environ['XDG_RUNTIME_DIR'] = rd
self.assertEqual(BaseDirectory.get_runtime_dir(strict=True), rd)
self.assertEqual(BaseDirectory.get_runtime_dir(strict=False), rd)
def test_runtime_dir_notset(self):
environ.pop('XDG_RUNTIME_DIR', None)
self.assertRaises(KeyError, BaseDirectory.get_runtime_dir, strict=True)
fallback = BaseDirectory.get_runtime_dir(strict=False)
assert fallback.startswith('/tmp/'), fallback
assert os.path.isdir(fallback), fallback
mode = stat.S_IMODE(os.stat(fallback).st_mode)
self.assertEqual(mode, stat.S_IRUSR|stat.S_IWUSR|stat.S_IXUSR)
# Calling it again should return the same directory.
fallback2 = BaseDirectory.get_runtime_dir(strict=False)
self.assertEqual(fallback, fallback2)
mode = stat.S_IMODE(os.stat(fallback2).st_mode)
self.assertEqual(mode, stat.S_IRUSR|stat.S_IWUSR|stat.S_IXUSR)
|
