fix: return correct data when using select field

Author: thomasconner

src/ipdata.ts

@@ -176,7 +176,11 @@ export default class IPData {
 
     try {
       const response = await axios.get(url, { params });
-      const data = { ...response.data, status: response.status };
+      let data = { ...response.data, status: response.status };
+
+      if (selectField) {
+        data = { [selectField]: response.data, status: response.status };
+      }
 
       if ((!selectField || !fields) && this.useCache) {
         cache.set(ip || DEFAULT_IP, data);

test/ipdata.test.ts

@@ -69,10 +69,10 @@ describe('lookup()', () => {
       await expect(ipdata.lookup(null, field)).rejects.toThrowError(`${field} is not a valid field.`);
     });
 
-    it('should return a response with only the field', async () => {
+    it.only('should return a response with only the field', async () => {
       const field = 'is_eu';
       const info = await ipdata.lookup(null, field);
-      expect(info).toHaveProperty('select_field', false);
+      expect(info).toHaveProperty(field, false);
       expect(info).toHaveProperty('status');
     });
   });