(self, mock_client)
| 93 | |
| 94 | @patch(MODULE_TO_MOCK + ".CosmosDBSQLBackend._client") |
| 95 | def test_get(self, mock_client): |
| 96 | self.backend.get(b"mykey") |
| 97 | |
| 98 | mock_client.ReadDocument.assert_has_calls( |
| 99 | [call("dbs/celerydb/colls/celerycol/docs/mykey", |
| 100 | {"partitionKey": "mykey"}), |
| 101 | call().get("value")]) |
| 102 | |
| 103 | @patch(MODULE_TO_MOCK + ".CosmosDBSQLBackend._client") |
| 104 | def test_get_missing(self, mock_client): |