def test_get_classification_no_classification():
""" Test attempting to get the classification element when none is set. """
inst = MemoryDetectionElement(0)
with pytest.raises(NoDetectionError,
match="Missing or empty classification for in-memory "
"detection with UUID 0"):
inst.get_classification()
def test_get_classification_empty_classification():
""" Test attempting to get the classification element when it is empty. """
c_elem = mock.MagicMock(spec_set=ClassificationElement)
# Simulate an empty ClassificationElement
c_elem.has_classifications.return_value = False
inst = MemoryDetectionElement(0)
inst._classification = c_elem
with pytest.raises(NoDetectionError,
match="Missing or empty classification for in-memory "
"detection with UUID 0"):
inst.get_classification()
def test_get_classification():
""" Test successfully getting the detection classification element. """
c_elem = mock.MagicMock(spec_set=ClassificationElement)
# Simulate a populated ClassificationElement
c_elem.has_classifications.return_value = True
inst = MemoryDetectionElement(0)
inst._classification = c_elem
assert inst.get_classification() == c_elem
