class InternalIdMapperTest(unittest.TestCase):
    """
    Tests for `InternalIdMapper`.
    """
    _INTERNAL_IDS = [123, 456, 789]

    def setUp(self):
        self._mapper = MockInternalIdMapper()

    def test_get_by_id_with_value(self):
        self._mapper.get_by_id(InternalIdMapperTest._INTERNAL_IDS[0])
        self._mapper._get_by_property_value_sequence.assert_called_once_with(
            Property.INTERNAL_ID, [InternalIdMapperTest._INTERNAL_IDS[0]])

    def test_get_by_id_with_list(self):
        self._mapper.get_by_id(InternalIdMapperTest._INTERNAL_IDS)
        self._mapper._get_by_property_value_sequence.assert_called_once_with(
            Property.INTERNAL_ID, InternalIdMapperTest._INTERNAL_IDS)
Beispiel #2
0
class InternalIdMapperTest(unittest.TestCase):
    """
    Tests for `InternalIdMapper`.
    """
    _INTERNAL_IDS = [123, 456, 789]

    def setUp(self):
        self._mapper = MockInternalIdMapper()

    def test_get_by_id_with_value(self):
        self._mapper.get_by_id(InternalIdMapperTest._INTERNAL_IDS[0])
        self._mapper._get_by_property_value_sequence.assert_called_once_with(
            Property.INTERNAL_ID, [InternalIdMapperTest._INTERNAL_IDS[0]])

    def test_get_by_id_with_list(self):
        self._mapper.get_by_id(InternalIdMapperTest._INTERNAL_IDS)
        self._mapper._get_by_property_value_sequence.assert_called_once_with(
            Property.INTERNAL_ID, InternalIdMapperTest._INTERNAL_IDS)
 def setUp(self):
     self._mapper = MockInternalIdMapper()
Beispiel #4
0
 def setUp(self):
     self._mapper = MockInternalIdMapper()