90 lines
4.0 KiB
Python
90 lines
4.0 KiB
Python
"""Resource parser/fixture cases; not executed in the current batch."""
|
|
from proxydoctor.resources import cgroup_limits, collect, cpu_ticks, pressure, steal_delta
|
|
|
|
|
|
def fixture_reader(files):
|
|
return lambda path: files.get(path)
|
|
|
|
|
|
def test_nested_v2_uses_strictest_visible_ancestor():
|
|
read = fixture_reader({
|
|
'/proc/self/cgroup': '0::/tenant/app\n',
|
|
'/proc/self/mountinfo': '1 0 0:1 / /sys/fs/cgroup rw - cgroup2 cgroup rw',
|
|
'/sys/fs/cgroup/tenant/app/cpu.max': '200000 100000',
|
|
'/sys/fs/cgroup/tenant/cpu.max': '50000 100000',
|
|
'/sys/fs/cgroup/cpu.max': 'max 100000',
|
|
'/sys/fs/cgroup/tenant/app/memory.max': 'max',
|
|
'/sys/fs/cgroup/tenant/memory.max': '536870912',
|
|
})
|
|
result = cgroup_limits(read)
|
|
assert result['cpu_quota_cores'] == 0.5
|
|
assert result['memory_limit_bytes'] == 512*1024**2
|
|
|
|
|
|
def test_v1_controller_mount_and_unlimited_sentinel():
|
|
read = fixture_reader({
|
|
'/proc/self/cgroup': '2:cpu,cpuacct:/app\n3:memory:/app',
|
|
'/proc/self/mountinfo': '1 0 0:1 / /cg/cpu rw - cgroup cgroup rw,cpu,cpuacct\n2 0 0:2 / /cg/memory rw - cgroup cgroup rw,memory',
|
|
'/cg/cpu/app/cpu.cfs_quota_us': '25000',
|
|
'/cg/cpu/app/cpu.cfs_period_us': '100000',
|
|
'/cg/memory/app/memory.limit_in_bytes': '9223372036854771712',
|
|
})
|
|
result = cgroup_limits(read)
|
|
assert result['cpu_quota_cores'] == 0.25
|
|
assert result['memory_limit_bytes'] is None
|
|
|
|
|
|
def test_subtree_mount_maps_membership_without_escaping_visible_root():
|
|
read = fixture_reader({
|
|
'/proc/self/cgroup': '0::/tenant/app',
|
|
'/proc/self/mountinfo': '1 0 0:1 /tenant /sys/fs/cgroup rw - cgroup2 cgroup rw',
|
|
'/sys/fs/cgroup/app/cpu.max': '100000 100000',
|
|
'/sys/fs/cgroup/cpu.max': '50000 100000',
|
|
})
|
|
result = cgroup_limits(read)
|
|
assert result['cpu_quota_cores'] == 0.5
|
|
assert all('/tenant/' not in row['source'] for row in result['observations'])
|
|
|
|
|
|
def test_missing_or_invalid_cgroup_is_unknown_not_unlimited():
|
|
assert cgroup_limits(fixture_reader({}))['cpu_quota_cores'] is None
|
|
assert pressure('some avg10=nan avg60=0 avg300=0 total=0') is None
|
|
assert pressure('some avg10=0.25 avg60=0 avg300=0 total=12')['some']['avg10'] == 0.25
|
|
|
|
|
|
def test_steal_uses_delta_and_does_not_double_count_guest():
|
|
before = cpu_ticks('cpu 100 0 0 100 0 0 0 0 90 0')
|
|
after = cpu_ticks('cpu 120 0 0 170 0 0 0 10 100 0')
|
|
assert steal_delta(before, after) == 10
|
|
assert steal_delta(after, before) is None
|
|
assert steal_delta(before, before) is None
|
|
|
|
|
|
def test_collect_keeps_numeric_units_and_passive_sampling():
|
|
snapshots = iter(['cpu 100 0 0 100 0 0 0 0', 'cpu 120 0 0 170 0 0 0 10'])
|
|
files = {'/proc/meminfo': 'MemTotal: 1024 kB\nMemAvailable: 512 kB\nSwapTotal: 256 kB\nSwapFree: 128 kB'}
|
|
times = iter([10.0, 10.2])
|
|
waits = []
|
|
result = collect(read=lambda path: next(snapshots) if path == '/proc/stat' else files.get(path),
|
|
system='Linux', cpu_count=lambda: 8, affinity=lambda: {0,1},
|
|
sleep=waits.append, clock=lambda: next(times))
|
|
assert waits == [0.2]
|
|
assert result['metrics']['visible_cpu_upper_bound_cores'] == 2
|
|
assert result['metrics']['swap_used_bytes'] == 128*1024
|
|
assert result['metrics']['steal_percent'] == 10
|
|
|
|
|
|
def test_non_linux_does_not_read_proc_or_wait():
|
|
def unexpected(*args):
|
|
raise AssertionError('Unsupported system must not probe Linux files')
|
|
assert collect(read=unexpected, sleep=unexpected, system='Windows')['status'] == 'unsupported'
|
|
|
|
|
|
def test_container_without_memory_limit_does_not_claim_host_ram():
|
|
files = {'/.dockerenv': '', '/proc/meminfo': 'MemTotal: 67108864 kB'}
|
|
result = collect(read=fixture_reader(files), system='Linux', sleep=lambda _: None,
|
|
cpu_count=lambda: 64, affinity=lambda: {0,1})
|
|
assert result['allocation']['memory_limit_bytes'] is None
|
|
assert result['allocation']['dedicated_cpu'] == 'unknown'
|
|
assert result['metrics']['memory_total_bytes'] == 64*1024**3
|