From 19f1a8e6027e9bcadfe6b4ff41ff6bd52cba56c1 Mon Sep 17 00:00:00 2001
From: Lee ByeongJun <lbj199874@gmail.com>
Date: Mon, 9 Dec 2024 15:00:02 +0900
Subject: [PATCH] test: position_update

---
 pool/position_update_test.gno | 85 +++++++++++++++++++++++++++++++++++
 1 file changed, 85 insertions(+)
 create mode 100644 pool/position_update_test.gno

diff --git a/pool/position_update_test.gno b/pool/position_update_test.gno
new file mode 100644
index 000000000..24ca4cb55
--- /dev/null
+++ b/pool/position_update_test.gno
@@ -0,0 +1,85 @@
+package pool
+
+import (
+	"testing"
+
+	"std"
+
+	"gno.land/p/demo/uassert"
+
+	"gno.land/r/gnoswap/v1/consts"
+	i256 "gno.land/p/gnoswap/int256"
+	u256 "gno.land/p/gnoswap/uint256"
+)
+
+func TestUpdatePosition(t *testing.T) {
+	poolParams := &createPoolParams{
+		token0Path:    "token0",
+		token1Path:    "token1",
+		fee:          500,
+		tickSpacing:  10,
+		sqrtPriceX96: u256.MustFromDecimal("1000000000000000000"), // 1.0
+	}
+	p := newPool(poolParams)
+
+	tests := []struct {
+		name           string
+		positionParams ModifyPositionParams
+		expectLiquidity *u256.Uint
+	}{
+		{
+			name: "add new position",
+			positionParams: ModifyPositionParams{
+				owner:          consts.POSITION_ADDR,
+				tickLower:      -100,
+				tickUpper:      100,
+				liquidityDelta: i256.MustFromDecimal("1000000"),
+			},
+			expectLiquidity: u256.MustFromDecimal("1000000"),
+		},
+		{
+			name: "add liquidity to existing position",
+			positionParams: ModifyPositionParams{
+				owner:          consts.POSITION_ADDR,
+				tickLower:      -100,
+				tickUpper:      100,
+				liquidityDelta: i256.MustFromDecimal("500000"),
+			},
+			expectLiquidity: u256.MustFromDecimal("1500000"),
+		},
+		{
+			name: "remove liquidity from position",
+			positionParams: ModifyPositionParams{
+				owner:          consts.POSITION_ADDR,
+				tickLower:      -100,
+				tickUpper:      100,
+				liquidityDelta: i256.MustFromDecimal("-500000"),
+			},
+			expectLiquidity: u256.MustFromDecimal("1000000"),
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			position := p.updatePosition(tt.positionParams)
+			
+			if !position.liquidity.Eq(tt.expectLiquidity) {
+				t.Errorf("liquidity mismatch: expected %s, got %s", 
+					tt.expectLiquidity.ToString(), 
+					position.liquidity.ToString())
+			}
+
+			if !tt.positionParams.liquidityDelta.IsZero() {
+				lowerTick := p.ticks[tt.positionParams.tickLower]
+				upperTick := p.ticks[tt.positionParams.tickUpper]
+
+				if !lowerTick.initialized {
+					t.Error("lower tick not initialized")
+				}
+				if !upperTick.initialized {
+					t.Error("upper tick not initialized")
+				}
+			}
+		})
+	}
+}