Merge pull request #384 from InjectiveLabs/chore/add_positions_in_market_endpoint

aarmoa · web-flow · commit c86686fd2a0b · 2025-06-23T16:05:26.000+02:00
[chore] add positions in market endpoint
diff --git a/examples/chain_client/exchange/query/65_PositionsInMarket.py b/examples/chain_client/exchange/query/65_PositionsInMarket.py
@@ -0,0 +1,21 @@
+import asyncio
+import json
+
+from pyinjective.async_client_v2 import AsyncClient
+from pyinjective.core.network import Network
+
+
+async def main() -> None:
+    # select network: local, testnet, mainnet
+    network = Network.testnet()
+
+    # initialize grpc client
+    client = AsyncClient(network)
+
+    market_id = "0x17ef48032cb24375ba7c2e39f384e56433bcab20cbee9a7357e4cba2eb00abe6"
+    positions = await client.fetch_chain_positions_in_market(market_id=market_id)
+    print(json.dumps(positions, indent=2))
+
+
+if __name__ == "__main__":
+    asyncio.get_event_loop().run_until_complete(main())
diff --git a/pyinjective/async_client_v2.py b/pyinjective/async_client_v2.py
@@ -734,6 +734,9 @@ async def fetch_chain_derivative_market(
     async def fetch_chain_positions(self) -> Dict[str, Any]:
         return await self.chain_exchange_v2_api.fetch_positions()
 
+    async def fetch_chain_positions_in_market(self, market_id: str) -> Dict[str, Any]:
+        return await self.chain_exchange_v2_api.fetch_positions_in_market(market_id=market_id)
+
     async def fetch_chain_subaccount_positions(self, subaccount_id: str) -> Dict[str, Any]:
         return await self.chain_exchange_v2_api.fetch_subaccount_positions(subaccount_id=subaccount_id)
 
diff --git a/pyinjective/client/chain/grpc/chain_grpc_exchange_v2_api.py b/pyinjective/client/chain/grpc/chain_grpc_exchange_v2_api.py
@@ -378,6 +378,12 @@ async def fetch_positions(self) -> Dict[str, Any]:
 
         return response
 
+    async def fetch_positions_in_market(self, market_id: str) -> Dict[str, Any]:
+        request = exchange_query_pb.QueryPositionsInMarketRequest(market_id=market_id)
+        response = await self._execute_call(call=self._stub.PositionsInMarket, request=request)
+
+        return response
+
     async def fetch_subaccount_positions(self, subaccount_id: str) -> Dict[str, Any]:
         request = exchange_query_pb.QuerySubaccountPositionsRequest(subaccount_id=subaccount_id)
         response = await self._execute_call(call=self._stub.SubaccountPositions, request=request)
diff --git a/tests/client/chain/grpc/configurable_exchange_v2_query_servicer.py b/tests/client/chain/grpc/configurable_exchange_v2_query_servicer.py
@@ -41,6 +41,7 @@ def __init__(self):
         self.derivative_market_address_responses = deque()
         self.subaccount_trade_nonce_responses = deque()
         self.positions_responses = deque()
+        self.positions_in_market_responses = deque()
         self.subaccount_positions_responses = deque()
         self.subaccount_position_in_market_responses = deque()
         self.subaccount_effective_position_in_market_responses = deque()
@@ -221,6 +222,11 @@ async def SubaccountTradeNonce(
     async def Positions(self, request: exchange_query_pb.QueryPositionsRequest, context=None, metadata=None):
         return self.positions_responses.pop()
 
+    async def PositionsInMarket(
+        self, request: exchange_query_pb.QueryPositionsInMarketRequest, context=None, metadata=None
+    ):
+        return self.positions_in_market_responses.pop()
+
     async def SubaccountPositions(
         self, request: exchange_query_pb.QuerySubaccountPositionsRequest, context=None, metadata=None
     ):
diff --git a/tests/client/chain/grpc/test_chain_grpc_exchange_v2_api.py b/tests/client/chain/grpc/test_chain_grpc_exchange_v2_api.py
@@ -23,7 +23,7 @@ def exchange_servicer():
     return ConfigurableExchangeV2QueryServicer()
 
 
-class TestChainGrpcBankApi:
+class TestChainGrpcExchangeV2Api:
     @pytest.mark.asyncio
     async def test_fetch_exchange_params(
         self,
@@ -1554,6 +1554,50 @@ async def test_fetch_positions(
 
         assert positions == expected_positions
 
+    @pytest.mark.asyncio
+    async def test_fetch_positions_in_market(
+        self,
+        exchange_servicer,
+    ):
+        position = exchange_pb.Position(
+            isLong=True,
+            quantity="1000000000000000",
+            entry_price="2000000000000000000",
+            margin="2000000000000000000000000000000000",
+            cumulative_funding_entry="4000000",
+        )
+        derivative_position = exchange_pb.DerivativePosition(
+            subaccount_id="0x17ef48032cb24375ba7c2e39f384e56433bcab20000000000000000000000000",
+            market_id="0x17ef48032cb24375ba7c2e39f384e56433bcab20cbee9a7357e4cba2eb00abe6",
+            position=position,
+        )
+        exchange_servicer.positions_in_market_responses.append(
+            exchange_query_pb.QueryPositionsInMarketResponse(state=[derivative_position])
+        )
+
+        api = self._api_instance(servicer=exchange_servicer)
+
+        positions = await api.fetch_positions_in_market(
+            market_id="0x17ef48032cb24375ba7c2e39f384e56433bcab20cbee9a7357e4cba2eb00abe6",
+        )
+        expected_positions = {
+            "state": [
+                {
+                    "subaccountId": derivative_position.subaccount_id,
+                    "marketId": derivative_position.market_id,
+                    "position": {
+                        "isLong": position.isLong,
+                        "quantity": position.quantity,
+                        "entryPrice": position.entry_price,
+                        "margin": position.margin,
+                        "cumulativeFundingEntry": position.cumulative_funding_entry,
+                    },
+                },
+            ],
+        }
+
+        assert positions == expected_positions
+
     @pytest.mark.asyncio
     async def test_fetch_subaccount_positions(
         self,
