@drift-labs/sdk 2.66.0-beta.0 → 2.66.0-beta.2

package/VERSION

@@ -1 +1 @@
-2.66.0-beta.0
+2.66.0-beta.2

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@drift-labs/sdk",
-	"version": "2.66.0-beta.0",
+	"version": "2.66.0-beta.2",
 	"main": "lib/index.js",
 	"types": "lib/index.d.ts",
 	"author": "crispheaney",

package/tests/amm/test.ts

@@ -614,8 +614,14 @@ describe('AMM Tests', () => {
 		// console.log(termsSuiExample);
 		assert(termsSuiExample.effectiveLeverageCapped <= 1.000001);
 		assert(termsSuiExample.inventorySpreadScale == 1.00007);
-		assert(termsSuiExample.longSpread == 259073);
-		assert(termsSuiExample.shortSpread == 3712);
+		assert(
+			termsSuiExample.longSpread == 269818,
+			`SUI long spread got ${termsSuiExample.longSpread}`
+		);
+		assert(
+			termsSuiExample.shortSpread == 3920,
+			`SUI short spread got ${termsSuiExample.shortSpread}`
+		);
 
 		// reset amm reserves/peg to balanced values s.t. liquidity/price is the same
 		// to avoid error prone int math
@@ -674,7 +680,26 @@ describe('AMM Tests', () => {
 		);
 
 		console.log(termsSuiExampleMod2);
-		// assert(_.isEqual(termsSuiExampleMod2, termsSuiExampleMod1));
+		assert(
+			_.isEqual(
+				termsSuiExampleMod2.maxTargetSpread,
+				termsSuiExampleMod1.maxTargetSpread
+			)
+		);
+		assert(
+			_.isEqual(
+				termsSuiExampleMod2.shortSpreadwPS,
+				termsSuiExampleMod1.shortSpreadwPS
+			)
+		);
+		assert(
+			_.isEqual(
+				termsSuiExampleMod2.longSpreadwPS,
+				termsSuiExampleMod1.longSpreadwPS
+			)
+		);
+
+		// note: effectiveLeverage as currently implemented is sensitive to peg change
 	});
 
 	it('Spread Reserves (with offset)', () => {

package/tests/dlob/helpers.ts

@@ -24,6 +24,7 @@ import {
 	SPOT_MARKET_CUMULATIVE_INTEREST_PRECISION,
 	SPOT_MARKET_WEIGHT_PRECISION,
 	PRICE_PRECISION,
+	DataAndSlot,
 } from '../../src';
 
 export const mockPerpPosition: PerpPosition = {
@@ -639,6 +640,10 @@ export class MockUserMap implements UserMapInterface {
 		return undefined;
 	}
 
+	public getWithSlot(_key: string): DataAndSlot<User> | undefined {
+		return undefined;
+	}
+
 	public async mustGet(_key: string): Promise<User> {
 		return new User({
 			driftClient: this.driftClient,
@@ -646,6 +651,16 @@ export class MockUserMap implements UserMapInterface {
 		});
 	}
 
+	public async mustGetWithSlot(_key: string): Promise<DataAndSlot<User>> {
+		return {
+			data: new User({
+				driftClient: this.driftClient,
+				userAccountPublicKey: PublicKey.default,
+			}),
+			slot: 0,
+		};
+	}
+
 	public getUserAuthority(key: string): PublicKey | undefined {
 		return new PublicKey(
 			this.userAccountToAuthority.get(key) || PublicKey.default.toBase58()
@@ -657,4 +672,29 @@ export class MockUserMap implements UserMapInterface {
 	public values(): IterableIterator<User> {
 		return this.userMap.values();
 	}
+
+	public *valuesWithSlot(): IterableIterator<DataAndSlot<User>> {
+		for (const user of this.userMap.values()) {
+			yield {
+				data: user,
+				slot: 0,
+			};
+		}
+	}
+
+	public entries(): IterableIterator<[string, User]> {
+		return this.userMap.entries();
+	}
+
+	public *entriesWithSlot(): IterableIterator<[string, DataAndSlot<User>]> {
+		for (const [key, user] of this.userMap.entries()) {
+			yield [
+				key,
+				{
+					data: user,
+					slot: 0,
+				},
+			];
+		}
+	}
 }