async def test_delete_not_found(self):
m = kerastest.new_model()
self.storage.delete = asynctest.AsyncMagicMock()
cache = await Cache.new(storage=self.storage)
await cache.delete(m.name, m.tag)
self.storage.delete.assert_called()
self.assertNotIn(m.key, cache.models)
async def test_save(self):
m1 = kerastest.new_model()
self.storage.save = asynctest.AsyncMagicMock(return_value=m1)
cache = await Cache.new(storage=self.storage)
m2 = await cache.save(m1.name, m1.tag, None)
self.storage.save.assert_called()
self.assertEqual(m1, m2)
self.assertIn(m1.key, cache.models)
async def test_load_not_found(self):
m1 = kerastest.new_model()
self.storage.load = asynctest.AsyncMagicMock(return_value=m1)
cache = await Cache.new(storage=self.storage)
m2 = await cache.load(m1.name, m1.tag)
self.storage.load.assert_called()
self.assertIn(m1.key, cache.models)
self.assertEqual(m1, m2)
async def test_load(self):
m1 = kerastest.new_model()
m1.model = unittest.mock.MagicMock()
self.storage.load = asynctest.AsyncMagicMock()
cache = await Cache.new(storage=self.storage)
cache.models[m1.key] = m1
m2 = await cache.load(m1.name, m1.tag)
self.storage.load.assert_not_called()
self.assertIn(m1.key, cache.models)
self.assertEqual(m1, m2)
async def handle_request(self,
method: str,
path: str,
resp: aiohttp.web.Response = None,
) -> client.Model:
resp = aiohttp.web.Response() if resp is None else resp
handler_mock = asynctest.AsyncMagicMock(return_value=resp)
app = aiohttp.web.Application()
route = aiohttp.web.RouteDef(
method, path, asynctest.unittest_handler(handler_mock), {})
app.add_routes([route])
async with aiohttptest.TestServer(app) as server:
service_url = str(server.make_url(""))
yield client.Model(client.Session(service_url))
handler_mock.assert_called()