def test_md_check_array_membership_wrong_array(self, mock_uuid, mock_query, mock_examine): mdname = '/dev/md0' devices = ['/dev/vda', '/dev/vdb', '/dev/vdc', '/dev/vdd'] md_uuid = '93a73e10:427f280b:b7076c02:204b8f7a' md_dict = {'MD_UUID': '11111111:427f280b:b7076c02:204b8f7a'} mock_query.return_value = md_dict mock_uuid.return_value = md_uuid mock_examine.side_effect = [md_dict] * len(devices) with self.assertRaises(ValueError): mdadm.md_check_array_membership(mdname, devices)
def test_md_check_array_membership(self, mock_uuid, mock_query, mock_examine): mdname = '/dev/md0' devices = ['/dev/vda', '/dev/vdb', '/dev/vdc', '/dev/vdd'] md_uuid = '93a73e10:427f280b:b7076c02:204b8f7a' md_dict = {'MD_UUID': md_uuid} mock_query.return_value = md_dict mock_uuid.return_value = md_uuid mock_examine.side_effect = [md_dict] * len(devices) expected_calls = [] for dev in devices: expected_calls.append(call(dev, export=False)) rv = mdadm.md_check_array_membership(mdname, devices) self.assertEqual(rv, None) mock_uuid.assert_has_calls([call(mdname)]) mock_examine.assert_has_calls(expected_calls)