Fix a test

Azure · Jul 30, 2021 · b77b98b · b77b98b
1 parent d8ce878
commit b77b98b
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/sdk/cosmos/src/cosmos_entity.rs b/sdk/cosmos/src/cosmos_entity.rs
@@ -12,6 +12,13 @@ pub trait CosmosEntity<'a> {
     fn partition_key(&'a self) -> Self::Entity;
 }
 
+impl<'a> CosmosEntity<'a> for serde_json::Value {
+    type Entity = &'a Self;
+    fn partition_key(&'a self) -> Self::Entity {
+        self
+    }
+}
+
 /// Serialize the partition key in the format CosmosDB expects.
 pub(crate) fn serialize_partition_key<PK: Serialize>(pk: &PK) -> Result<String, serde_json::Error> {
     // this must be serialized as an array even tough CosmosDB supports only a sigle