Cesarus85--mjolnir-vr/tools/test_audit_object_admission.py

import unittest

from tools.audit_object_admission import classify_units


def _unit(name, complete, total_functions, matched_functions, total_data, matched_data):
    return {
        "name": name,
        "metadata": {"complete": complete},
        "measures": {
            "total_functions": total_functions,
            "matched_functions": matched_functions,
            "total_data": total_data,
            "matched_data": matched_data,
        },
    }


class ObjectAdmissionAuditTests(unittest.TestCase):
    def test_lists_unmarked_zero_gap_unit_without_admitting_it(self):
        raw = {"units": [_unit("source/a", False, 2, 1, 8, 8)]}
        strict = {"units": [_unit("source/a", False, 2, 2, 8, 8)]}

        result = classify_units(raw, strict)

        self.assertEqual([item["unit"] for item in result["candidates"]], ["source/a"])
        self.assertEqual(
            result["candidates"][0]["decision"],
            "audit-coff-ownership-before-admission",
        )
        self.assertEqual(result["contradicted"], [])

    def test_lists_complete_label_with_strict_gap_as_contradicted(self):
        raw = {"units": [_unit("source/b", True, 3, 3, 12, 12)]}
        strict = {"units": [_unit("source/b", False, 3, 2, 12, 12)]}

        result = classify_units(raw, strict)

        self.assertEqual(result["candidates"], [])
        self.assertEqual(
            [item["unit"] for item in result["contradicted"]], ["source/b"]
        )

    def test_does_not_list_incomplete_or_already_complete_units(self):
        units = [
            _unit("source/incomplete", False, 2, 1, 0, 0),
            _unit("source/complete", True, 2, 2, 0, 0),
        ]

        result = classify_units({"units": units}, {"units": units})

        self.assertEqual(result, {"candidates": [], "contradicted": []})

    def test_does_not_list_zero_content_placeholder_as_candidate(self):
        unit = _unit("libs/empty", False, 0, 0, 0, 0)

        result = classify_units({"units": [unit]}, {"units": [unit]})

        self.assertEqual(result, {"candidates": [], "contradicted": []})


if __name__ == "__main__":
    unittest.main()