]>
git.proxmox.com Git - ceph.git/blob - ceph/qa/tasks/cephfs/test_mantle.py
1 from io
import StringIO
3 from tasks
.cephfs
.cephfs_test_case
import CephFSTestCase
7 log
= logging
.getLogger(__name__
)
8 failure
= "using old balancer; mantle failed for balancer="
9 success
= "mantle balancer version changed: "
11 class TestMantle(CephFSTestCase
):
12 def start_mantle(self
):
13 self
.wait_for_health_clear(timeout
=30)
14 self
.fs
.set_max_mds(2)
15 self
.wait_until_equal(lambda: len(self
.fs
.get_active_names()), 2, 30,
16 reject_fn
=lambda v
: v
> 2 or v
< 1)
18 for m
in self
.fs
.get_active_names():
19 self
.fs
.mds_asok(['config', 'set', 'debug_objecter', '20'], mds_id
=m
)
20 self
.fs
.mds_asok(['config', 'set', 'debug_ms', '0'], mds_id
=m
)
21 self
.fs
.mds_asok(['config', 'set', 'debug_mds', '0'], mds_id
=m
)
22 self
.fs
.mds_asok(['config', 'set', 'debug_mds_balancer', '5'], mds_id
=m
)
24 def push_balancer(self
, obj
, lua_code
, expect
):
25 self
.fs
.mon_manager
.raw_cluster_cmd_result('fs', 'set', self
.fs
.name
, 'balancer', obj
)
26 self
.fs
.radosm(["put", obj
, "-"], stdin
=StringIO(lua_code
))
27 with self
.assert_cluster_log(failure
+ obj
+ " " + expect
):
28 log
.info("run a " + obj
+ " balancer that expects=" + expect
)
30 def test_version_empty(self
):
32 expect
= " : (2) No such file or directory"
34 ret
= self
.fs
.mon_manager
.raw_cluster_cmd_result('fs', 'set', self
.fs
.name
, 'balancer')
35 assert(ret
== 22) # EINVAL
37 self
.fs
.mon_manager
.raw_cluster_cmd_result('fs', 'set', self
.fs
.name
, 'balancer', " ")
38 with self
.assert_cluster_log(failure
+ " " + expect
): pass
40 def test_version_not_in_rados(self
):
42 expect
= failure
+ "ghost.lua : (2) No such file or directory"
43 self
.fs
.mon_manager
.raw_cluster_cmd_result('fs', 'set', self
.fs
.name
, 'balancer', "ghost.lua")
44 with self
.assert_cluster_log(expect
): pass
46 def test_balancer_invalid(self
):
48 expect
= ": (22) Invalid argument"
50 lua_code
= "this is invalid lua code!"
51 self
.push_balancer("invalid.lua", lua_code
, expect
)
53 lua_code
= "BAL_LOG()"
54 self
.push_balancer("invalid_log.lua", lua_code
, expect
)
56 lua_code
= "BAL_LOG(0)"
57 self
.push_balancer("invalid_log_again.lua", lua_code
, expect
)
59 def test_balancer_valid(self
):
61 lua_code
= "BAL_LOG(0, \"test\")\nreturn {3, 4}"
62 self
.fs
.mon_manager
.raw_cluster_cmd_result('fs', 'set', self
.fs
.name
, 'balancer', "valid.lua")
63 self
.fs
.radosm(["put", "valid.lua", "-"], stdin
=StringIO(lua_code
))
64 with self
.assert_cluster_log(success
+ "valid.lua"):
65 log
.info("run a valid.lua balancer")
67 def test_return_invalid(self
):
69 expect
= ": (22) Invalid argument"
71 lua_code
= "return \"hello\""
72 self
.push_balancer("string.lua", lua_code
, expect
)
75 self
.push_balancer("number.lua", lua_code
, expect
)
77 lua_code
= "return {}"
78 self
.push_balancer("dict_empty.lua", lua_code
, expect
)
80 lua_code
= "return {\"this\", \"is\", \"a\", \"test\"}"
81 self
.push_balancer("dict_of_strings.lua", lua_code
, expect
)
83 lua_code
= "return {3, \"test\"}"
84 self
.push_balancer("dict_of_mixed.lua", lua_code
, expect
)
86 lua_code
= "return {3}"
87 self
.push_balancer("not_enough_numbers.lua", lua_code
, expect
)
89 lua_code
= "return {3, 4, 5, 6, 7, 8, 9}"
90 self
.push_balancer("too_many_numbers.lua", lua_code
, expect
)
92 def test_dead_osd(self
):
94 expect
= " : (110) Connection timed out"
96 # kill the OSDs so that the balancer pull from RADOS times out
97 osd_map
= json
.loads(self
.fs
.mon_manager
.raw_cluster_cmd('osd', 'dump', '--format=json-pretty'))
98 for i
in range(0, len(osd_map
['osds'])):
99 self
.fs
.mon_manager
.raw_cluster_cmd_result('osd', 'down', str(i
))
100 self
.fs
.mon_manager
.raw_cluster_cmd_result('osd', 'out', str(i
))
102 # trigger a pull from RADOS
103 self
.fs
.mon_manager
.raw_cluster_cmd_result('fs', 'set', self
.fs
.name
, 'balancer', "valid.lua")
105 # make the timeout a little longer since dead OSDs spam ceph -w
106 with self
.assert_cluster_log(failure
+ "valid.lua" + expect
, timeout
=30):
107 log
.info("run a balancer that should timeout")
110 for i
in range(0, len(osd_map
['osds'])):
111 self
.fs
.mon_manager
.raw_cluster_cmd_result('osd', 'in', str(i
))