def test_save_phone():
lf = LocalFiles(BASE_DIR)
lf.save_phone(434353453, 'barsik1')
assert os.path.isfile(os.path.join(lf.base_path, 'Austria', 'barsik1'))
lf.save_phone(434353453, 'barsik2')
assert not os.path.isfile(os.path.join(lf.base_path, 'Austria', 'barsik2'))
lf.save_phone('eqweqqwed', 'Cat2')
tree = os.walk(BASE_DIR)
for i in tree:
if len(i[2]) != 0:
for j in i[2]:
assert j != 'Cat2'
lf.save_phone('1415eqweq62qwe73d', 'Cat')
tree = os.walk(BASE_DIR)
for i in tree:
if len(i[2]) != 0:
for j in i[2]:
assert j != 'Cat'
lf.save_phone(434353455, 'Bob')
lf.save_phone(434353452, 'Bob')
with open(os.path.join(lf.base_path, 'Austria', 'Bob')) as f:
assert f.read() == '434353455'
assert not f.read() == '434353452'
def save(phone, name, path):
lf = LocalFiles() if path is None else LocalFiles(path)
lf.save_phone(phone, name)