chore:comment addresed

rudderlabs · Dec 22, 2023 · 55d5fd8 · 55d5fd8
1 parent 11da01f
commit 55d5fd8
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/src/v0/destinations/am/util.test.js b/src/v0/destinations/am/util.test.js
@@ -1,6 +1,15 @@
 const { getUnsetObj } = require('./utils');
 
 describe('getUnsetObj', () => {
+  it("should return undefined when 'message.integrations.Amplitude.fieldsToUnset' is not array", () => {
+    const message = {
+      integrations: {
+        Amplitude: { fieldsToUnset: 'field_name' },
+      },
+    };
+    const result = getUnsetObj(message);
+    expect(result).toBeUndefined();
+  });
   it("should return undefined when 'message.integrations.Amplitude.fieldsToUnset' is undefined", () => {
     const message = {
       integrations: {

diff --git a/src/v0/destinations/am/utils.js b/src/v0/destinations/am/utils.js
@@ -101,7 +101,7 @@ const getEventId = (payload, sourceKey) => {
 const getUnsetObj = (message) => {
   const fieldsToUnset = get(message, 'integrations.Amplitude.fieldsToUnset');
   let unsetObject;
-  if (fieldsToUnset) {
+  if (fieldsToUnset && Array.isArray(fieldsToUnset)) {
     unsetObject = Object.fromEntries(fieldsToUnset.map((field) => [field, '-']));
   }