|
4 | 4 | from tasks.cephfs.fuse_mount import FuseMount |
5 | 5 | from tasks.cephfs.cephfs_test_case import CephFSTestCase |
6 | 6 | from teuthology.exceptions import CommandFailedError |
| 7 | +from teuthology.contextutil import safe_while, MaxWhileTries |
7 | 8 |
|
8 | 9 | log = logging.getLogger(__name__) |
9 | 10 |
|
@@ -628,3 +629,98 @@ def test_ephemeral_pin_shrink_mds(self): |
628 | 629 | log.info("{0} migrations have occured due to the cluster resizing".format(count)) |
629 | 630 | # rebalancing from 3 -> 2 may cause half of rank 0/1 to move and all of rank 2 |
630 | 631 | self.assertLessEqual((count/len(subtrees_old)), (1.0/3.0/2.0 + 1.0/3.0/2.0 + 1.0/3.0)*1.25) # aka .66 with 25% overbudget |
| 632 | + |
| 633 | +class TestDumpExportStates(CephFSTestCase): |
| 634 | + MDSS_REQUIRED = 2 |
| 635 | + CLIENTS_REQUIRED = 1 |
| 636 | + |
| 637 | + EXPORT_STATES = ['locking', 'discovering', 'freezing', 'prepping', 'warning', 'exporting'] |
| 638 | + |
| 639 | + def setUp(self): |
| 640 | + super().setUp() |
| 641 | + |
| 642 | + self.fs.set_max_mds(self.MDSS_REQUIRED) |
| 643 | + self.status = self.fs.wait_for_daemons() |
| 644 | + |
| 645 | + self.mount_a.run_shell_payload('mkdir -p test/export') |
| 646 | + |
| 647 | + def tearDown(self): |
| 648 | + super().tearDown() |
| 649 | + |
| 650 | + def _wait_for_export_target(self, source, target, sleep=2, timeout=10): |
| 651 | + try: |
| 652 | + with safe_while(sleep=sleep, tries=timeout//sleep) as proceed: |
| 653 | + while proceed(): |
| 654 | + info = self.fs.getinfo().get_rank(self.fs.id, source) |
| 655 | + log.info(f'waiting for rank {target} to be added to the export target') |
| 656 | + if target in info['export_targets']: |
| 657 | + return |
| 658 | + except MaxWhileTries as e: |
| 659 | + raise RuntimeError(f'rank {target} has not been added to export target after {timeout}s') from e |
| 660 | + |
| 661 | + def _dump_export_state(self, rank): |
| 662 | + states = self.fs.rank_asok(['dump_export_states'], rank=rank, status=self.status) |
| 663 | + self.assertTrue(type(states) is list) |
| 664 | + self.assertEqual(len(states), 1) |
| 665 | + return states[0] |
| 666 | + |
| 667 | + def _test_base(self, path, source, target, state_index, kill): |
| 668 | + self.fs.rank_asok(['config', 'set', 'mds_kill_import_at', str(kill)], rank=target, status=self.status) |
| 669 | + |
| 670 | + self.fs.rank_asok(['export', 'dir', path, str(target)], rank=source, status=self.status) |
| 671 | + self._wait_for_export_target(source, target) |
| 672 | + |
| 673 | + target_rank = self.fs.get_rank(rank=target, status=self.status) |
| 674 | + self.delete_mds_coredump(target_rank['name']) |
| 675 | + |
| 676 | + state = self._dump_export_state(source) |
| 677 | + |
| 678 | + self.assertTrue(type(state['tid']) is int) |
| 679 | + self.assertEqual(state['path'], path) |
| 680 | + self.assertEqual(state['state'], self.EXPORT_STATES[state_index]) |
| 681 | + self.assertEqual(state['peer'], target) |
| 682 | + |
| 683 | + return state |
| 684 | + |
| 685 | + def _test_state_history(self, state): |
| 686 | + history = state['state_history'] |
| 687 | + self.assertTrue(type(history) is dict) |
| 688 | + size = 0 |
| 689 | + for name in self.EXPORT_STATES: |
| 690 | + self.assertTrue(type(history[name]) is dict) |
| 691 | + size += 1 |
| 692 | + if name == state['state']: |
| 693 | + break |
| 694 | + self.assertEqual(len(history), size) |
| 695 | + |
| 696 | + def _test_freeze_tree(self, state, waiters): |
| 697 | + self.assertTrue(type(state['freeze_tree_time']) is float) |
| 698 | + self.assertEqual(state['unfreeze_tree_waiters'], waiters) |
| 699 | + |
| 700 | + def test_discovering(self): |
| 701 | + state = self._test_base('/test', 0, 1, 1, 1) |
| 702 | + |
| 703 | + self._test_state_history(state) |
| 704 | + self._test_freeze_tree(state, 0) |
| 705 | + |
| 706 | + self.assertEqual(state['last_cum_auth_pins'], 0) |
| 707 | + self.assertEqual(state['num_remote_waiters'], 0) |
| 708 | + |
| 709 | + def test_prepping(self): |
| 710 | + client_id = self.mount_a.get_global_id() |
| 711 | + |
| 712 | + state = self._test_base('/test', 0, 1, 3, 3) |
| 713 | + |
| 714 | + self._test_state_history(state) |
| 715 | + self._test_freeze_tree(state, 0) |
| 716 | + |
| 717 | + self.assertEqual(state['flushed_clients'], [client_id]) |
| 718 | + self.assertTrue(type(state['warning_ack_waiting']) is list) |
| 719 | + |
| 720 | + def test_exporting(self): |
| 721 | + state = self._test_base('/test', 0, 1, 5, 5) |
| 722 | + |
| 723 | + self._test_state_history(state) |
| 724 | + self._test_freeze_tree(state, 0) |
| 725 | + |
| 726 | + self.assertTrue(type(state['notify_ack_waiting']) is list) |
0 commit comments