def test_raw_returns_function_when_exists(self): c = MedleyContainer() c._keys = Mock() c._keys.__contains__ = Mock(return_value=True) c._raw = {'foo': self.foo} self.assertEqual(c.raw('foo'), self.foo)
def test_getitem_returns_raw_function_when_id_in_raw_set(self): c = MedleyContainer() c._keys = Mock(__contains__=Mock(return_value=True)) c._protected = Mock(__contains__=Mock(return_value=False)) c._raw = {'foo': self.foo} c._values = {'foo': self.foo} self.assertEqual(c.__getitem__('foo'), self.foo)
def test_raw_returns_raw_function_from_values_attr_when_it_does_not_exist_in_raw_attr( self): c = MedleyContainer() c._keys = Mock() c._keys.__contains__ = Mock(return_value=True) c._raw = Mock() c._raw.__contains__ = Mock(return_value=False) c._values = {'foo': self.foo} self.assertEqual(c.raw('foo'), self.foo) c._keys.__contains__.assert_called_with('foo') c._raw.__contains__.assert_called_with('foo')
def test_delitem_deletes_from_everywhere_wwhen_id_is_function(self): c = MedleyContainer() c._keys = Mock(__contains__=Mock(return_value=True), discard=Mock()) c._protected = Mock(discard=Mock()) c._factories = Mock(discard=Mock()) c._frozen = Mock(discard=Mock()) c._values = {'foo': self.foo} c._raw = {'foo': self.foo} self.assertFalse(c.__delitem__('foo')) c._factories.discard.assert_called_with(self.foo) c._protected.discard.assert_called_with(self.foo) self.assertEqual(c._values, {}) self.assertEqual(c._raw, {}) c._frozen.discard.assert_called_with('foo') c._keys.discard.assert_called_with('foo')