fix tests

Author: fuziontech

posthog/test/client.py

@@ -208,7 +208,7 @@ def mock_post_fn(*args, **kwargs):
 
         # the post function should be called 2 times, with a batch size of 10
         # each time.
-        with mock.patch('posthog.consumer.post', side_effect=mock_post_fn) \
+        with mock.patch('posthog.consumer.batch_post', side_effect=mock_post_fn) \
                 as mock_post:
             for _ in range(20):
                 client.identify('distinct_id', {'trait': 'value'})
@@ -269,7 +269,7 @@ def test_feature_enabled_request(self, patch_get):
             'id': 1,
             'name': 'Beta Feature',
             'key': 'beta-feature',
-            'is_simple_flag': False,
+            'is_simple_flag': True,
             'rollout_percentage': 100
         }]
         self.assertTrue(client.feature_enabled('beta-feature', 'distinct_id'))

posthog/test/consumer.py

@@ -60,7 +60,7 @@ def test_flush_interval(self):
         flush_interval = 0.3
         consumer = Consumer(q, TEST_API_KEY, flush_at=10,
                             flush_interval=flush_interval)
-        with mock.patch('posthog.consumer.post') as mock_post:
+        with mock.patch('posthog.consumer.batch_post') as mock_post:
             consumer.start()
             for i in range(0, 3):
                 track = {
@@ -80,7 +80,7 @@ def test_multiple_uploads_per_interval(self):
         flush_at = 10
         consumer = Consumer(q, TEST_API_KEY, flush_at=flush_at,
                             flush_interval=flush_interval)
-        with mock.patch('posthog.consumer.post') as mock_post:
+        with mock.patch('posthog.consumer.batch_post') as mock_post:
             consumer.start()
             for i in range(0, flush_at * 2):
                 track = {
@@ -110,7 +110,7 @@ def mock_post(*args, **kwargs):
                 raise expected_exception
         mock_post.call_count = 0
 
-        with mock.patch('posthog.consumer.post',
+        with mock.patch('posthog.consumer.batch_post',
                         mock.Mock(side_effect=mock_post)):
             track = {
                 'type': 'track',

posthog/test/request.py

@@ -18,11 +18,11 @@ def test_valid_request(self):
         self.assertEqual(res.status_code, 200)
 
     def test_invalid_request_error(self):
-        self.assertRaises(Exception, post, 'testsecret',
+        self.assertRaises(Exception, batch_post, 'testsecret',
                           'https://t.posthog.com', False, '[{]')
 
     def test_invalid_host(self):
-        self.assertRaises(Exception, post, 'testsecret',
+        self.assertRaises(Exception, batch_post, 'testsecret',
                           't.posthog.com/', batch=[])
 
     def test_datetime_serialization(self):