statsig-node-vercel 0.0.1-beta.1 → 0.0.1-beta.2

package/EdgeConfigDataAdapter.ts

@@ -1,4 +1,4 @@
-import { AdapterResponse, IDataAdapter, DataAdapterKey } from 'statsig-node';
+import { AdapterResponse, IDataAdapter } from 'statsig-node';
 import { get } from '@vercel/edge-config';
 
 type Data = {
@@ -20,8 +20,8 @@ export class EdgeConfigDataAdapter implements IDataAdapter {
   }
 
   // eslint-disable-next-line @typescript-eslint/require-await
-  public async get(key: DataAdapterKey): Promise<AdapterResponse> {
-    if (key !== DataAdapterKey.Rulesets) {
+  public async get(key: string): Promise<AdapterResponse> {
+    if (key !== "statsig.cache") {
       return { error: new Error(`Edge Config Adapter Only Supports Config Specs`) };
     }
 
@@ -34,7 +34,7 @@ export class EdgeConfigDataAdapter implements IDataAdapter {
 
   // eslint-disable-next-line @typescript-eslint/require-await
   public async set(
-    key: DataAdapterKey,
+    key: string,
     value: string,
     time?: number | undefined,
   ): Promise<void> {
@@ -54,8 +54,8 @@ export class EdgeConfigDataAdapter implements IDataAdapter {
     
   }
 
-  public supportsPollingUpdatesFor(key: DataAdapterKey): boolean {
-    if (key === DataAdapterKey.Rulesets) {
+  public supportsPollingUpdatesFor(key: string): boolean {
+    if (key === "statsig.cache") {
       return this.supportConfigSpecPolling;
     }
     return false;

package/__tests__/EdgeConfigDataAdapter.test.ts

@@ -1,6 +1,5 @@
 import { EdgeConfigDataAdapter } from '../EdgeConfigDataAdapter';
 import fetchMock from "jest-fetch-mock"
-import { DataAdapterKey } from 'statsig-node';
 
 describe('Validate edge config adapter functionality', () => {
   const dataAdapter = new EdgeConfigDataAdapter(
@@ -9,10 +8,7 @@ describe('Validate edge config adapter functionality', () => {
 
   beforeEach(async () => {
     fetchMock.enableMocks()
-    fetchMock.mockResponse(JSON.stringify({
-        body: '{ "items": { "statsig-companyid": {} }}',
-      })
-    );
+    fetchMock.mockResponse('"test123"');
     await dataAdapter.initialize();
   }); 
 
@@ -20,11 +16,11 @@ describe('Validate edge config adapter functionality', () => {
     await dataAdapter.shutdown();
   });
 
-  test('Simple get/set', async () => {
-    const { result: gates } = await dataAdapter.get(DataAdapterKey.Rulesets);
+  test('Simple get', async () => {
+    const { result: gates } = await dataAdapter.get("statsig.cache");
     if (gates == null) {
       return;
     }
-    expect(gates).toEqual('test123');
+    expect(gates).toEqual('"test123"');
   });
 })

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "statsig-node-vercel",
-  "version": "0.0.1-beta.1",
+  "version": "0.0.1-beta.2",
   "description": "",
   "main": "index.js",
   "scripts": {
@@ -13,7 +13,7 @@
   "license": "ISC",
   "dependencies": {
     "@vercel/edge-config": "^0.1.0-canary.15",
-    "statsig-node": "5.1.0-beta.2"
+    "statsig-node": "5.1.0-beta.5"
   },
   "devDependencies": {
     "@babel/core": "^7.18.13",