21 lines
611 B
Python
21 lines
611 B
Python
import os
|
|
from unittest import mock
|
|
|
|
from babi.user_data import xdg_data
|
|
|
|
|
|
def test_when_xdg_data_home_is_set():
|
|
with mock.patch.dict(os.environ, {'XDG_DATA_HOME': '/foo'}):
|
|
ret = xdg_data('history', 'command')
|
|
assert ret == '/foo/babi/history/command'
|
|
|
|
|
|
def test_when_xdg_data_home_is_not_set():
|
|
def fake_expanduser(s):
|
|
return s.replace('~', '/home/username')
|
|
|
|
with mock.patch.object(os.path, 'expanduser', fake_expanduser):
|
|
with mock.patch.dict(os.environ, clear=True):
|
|
ret = xdg_data('history')
|
|
assert ret == '/home/username/.local/share/babi/history'
|