better testing and org

Author: fuziontech

posthog/client.py

@@ -280,8 +280,7 @@ def feature_enabled(self, key, distinct_id, default=False):
                         "distinct_id": distinct_id,
                         "personal_api_key": self.personal_api_key,
                     }
-                    request = decide(self.api_key, self.host, timeout=10, **request_data)
-                    resp_data = request.json()
+                    resp_data = decide(self.api_key, self.host, timeout=10, **request_data)
                     response = key in resp_data['featureFlags']
                 except Exception as e:
                     response = default

posthog/request.py

@@ -52,7 +52,7 @@ def decide(api_key, host=None, gzip=False, timeout=15, **kwargs):
     res = post(api_key, host, '/decide/', gzip, timeout, **kwargs)
     if res.status_code == 200:
         log.debug('Feature flags decided successfully')
-        return res
+        return res.json()
     try:
         payload = res.json()
         log.debug('received response: %s', payload)

posthog/test/client.py

@@ -259,7 +259,7 @@ def test_feature_enabled_simple(self, patch_get):
         }]
         self.assertTrue(client.feature_enabled('beta-feature', 'distinct_id'))
 
-    @mock.patch('posthog.client.get')
+    @mock.patch('posthog.client.decide')
     def test_feature_enabled_request(self, patch_get):
         patch_get.return_value = {
             'featureFlags': ['beta-feature']
@@ -269,7 +269,7 @@ def test_feature_enabled_request(self, patch_get):
             'id': 1,
             'name': 'Beta Feature',
             'key': 'beta-feature',
-            'is_simple_flag': True,
+            'is_simple_flag': False,
             'rollout_percentage': 100
         }]
         self.assertTrue(client.feature_enabled('beta-feature', 'distinct_id'))