iframe-proxy | Sunbelt Computer Software

69 lines (55 loc) · 2 KB

from __future__ import unicode_literals
import unittest
from prometheus_client import CollectorRegistry, PlatformCollector
class TestPlatformCollector(unittest.TestCase):
    def setUp(self):
        self.registry = CollectorRegistry()
        self.platform = _MockPlatform()
    def test_python_info(self):
        PlatformCollector(registry=self.registry, platform=self.platform)
        self.assertLabels("python_info", {
            "version": "python_version",
            "implementation": "python_implementation",
            "major": "pvt_major",
            "minor": "pvt_minor",
            "patchlevel": "pvt_patchlevel"
    def test_system_info_java(self):
        self.platform._system = "Java"
        PlatformCollector(registry=self.registry, platform=self.platform)
        self.assertLabels("python_info", {
            "version": "python_version",
            "implementation": "python_implementation",
            "major": "pvt_major",
            "minor": "pvt_minor",
            "patchlevel": "pvt_patchlevel",
            "jvm_version": "jv_release",
            "jvm_release": "vm_release",
            "jvm_vendor": "vm_vendor",
            "jvm_name": "vm_name"
    def assertLabels(self, name, labels):
        for metric in self.registry.collect():
            for n, l, value in metric.samples:
                if n == name:
                    assert l == labels
                    return
        assert False
class _MockPlatform(object):
    def __init__(self):
        self._system = "system"
    def python_version_tuple(self):
        return "pvt_major", "pvt_minor", "pvt_patchlevel"
    def python_version(self):
        return "python_version"
    def python_implementation(self):
        return "python_implementation"
    def system(self):
        return self._system
    def java_ver(self):
        return (
            "jv_release",
            "jv_vendor",
            ("vm_name", "vm_release", "vm_vendor"),
            ("os_name", "os_version", "os_arch")

Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

test_platform_collector.py

Latest commit

History

test_platform_collector.py

File metadata and controls