test_transfer.py 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716
  1. import logging
  2. import pytest
  3. from flexmock import flexmock
  4. from borgmatic.borg import transfer as module
  5. from ..test_verbosity import insert_logging_mock
  6. def test_transfer_archives_calls_borg_with_flags():
  7. flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
  8. flexmock(module.logging).ANSWER = module.borgmatic.logger.ANSWER
  9. flexmock(module.flags).should_receive('make_flags').and_return(())
  10. flexmock(module.flags).should_receive('make_match_archives_flags').and_return(())
  11. flexmock(module.flags).should_receive('make_flags_from_arguments').and_return(())
  12. flexmock(module.flags).should_receive('make_repository_flags').and_return(('--repo', 'repo'))
  13. flexmock(module.environment).should_receive('make_environment')
  14. flexmock(module.borgmatic.config.paths).should_receive('get_working_directory').and_return(None)
  15. flexmock(module).should_receive('execute_command').with_args(
  16. ('borg', 'transfer', '--repo', 'repo'),
  17. output_log_level=module.borgmatic.logger.ANSWER,
  18. output_file=None,
  19. environment=None,
  20. working_directory=None,
  21. borg_local_path='borg',
  22. borg_exit_codes=None,
  23. )
  24. module.transfer_archives(
  25. dry_run=False,
  26. repository_path='repo',
  27. config={},
  28. local_borg_version='2.3.4',
  29. transfer_arguments=flexmock(
  30. archive=None,
  31. progress=None,
  32. match_archives=None,
  33. source_repository=None,
  34. ),
  35. global_arguments=flexmock(),
  36. )
  37. def test_transfer_archives_with_dry_run_calls_borg_with_dry_run_flag():
  38. flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
  39. flexmock(module.logging).ANSWER = module.borgmatic.logger.ANSWER
  40. flexmock(module.flags).should_receive('make_flags').and_return(())
  41. flexmock(module.flags).should_receive('make_flags').with_args('dry-run', True).and_return(
  42. ('--dry-run',),
  43. )
  44. flexmock(module.flags).should_receive('make_match_archives_flags').and_return(())
  45. flexmock(module.flags).should_receive('make_flags_from_arguments').and_return(())
  46. flexmock(module.flags).should_receive('make_repository_flags').and_return(('--repo', 'repo'))
  47. flexmock(module.environment).should_receive('make_environment')
  48. flexmock(module.borgmatic.config.paths).should_receive('get_working_directory').and_return(None)
  49. flexmock(module).should_receive('execute_command').with_args(
  50. ('borg', 'transfer', '--repo', 'repo', '--dry-run'),
  51. output_log_level=module.borgmatic.logger.ANSWER,
  52. output_file=None,
  53. environment=None,
  54. working_directory=None,
  55. borg_local_path='borg',
  56. borg_exit_codes=None,
  57. )
  58. module.transfer_archives(
  59. dry_run=True,
  60. repository_path='repo',
  61. config={},
  62. local_borg_version='2.3.4',
  63. transfer_arguments=flexmock(
  64. archive=None,
  65. progress=None,
  66. match_archives=None,
  67. source_repository=None,
  68. ),
  69. global_arguments=flexmock(),
  70. )
  71. def test_transfer_archives_with_log_info_calls_borg_with_info_flag():
  72. flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
  73. flexmock(module.logging).ANSWER = module.borgmatic.logger.ANSWER
  74. flexmock(module.flags).should_receive('make_flags').and_return(())
  75. flexmock(module.flags).should_receive('make_match_archives_flags').and_return(())
  76. flexmock(module.flags).should_receive('make_flags_from_arguments').and_return(())
  77. flexmock(module.flags).should_receive('make_repository_flags').and_return(('--repo', 'repo'))
  78. flexmock(module.environment).should_receive('make_environment')
  79. flexmock(module.borgmatic.config.paths).should_receive('get_working_directory').and_return(None)
  80. flexmock(module).should_receive('execute_command').with_args(
  81. ('borg', 'transfer', '--info', '--repo', 'repo'),
  82. output_log_level=module.borgmatic.logger.ANSWER,
  83. output_file=None,
  84. environment=None,
  85. working_directory=None,
  86. borg_local_path='borg',
  87. borg_exit_codes=None,
  88. )
  89. insert_logging_mock(logging.INFO)
  90. module.transfer_archives(
  91. dry_run=False,
  92. repository_path='repo',
  93. config={},
  94. local_borg_version='2.3.4',
  95. transfer_arguments=flexmock(
  96. archive=None,
  97. progress=None,
  98. match_archives=None,
  99. source_repository=None,
  100. ),
  101. global_arguments=flexmock(),
  102. )
  103. def test_transfer_archives_with_log_debug_calls_borg_with_debug_flag():
  104. flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
  105. flexmock(module.logging).ANSWER = module.borgmatic.logger.ANSWER
  106. flexmock(module.flags).should_receive('make_flags').and_return(())
  107. flexmock(module.flags).should_receive('make_match_archives_flags').and_return(())
  108. flexmock(module.flags).should_receive('make_flags_from_arguments').and_return(())
  109. flexmock(module.flags).should_receive('make_repository_flags').and_return(('--repo', 'repo'))
  110. flexmock(module.environment).should_receive('make_environment')
  111. flexmock(module.borgmatic.config.paths).should_receive('get_working_directory').and_return(None)
  112. flexmock(module).should_receive('execute_command').with_args(
  113. ('borg', 'transfer', '--debug', '--show-rc', '--repo', 'repo'),
  114. output_log_level=module.borgmatic.logger.ANSWER,
  115. output_file=None,
  116. environment=None,
  117. working_directory=None,
  118. borg_local_path='borg',
  119. borg_exit_codes=None,
  120. )
  121. insert_logging_mock(logging.DEBUG)
  122. module.transfer_archives(
  123. dry_run=False,
  124. repository_path='repo',
  125. config={},
  126. local_borg_version='2.3.4',
  127. transfer_arguments=flexmock(
  128. archive=None,
  129. progress=None,
  130. match_archives=None,
  131. source_repository=None,
  132. ),
  133. global_arguments=flexmock(),
  134. )
  135. def test_transfer_archives_with_archive_calls_borg_with_match_archives_flag():
  136. flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
  137. flexmock(module.logging).ANSWER = module.borgmatic.logger.ANSWER
  138. flexmock(module.flags).should_receive('make_flags').and_return(())
  139. flexmock(module.flags).should_receive('make_match_archives_flags').with_args(
  140. 'archive',
  141. 'bar-{now}',
  142. '2.3.4',
  143. ).and_return(('--match-archives', 'archive'))
  144. flexmock(module.flags).should_receive('make_flags_from_arguments').and_return(())
  145. flexmock(module.flags).should_receive('make_repository_flags').and_return(('--repo', 'repo'))
  146. flexmock(module.environment).should_receive('make_environment')
  147. flexmock(module.borgmatic.config.paths).should_receive('get_working_directory').and_return(None)
  148. flexmock(module).should_receive('execute_command').with_args(
  149. ('borg', 'transfer', '--match-archives', 'archive', '--repo', 'repo'),
  150. output_log_level=module.borgmatic.logger.ANSWER,
  151. output_file=None,
  152. environment=None,
  153. working_directory=None,
  154. borg_local_path='borg',
  155. borg_exit_codes=None,
  156. )
  157. module.transfer_archives(
  158. dry_run=False,
  159. repository_path='repo',
  160. config={'archive_name_format': 'bar-{now}'},
  161. local_borg_version='2.3.4',
  162. transfer_arguments=flexmock(
  163. archive='archive',
  164. progress=None,
  165. match_archives=None,
  166. source_repository=None,
  167. ),
  168. global_arguments=flexmock(),
  169. )
  170. def test_transfer_archives_with_match_archives_calls_borg_with_match_archives_flag():
  171. flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
  172. flexmock(module.logging).ANSWER = module.borgmatic.logger.ANSWER
  173. flexmock(module.flags).should_receive('make_flags').and_return(())
  174. flexmock(module.flags).should_receive('make_match_archives_flags').with_args(
  175. 'sh:foo*',
  176. 'bar-{now}',
  177. '2.3.4',
  178. ).and_return(('--match-archives', 'sh:foo*'))
  179. flexmock(module.flags).should_receive('make_flags_from_arguments').and_return(())
  180. flexmock(module.flags).should_receive('make_repository_flags').and_return(('--repo', 'repo'))
  181. flexmock(module.environment).should_receive('make_environment')
  182. flexmock(module.borgmatic.config.paths).should_receive('get_working_directory').and_return(None)
  183. flexmock(module).should_receive('execute_command').with_args(
  184. ('borg', 'transfer', '--match-archives', 'sh:foo*', '--repo', 'repo'),
  185. output_log_level=module.borgmatic.logger.ANSWER,
  186. output_file=None,
  187. environment=None,
  188. working_directory=None,
  189. borg_local_path='borg',
  190. borg_exit_codes=None,
  191. )
  192. module.transfer_archives(
  193. dry_run=False,
  194. repository_path='repo',
  195. config={'archive_name_format': 'bar-{now}', 'match_archives': 'sh:foo*'},
  196. local_borg_version='2.3.4',
  197. transfer_arguments=flexmock(
  198. archive=None,
  199. progress=None,
  200. match_archives='sh:foo*',
  201. source_repository=None,
  202. ),
  203. global_arguments=flexmock(),
  204. )
  205. def test_transfer_archives_with_archive_name_format_calls_borg_with_match_archives_flag():
  206. flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
  207. flexmock(module.logging).ANSWER = module.borgmatic.logger.ANSWER
  208. flexmock(module.flags).should_receive('make_flags').and_return(())
  209. flexmock(module.flags).should_receive('make_match_archives_flags').with_args(
  210. None,
  211. 'bar-{now}',
  212. '2.3.4',
  213. ).and_return(('--match-archives', 'sh:bar-*'))
  214. flexmock(module.flags).should_receive('make_flags_from_arguments').and_return(())
  215. flexmock(module.flags).should_receive('make_repository_flags').and_return(('--repo', 'repo'))
  216. flexmock(module.environment).should_receive('make_environment')
  217. flexmock(module.borgmatic.config.paths).should_receive('get_working_directory').and_return(None)
  218. flexmock(module).should_receive('execute_command').with_args(
  219. ('borg', 'transfer', '--match-archives', 'sh:bar-*', '--repo', 'repo'),
  220. output_log_level=module.borgmatic.logger.ANSWER,
  221. output_file=None,
  222. environment=None,
  223. working_directory=None,
  224. borg_local_path='borg',
  225. borg_exit_codes=None,
  226. )
  227. module.transfer_archives(
  228. dry_run=False,
  229. repository_path='repo',
  230. config={'archive_name_format': 'bar-{now}'},
  231. local_borg_version='2.3.4',
  232. transfer_arguments=flexmock(
  233. archive=None,
  234. progress=None,
  235. match_archives=None,
  236. source_repository=None,
  237. ),
  238. global_arguments=flexmock(),
  239. )
  240. def test_transfer_archives_with_local_path_calls_borg_via_local_path():
  241. flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
  242. flexmock(module.logging).ANSWER = module.borgmatic.logger.ANSWER
  243. flexmock(module.flags).should_receive('make_flags').and_return(())
  244. flexmock(module.flags).should_receive('make_match_archives_flags').and_return(())
  245. flexmock(module.flags).should_receive('make_flags_from_arguments').and_return(())
  246. flexmock(module.flags).should_receive('make_repository_flags').and_return(('--repo', 'repo'))
  247. flexmock(module.environment).should_receive('make_environment')
  248. flexmock(module.borgmatic.config.paths).should_receive('get_working_directory').and_return(None)
  249. flexmock(module).should_receive('execute_command').with_args(
  250. ('borg2', 'transfer', '--repo', 'repo'),
  251. output_log_level=module.borgmatic.logger.ANSWER,
  252. output_file=None,
  253. environment=None,
  254. working_directory=None,
  255. borg_local_path='borg2',
  256. borg_exit_codes=None,
  257. )
  258. module.transfer_archives(
  259. dry_run=False,
  260. repository_path='repo',
  261. config={},
  262. local_borg_version='2.3.4',
  263. transfer_arguments=flexmock(
  264. archive=None,
  265. progress=None,
  266. match_archives=None,
  267. source_repository=None,
  268. ),
  269. global_arguments=flexmock(),
  270. local_path='borg2',
  271. )
  272. def test_transfer_archives_with_exit_codes_calls_borg_using_them():
  273. flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
  274. flexmock(module.logging).ANSWER = module.borgmatic.logger.ANSWER
  275. flexmock(module.flags).should_receive('make_flags').and_return(())
  276. flexmock(module.flags).should_receive('make_match_archives_flags').and_return(())
  277. flexmock(module.flags).should_receive('make_flags_from_arguments').and_return(())
  278. flexmock(module.flags).should_receive('make_repository_flags').and_return(('--repo', 'repo'))
  279. flexmock(module.environment).should_receive('make_environment')
  280. flexmock(module.borgmatic.config.paths).should_receive('get_working_directory').and_return(None)
  281. borg_exit_codes = flexmock()
  282. flexmock(module).should_receive('execute_command').with_args(
  283. ('borg', 'transfer', '--repo', 'repo'),
  284. output_log_level=module.borgmatic.logger.ANSWER,
  285. output_file=None,
  286. environment=None,
  287. working_directory=None,
  288. borg_local_path='borg',
  289. borg_exit_codes=borg_exit_codes,
  290. )
  291. module.transfer_archives(
  292. dry_run=False,
  293. repository_path='repo',
  294. config={'borg_exit_codes': borg_exit_codes},
  295. local_borg_version='2.3.4',
  296. transfer_arguments=flexmock(
  297. archive=None,
  298. progress=None,
  299. match_archives=None,
  300. source_repository=None,
  301. ),
  302. global_arguments=flexmock(),
  303. )
  304. def test_transfer_archives_with_remote_path_calls_borg_with_remote_path_flags():
  305. flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
  306. flexmock(module.logging).ANSWER = module.borgmatic.logger.ANSWER
  307. flexmock(module.flags).should_receive('make_flags').and_return(())
  308. flexmock(module.flags).should_receive('make_flags').with_args(
  309. 'remote-path',
  310. 'borg2',
  311. ).and_return(('--remote-path', 'borg2'))
  312. flexmock(module.flags).should_receive('make_match_archives_flags').and_return(())
  313. flexmock(module.flags).should_receive('make_flags_from_arguments').and_return(())
  314. flexmock(module.flags).should_receive('make_repository_flags').and_return(('--repo', 'repo'))
  315. flexmock(module.environment).should_receive('make_environment')
  316. flexmock(module.borgmatic.config.paths).should_receive('get_working_directory').and_return(None)
  317. flexmock(module).should_receive('execute_command').with_args(
  318. ('borg', 'transfer', '--remote-path', 'borg2', '--repo', 'repo'),
  319. output_log_level=module.borgmatic.logger.ANSWER,
  320. output_file=None,
  321. environment=None,
  322. working_directory=None,
  323. borg_local_path='borg',
  324. borg_exit_codes=None,
  325. )
  326. module.transfer_archives(
  327. dry_run=False,
  328. repository_path='repo',
  329. config={},
  330. local_borg_version='2.3.4',
  331. transfer_arguments=flexmock(
  332. archive=None,
  333. progress=None,
  334. match_archives=None,
  335. source_repository=None,
  336. ),
  337. global_arguments=flexmock(),
  338. remote_path='borg2',
  339. )
  340. def test_transfer_archives_with_umask_calls_borg_with_umask_flags():
  341. flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
  342. flexmock(module.logging).ANSWER = module.borgmatic.logger.ANSWER
  343. flexmock(module.flags).should_receive('make_flags').replace_with(
  344. lambda name, value: (f'--{name}', value) if value else (),
  345. )
  346. flexmock(module.flags).should_receive('make_match_archives_flags').and_return(())
  347. flexmock(module.flags).should_receive('make_flags_from_arguments').and_return(())
  348. flexmock(module.flags).should_receive('make_repository_flags').and_return(('--repo', 'repo'))
  349. flexmock(module.environment).should_receive('make_environment')
  350. flexmock(module.borgmatic.config.paths).should_receive('get_working_directory').and_return(None)
  351. flexmock(module).should_receive('execute_command').with_args(
  352. ('borg', 'transfer', '--umask', '077', '--repo', 'repo'),
  353. output_log_level=module.borgmatic.logger.ANSWER,
  354. output_file=None,
  355. environment=None,
  356. working_directory=None,
  357. borg_local_path='borg',
  358. borg_exit_codes=None,
  359. )
  360. module.transfer_archives(
  361. dry_run=False,
  362. repository_path='repo',
  363. config={'umask': '077'},
  364. local_borg_version='2.3.4',
  365. transfer_arguments=flexmock(
  366. archive=None,
  367. progress=None,
  368. match_archives=None,
  369. source_repository=None,
  370. ),
  371. global_arguments=flexmock(),
  372. )
  373. def test_transfer_archives_with_log_json_calls_borg_with_log_json_flags():
  374. flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
  375. flexmock(module.logging).ANSWER = module.borgmatic.logger.ANSWER
  376. flexmock(module.flags).should_receive('make_flags').and_return(())
  377. flexmock(module.flags).should_receive('make_flags').with_args('log-json', True).and_return(
  378. ('--log-json',),
  379. )
  380. flexmock(module.flags).should_receive('make_match_archives_flags').and_return(())
  381. flexmock(module.flags).should_receive('make_flags_from_arguments').and_return(())
  382. flexmock(module.flags).should_receive('make_repository_flags').and_return(('--repo', 'repo'))
  383. flexmock(module.environment).should_receive('make_environment')
  384. flexmock(module.borgmatic.config.paths).should_receive('get_working_directory').and_return(None)
  385. flexmock(module).should_receive('execute_command').with_args(
  386. ('borg', 'transfer', '--log-json', '--repo', 'repo'),
  387. output_log_level=module.borgmatic.logger.ANSWER,
  388. output_file=None,
  389. environment=None,
  390. working_directory=None,
  391. borg_local_path='borg',
  392. borg_exit_codes=None,
  393. )
  394. module.transfer_archives(
  395. dry_run=False,
  396. repository_path='repo',
  397. config={'log_json': True},
  398. local_borg_version='2.3.4',
  399. transfer_arguments=flexmock(
  400. archive=None,
  401. progress=None,
  402. match_archives=None,
  403. source_repository=None,
  404. ),
  405. global_arguments=flexmock(),
  406. )
  407. def test_transfer_archives_with_lock_wait_calls_borg_with_lock_wait_flags():
  408. flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
  409. flexmock(module.logging).ANSWER = module.borgmatic.logger.ANSWER
  410. flexmock(module.flags).should_receive('make_flags').and_return(())
  411. flexmock(module.flags).should_receive('make_flags').with_args('lock-wait', 5).and_return(
  412. ('--lock-wait', '5'),
  413. )
  414. flexmock(module.flags).should_receive('make_match_archives_flags').and_return(())
  415. flexmock(module.flags).should_receive('make_flags_from_arguments').and_return(())
  416. flexmock(module.flags).should_receive('make_repository_flags').and_return(('--repo', 'repo'))
  417. config = {'lock_wait': 5}
  418. flexmock(module.environment).should_receive('make_environment')
  419. flexmock(module.borgmatic.config.paths).should_receive('get_working_directory').and_return(None)
  420. flexmock(module).should_receive('execute_command').with_args(
  421. ('borg', 'transfer', '--lock-wait', '5', '--repo', 'repo'),
  422. output_log_level=module.borgmatic.logger.ANSWER,
  423. output_file=None,
  424. environment=None,
  425. working_directory=None,
  426. borg_local_path='borg',
  427. borg_exit_codes=None,
  428. )
  429. module.transfer_archives(
  430. dry_run=False,
  431. repository_path='repo',
  432. config=config,
  433. local_borg_version='2.3.4',
  434. transfer_arguments=flexmock(
  435. archive=None,
  436. progress=None,
  437. match_archives=None,
  438. source_repository=None,
  439. ),
  440. global_arguments=flexmock(),
  441. )
  442. def test_transfer_archives_calls_borg_with_extra_borg_options():
  443. flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
  444. flexmock(module.logging).ANSWER = module.borgmatic.logger.ANSWER
  445. flexmock(module.flags).should_receive('make_flags').and_return(())
  446. flexmock(module.flags).should_receive('make_match_archives_flags').and_return(())
  447. flexmock(module.flags).should_receive('make_flags_from_arguments').and_return(())
  448. flexmock(module.flags).should_receive('make_repository_flags').and_return(('--repo', 'repo'))
  449. config = {'extra_borg_options': {'transfer': '--extra "value with space"'}}
  450. flexmock(module.environment).should_receive('make_environment')
  451. flexmock(module.borgmatic.config.paths).should_receive('get_working_directory').and_return(None)
  452. flexmock(module).should_receive('execute_command').with_args(
  453. ('borg', 'transfer', '--extra', 'value with space', '--repo', 'repo'),
  454. output_log_level=module.borgmatic.logger.ANSWER,
  455. output_file=None,
  456. environment=None,
  457. working_directory=None,
  458. borg_local_path='borg',
  459. borg_exit_codes=None,
  460. )
  461. module.transfer_archives(
  462. dry_run=False,
  463. repository_path='repo',
  464. config=config,
  465. local_borg_version='2.3.4',
  466. transfer_arguments=flexmock(
  467. archive=None,
  468. progress=None,
  469. match_archives=None,
  470. source_repository=None,
  471. ),
  472. global_arguments=flexmock(),
  473. )
  474. def test_transfer_archives_with_progress_calls_borg_with_progress_flags():
  475. flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
  476. flexmock(module.logging).ANSWER = module.borgmatic.logger.ANSWER
  477. flexmock(module.flags).should_receive('make_flags').and_return(())
  478. flexmock(module.flags).should_receive('make_flags').with_args('progress', True).and_return(
  479. ('--progress',),
  480. )
  481. flexmock(module.flags).should_receive('make_match_archives_flags').and_return(())
  482. flexmock(module.flags).should_receive('make_flags_from_arguments').and_return(())
  483. flexmock(module.flags).should_receive('make_repository_flags').and_return(('--repo', 'repo'))
  484. flexmock(module.environment).should_receive('make_environment')
  485. flexmock(module.borgmatic.config.paths).should_receive('get_working_directory').and_return(None)
  486. flexmock(module).should_receive('execute_command').with_args(
  487. ('borg', 'transfer', '--progress', '--repo', 'repo'),
  488. output_log_level=module.borgmatic.logger.ANSWER,
  489. output_file=module.DO_NOT_CAPTURE,
  490. environment=None,
  491. working_directory=None,
  492. borg_local_path='borg',
  493. borg_exit_codes=None,
  494. )
  495. module.transfer_archives(
  496. dry_run=False,
  497. repository_path='repo',
  498. config={'progress': True},
  499. local_borg_version='2.3.4',
  500. transfer_arguments=flexmock(
  501. archive=None,
  502. progress=None,
  503. match_archives=None,
  504. source_repository=None,
  505. ),
  506. global_arguments=flexmock(),
  507. )
  508. @pytest.mark.parametrize('argument_name', ('upgrader', 'sort_by', 'first', 'last'))
  509. def test_transfer_archives_passes_through_arguments_to_borg(argument_name):
  510. flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
  511. flexmock(module.logging).ANSWER = module.borgmatic.logger.ANSWER
  512. flag_name = f"--{argument_name.replace('_', ' ')}"
  513. flexmock(module.flags).should_receive('make_flags').and_return(())
  514. flexmock(module.flags).should_receive('make_match_archives_flags').and_return(())
  515. flexmock(module.flags).should_receive('make_flags_from_arguments').and_return(
  516. (flag_name, 'value'),
  517. )
  518. flexmock(module.flags).should_receive('make_repository_flags').and_return(('--repo', 'repo'))
  519. flexmock(module.environment).should_receive('make_environment')
  520. flexmock(module.borgmatic.config.paths).should_receive('get_working_directory').and_return(None)
  521. flexmock(module).should_receive('execute_command').with_args(
  522. ('borg', 'transfer', flag_name, 'value', '--repo', 'repo'),
  523. output_log_level=module.borgmatic.logger.ANSWER,
  524. output_file=None,
  525. environment=None,
  526. working_directory=None,
  527. borg_local_path='borg',
  528. borg_exit_codes=None,
  529. )
  530. module.transfer_archives(
  531. dry_run=False,
  532. repository_path='repo',
  533. config={},
  534. local_borg_version='2.3.4',
  535. transfer_arguments=flexmock(
  536. archive=None,
  537. progress=None,
  538. match_archives=None,
  539. source_repository=None,
  540. **{argument_name: 'value'},
  541. ),
  542. global_arguments=flexmock(),
  543. )
  544. def test_transfer_archives_with_source_repository_calls_borg_with_other_repo_flags():
  545. flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
  546. flexmock(module.flags).should_receive('make_flags').and_return(())
  547. flexmock(module.flags).should_receive('make_flags').with_args('other-repo', 'other').and_return(
  548. ('--other-repo', 'other'),
  549. )
  550. flexmock(module.flags).should_receive('make_match_archives_flags').and_return(())
  551. flexmock(module.flags).should_receive('make_flags_from_arguments').and_return(())
  552. flexmock(module.flags).should_receive('make_repository_flags').and_return(('--repo', 'repo'))
  553. flexmock(module.environment).should_receive('make_environment')
  554. flexmock(module.borgmatic.config.paths).should_receive('get_working_directory').and_return(None)
  555. flexmock(module).should_receive('execute_command').with_args(
  556. ('borg', 'transfer', '--repo', 'repo', '--other-repo', 'other'),
  557. output_log_level=module.borgmatic.logger.ANSWER,
  558. output_file=None,
  559. environment=None,
  560. working_directory=None,
  561. borg_local_path='borg',
  562. borg_exit_codes=None,
  563. )
  564. module.transfer_archives(
  565. dry_run=False,
  566. repository_path='repo',
  567. config={},
  568. local_borg_version='2.3.4',
  569. transfer_arguments=flexmock(
  570. archive=None,
  571. progress=None,
  572. match_archives=None,
  573. source_repository='other',
  574. ),
  575. global_arguments=flexmock(),
  576. )
  577. def test_transfer_archives_with_date_based_matching_calls_borg_with_date_based_flags():
  578. flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
  579. flexmock(module.flags).should_receive('make_flags').and_return(())
  580. flexmock(module.flags).should_receive('make_match_archives_flags').and_return(())
  581. flexmock(module.flags).should_receive('make_flags_from_arguments').and_return(
  582. ('--newer', '1d', '--newest', '1y', '--older', '1m', '--oldest', '1w'),
  583. )
  584. flexmock(module.flags).should_receive('make_repository_flags').and_return(('--repo', 'repo'))
  585. flexmock(module.environment).should_receive('make_environment')
  586. flexmock(module.borgmatic.config.paths).should_receive('get_working_directory').and_return(None)
  587. flexmock(module).should_receive('execute_command').with_args(
  588. (
  589. 'borg',
  590. 'transfer',
  591. '--newer',
  592. '1d',
  593. '--newest',
  594. '1y',
  595. '--older',
  596. '1m',
  597. '--oldest',
  598. '1w',
  599. '--repo',
  600. 'repo',
  601. ),
  602. output_log_level=module.borgmatic.logger.ANSWER,
  603. output_file=None,
  604. environment=None,
  605. working_directory=None,
  606. borg_local_path='borg',
  607. borg_exit_codes=None,
  608. )
  609. module.transfer_archives(
  610. dry_run=False,
  611. repository_path='repo',
  612. config={},
  613. local_borg_version='2.3.4',
  614. global_arguments=flexmock(),
  615. transfer_arguments=flexmock(
  616. archive=None,
  617. progress=None,
  618. source_repository='other',
  619. newer='1d',
  620. newest='1y',
  621. older='1m',
  622. oldest='1w',
  623. ),
  624. )
  625. def test_transfer_archives_calls_borg_with_working_directory():
  626. flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
  627. flexmock(module.logging).ANSWER = module.borgmatic.logger.ANSWER
  628. flexmock(module.flags).should_receive('make_flags').and_return(())
  629. flexmock(module.flags).should_receive('make_match_archives_flags').and_return(())
  630. flexmock(module.flags).should_receive('make_flags_from_arguments').and_return(())
  631. flexmock(module.flags).should_receive('make_repository_flags').and_return(('--repo', 'repo'))
  632. flexmock(module.environment).should_receive('make_environment')
  633. flexmock(module.borgmatic.config.paths).should_receive('get_working_directory').and_return(
  634. '/working/dir',
  635. )
  636. flexmock(module).should_receive('execute_command').with_args(
  637. ('borg', 'transfer', '--repo', 'repo'),
  638. output_log_level=module.borgmatic.logger.ANSWER,
  639. output_file=None,
  640. environment=None,
  641. working_directory='/working/dir',
  642. borg_local_path='borg',
  643. borg_exit_codes=None,
  644. )
  645. module.transfer_archives(
  646. dry_run=False,
  647. repository_path='repo',
  648. config={'working_directory': '/working/dir'},
  649. local_borg_version='2.3.4',
  650. transfer_arguments=flexmock(
  651. archive=None,
  652. progress=None,
  653. match_archives=None,
  654. source_repository=None,
  655. ),
  656. global_arguments=flexmock(),
  657. )