chore: Support migration payload contract tests. (#262)

launchdarkly · Aug 30, 2023 · 760567d · 760567d
1 parent ee4ebbf
commit 760567d
Showing 1 changed file with 29 additions and 8 deletions.
diff --git a/contract-tests/sdkClientEntity.js b/contract-tests/sdkClientEntity.js
@@ -80,6 +80,13 @@ function getExecution(order) {
   }
 }
 
+function makeMigrationPostOptions(payload) {
+  if (payload) {
+    return { body: payload };
+  }
+  return {};
+}
+
 export async function newSdkClientEntity(options) {
   const c = {};
   const log = Log(options.tag);
@@ -170,33 +177,45 @@ export async function newSdkClientEntity(options) {
           latencyTracking: migrationOperation.trackLatency,
           errorTracking: migrationOperation.trackErrors,
           check: migrationOperation.trackConsistency ? (a, b) => a === b : undefined,
-          readNew: async () => {
+          readNew: async (payload) => {
             try {
-              const res = await got.post(migrationOperation.newEndpoint, {});
+              const res = await got.post(
+                migrationOperation.newEndpoint,
+                makeMigrationPostOptions(payload),
+              );
               return LDMigrationSuccess(res.body);
             } catch (err) {
               return LDMigrationError(err.message);
             }
           },
-          writeNew: async () => {
+          writeNew: async (payload) => {
             try {
-              const res = await got.post(migrationOperation.newEndpoint, {});
+              const res = await got.post(
+                migrationOperation.newEndpoint,
+                makeMigrationPostOptions(payload),
+              );
               return LDMigrationSuccess(res.body);
             } catch (err) {
               return LDMigrationError(err.message);
             }
           },
-          readOld: async () => {
+          readOld: async (payload) => {
             try {
-              const res = await got.post(migrationOperation.oldEndpoint, {});
+              const res = await got.post(
+                migrationOperation.oldEndpoint,
+                makeMigrationPostOptions(payload),
+              );
               return LDMigrationSuccess(res.body);
             } catch (err) {
               return LDMigrationError(err.message);
             }
           },
-          writeOld: async () => {
+          writeOld: async (payload) => {
             try {
-              const res = await got.post(migrationOperation.oldEndpoint, {});
+              const res = await got.post(
+                migrationOperation.oldEndpoint,
+                makeMigrationPostOptions(payload),
+              );
               return LDMigrationSuccess(res.body);
             } catch (err) {
               return LDMigrationError(err.message);
@@ -210,6 +229,7 @@ export async function newSdkClientEntity(options) {
               migrationOperation.key,
               migrationOperation.context,
               migrationOperation.defaultStage,
+              migrationOperation.payload,
             );
             if (res.success) {
               return { result: res.result };
@@ -222,6 +242,7 @@ export async function newSdkClientEntity(options) {
               migrationOperation.key,
               migrationOperation.context,
               migrationOperation.defaultStage,
+              migrationOperation.payload,
             );
 
             if (res.authoritative.success) {