def test_exit(self):
storage = ContentStorage()
storage.close = Mock()
storage.__exit__()
storage.close.assert_called_once_with()
def test_enter(self):
storage = ContentStorage()
storage.open = Mock()
inst = storage.__enter__()
storage.open.assert_called_once_with()
self.assertEqual(inst, storage)
def test_open(self):
storage = ContentStorage()
storage.open()
def test_close(self):
storage = ContentStorage()
storage.close()
def test_abstract(self):
storage = ContentStorage()
self.assertRaises(NotImplementedError, storage.put, None, None)
self.assertRaises(NotImplementedError, storage.get, None)