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()