test_transfer.py 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681
  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_with_progress_calls_borg_with_progress_flags():
  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_flags').with_args('progress', True).and_return(
  447. ('--progress',),
  448. )
  449. flexmock(module.flags).should_receive('make_match_archives_flags').and_return(())
  450. flexmock(module.flags).should_receive('make_flags_from_arguments').and_return(())
  451. flexmock(module.flags).should_receive('make_repository_flags').and_return(('--repo', 'repo'))
  452. flexmock(module.environment).should_receive('make_environment')
  453. flexmock(module.borgmatic.config.paths).should_receive('get_working_directory').and_return(None)
  454. flexmock(module).should_receive('execute_command').with_args(
  455. ('borg', 'transfer', '--progress', '--repo', 'repo'),
  456. output_log_level=module.borgmatic.logger.ANSWER,
  457. output_file=module.DO_NOT_CAPTURE,
  458. environment=None,
  459. working_directory=None,
  460. borg_local_path='borg',
  461. borg_exit_codes=None,
  462. )
  463. module.transfer_archives(
  464. dry_run=False,
  465. repository_path='repo',
  466. config={'progress': True},
  467. local_borg_version='2.3.4',
  468. transfer_arguments=flexmock(
  469. archive=None,
  470. progress=None,
  471. match_archives=None,
  472. source_repository=None,
  473. ),
  474. global_arguments=flexmock(),
  475. )
  476. @pytest.mark.parametrize('argument_name', ('upgrader', 'sort_by', 'first', 'last'))
  477. def test_transfer_archives_passes_through_arguments_to_borg(argument_name):
  478. flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
  479. flexmock(module.logging).ANSWER = module.borgmatic.logger.ANSWER
  480. flag_name = f"--{argument_name.replace('_', ' ')}"
  481. flexmock(module.flags).should_receive('make_flags').and_return(())
  482. flexmock(module.flags).should_receive('make_match_archives_flags').and_return(())
  483. flexmock(module.flags).should_receive('make_flags_from_arguments').and_return(
  484. (flag_name, 'value'),
  485. )
  486. flexmock(module.flags).should_receive('make_repository_flags').and_return(('--repo', 'repo'))
  487. flexmock(module.environment).should_receive('make_environment')
  488. flexmock(module.borgmatic.config.paths).should_receive('get_working_directory').and_return(None)
  489. flexmock(module).should_receive('execute_command').with_args(
  490. ('borg', 'transfer', flag_name, 'value', '--repo', 'repo'),
  491. output_log_level=module.borgmatic.logger.ANSWER,
  492. output_file=None,
  493. environment=None,
  494. working_directory=None,
  495. borg_local_path='borg',
  496. borg_exit_codes=None,
  497. )
  498. module.transfer_archives(
  499. dry_run=False,
  500. repository_path='repo',
  501. config={},
  502. local_borg_version='2.3.4',
  503. transfer_arguments=flexmock(
  504. archive=None,
  505. progress=None,
  506. match_archives=None,
  507. source_repository=None,
  508. **{argument_name: 'value'},
  509. ),
  510. global_arguments=flexmock(),
  511. )
  512. def test_transfer_archives_with_source_repository_calls_borg_with_other_repo_flags():
  513. flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
  514. flexmock(module.flags).should_receive('make_flags').and_return(())
  515. flexmock(module.flags).should_receive('make_flags').with_args('other-repo', 'other').and_return(
  516. ('--other-repo', 'other'),
  517. )
  518. flexmock(module.flags).should_receive('make_match_archives_flags').and_return(())
  519. flexmock(module.flags).should_receive('make_flags_from_arguments').and_return(())
  520. flexmock(module.flags).should_receive('make_repository_flags').and_return(('--repo', 'repo'))
  521. flexmock(module.environment).should_receive('make_environment')
  522. flexmock(module.borgmatic.config.paths).should_receive('get_working_directory').and_return(None)
  523. flexmock(module).should_receive('execute_command').with_args(
  524. ('borg', 'transfer', '--repo', 'repo', '--other-repo', 'other'),
  525. output_log_level=module.borgmatic.logger.ANSWER,
  526. output_file=None,
  527. environment=None,
  528. working_directory=None,
  529. borg_local_path='borg',
  530. borg_exit_codes=None,
  531. )
  532. module.transfer_archives(
  533. dry_run=False,
  534. repository_path='repo',
  535. config={},
  536. local_borg_version='2.3.4',
  537. transfer_arguments=flexmock(
  538. archive=None,
  539. progress=None,
  540. match_archives=None,
  541. source_repository='other',
  542. ),
  543. global_arguments=flexmock(),
  544. )
  545. def test_transfer_archives_with_date_based_matching_calls_borg_with_date_based_flags():
  546. flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
  547. flexmock(module.flags).should_receive('make_flags').and_return(())
  548. flexmock(module.flags).should_receive('make_match_archives_flags').and_return(())
  549. flexmock(module.flags).should_receive('make_flags_from_arguments').and_return(
  550. ('--newer', '1d', '--newest', '1y', '--older', '1m', '--oldest', '1w'),
  551. )
  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. (
  557. 'borg',
  558. 'transfer',
  559. '--newer',
  560. '1d',
  561. '--newest',
  562. '1y',
  563. '--older',
  564. '1m',
  565. '--oldest',
  566. '1w',
  567. '--repo',
  568. 'repo',
  569. ),
  570. output_log_level=module.borgmatic.logger.ANSWER,
  571. output_file=None,
  572. environment=None,
  573. working_directory=None,
  574. borg_local_path='borg',
  575. borg_exit_codes=None,
  576. )
  577. module.transfer_archives(
  578. dry_run=False,
  579. repository_path='repo',
  580. config={},
  581. local_borg_version='2.3.4',
  582. global_arguments=flexmock(),
  583. transfer_arguments=flexmock(
  584. archive=None,
  585. progress=None,
  586. source_repository='other',
  587. newer='1d',
  588. newest='1y',
  589. older='1m',
  590. oldest='1w',
  591. ),
  592. )
  593. def test_transfer_archives_calls_borg_with_working_directory():
  594. flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
  595. flexmock(module.logging).ANSWER = module.borgmatic.logger.ANSWER
  596. flexmock(module.flags).should_receive('make_flags').and_return(())
  597. flexmock(module.flags).should_receive('make_match_archives_flags').and_return(())
  598. flexmock(module.flags).should_receive('make_flags_from_arguments').and_return(())
  599. flexmock(module.flags).should_receive('make_repository_flags').and_return(('--repo', 'repo'))
  600. flexmock(module.environment).should_receive('make_environment')
  601. flexmock(module.borgmatic.config.paths).should_receive('get_working_directory').and_return(
  602. '/working/dir',
  603. )
  604. flexmock(module).should_receive('execute_command').with_args(
  605. ('borg', 'transfer', '--repo', 'repo'),
  606. output_log_level=module.borgmatic.logger.ANSWER,
  607. output_file=None,
  608. environment=None,
  609. working_directory='/working/dir',
  610. borg_local_path='borg',
  611. borg_exit_codes=None,
  612. )
  613. module.transfer_archives(
  614. dry_run=False,
  615. repository_path='repo',
  616. config={'working_directory': '/working/dir'},
  617. local_borg_version='2.3.4',
  618. transfer_arguments=flexmock(
  619. archive=None,
  620. progress=None,
  621. match_archives=None,
  622. source_repository=None,
  623. ),
  624. global_arguments=flexmock(),
  625. )