feat(analytics): Introduce new Analytics class

Author: julienbourdeau

CHANGELOG.md

@@ -5,6 +5,11 @@
 
 <Contributors, please add your changes below this line>
 
+* Introduce AB Testing feature - PR [#408](https://github.com/algolia/algoliasearch-client-php/pull/#408)
+    List/Create/Stop/Delete AB Tests programmatically
+    Introduce new Analytics object, wrapper around the
+    [Analytics API](https://www.algolia.com/doc/rest-api/analytics/) (more methods to come).
+
 * 2 methods about taskID initially available in the `Index` moved to the `Client`. 
     You could get some taskID from the engine without necessarily have an instance of Index, 
     instead of instantiating an index that you won't need, you can now call wait_task and get_task_status on the client.

algoliasearch/analytics.py

@@ -0,0 +1,46 @@
+# -*- coding: utf-8 -*-
+from algoliasearch.helpers import AlgoliaException
+from .helpers import safe
+
+
+class Analytics:
+
+    def __init__(self, client, _transport):
+        self.client = client
+        self._transport = _transport
+        self._transport.read_hosts = ['analytics.algolia.com']
+        self._transport.write_hosts = ['analytics.algolia.com']
+
+    def get_ab_tests(self, args=None):
+        params = {'offest': 0, 'limit': 10}
+        if args is not None:
+            params.update(args)
+
+        return self._req('/2/abtests', 'GET', params)
+
+    def add_ab_test(self, ab_test):
+        return self._req('/2/abtests', 'POST', None, ab_test)
+
+    def get_ab_test(self, ab_test_id):
+        if ab_test_id == '':
+            raise AlgoliaException('ab_test_id cannot be empty')
+
+        return self._req('/2/abtests/%s' % safe(ab_test_id), 'GET')
+
+    def stop_ab_test(self, ab_test_id):
+        if ab_test_id == '':
+            raise AlgoliaException('ab_test_id cannot be empty')
+
+        return self._req('/2/abtests/%s/stop' % safe(ab_test_id), 'POST')
+
+    def delete_ab_test(self, ab_test_id):
+        if ab_test_id == '':
+            raise AlgoliaException('ab_test_id cannot be empty')
+
+        return self._req('/2/abtests/%s' % safe(ab_test_id), 'DELETE')
+
+    def wait_task(self, index_name, task_id, time_before_retry=100, request_options=None):
+        return self.client.wait_task(index_name, task_id, time_before_retry, request_options)
+
+    def _req(self, path, method, params=None, data=None):
+        return self._transport.req(False, path, method, params, data)

algoliasearch/client.py

@@ -28,6 +28,7 @@
 import random
 import sys
 import time
+import copy
 from platform import python_version
 
 try:
@@ -37,6 +38,7 @@
 
 from .version import VERSION
 from .index import Index
+from .analytics import Analytics
 
 from .transport import Transport
 from .helpers import deprecated
@@ -377,6 +379,9 @@ def init_index(self, index_name):
         """
         return Index(self, index_name)
 
+    def init_analytics(self):
+        return Analytics(self, copy.deepcopy(self._transport))
+
     @deprecated
     def listUserKeys(self):
         return self.list_user_keys()

tests/conftest.py

@@ -14,6 +14,11 @@ def client():
     return create_client()
 
 
+@pytest.fixture
+def analytics():
+    return create_client().init_analytics()
+
+
 @pytest.fixture
 def index(client):
     idx = create_index(client)

tests/test_analytics.py

@@ -0,0 +1,39 @@
+from __future__ import print_function
+from datetime import datetime
+
+from algoliasearch.helpers import AlgoliaException
+
+
+
+def test_ab_test_features(analytics, double_indexes):
+    ab_test = {
+        'name': "Python client integration test",
+        'variants': [
+            {'index': double_indexes[0].index_name, 'trafficPercentage': 90},
+            {'index': double_indexes[1].index_name, 'trafficPercentage': 10},
+        ],
+        'endAt':  datetime.utcnow().replace(day=29).strftime("%Y-%m-%dT%H:%M:%SZ"),
+    }
+    response = analytics.add_ab_test(ab_test)
+    analytics.wait_task(response['index'], response['taskID'])
+    ab_test_id = response['abTestID']
+
+    response = analytics.get_ab_test(ab_test_id)
+    assert response['name'] == ab_test['name']
+
+    response = analytics.get_ab_tests()
+    assert len(response['abtests']) == response['count']
+    assert response['total'] >= response['count']
+
+    response = analytics.stop_ab_test(ab_test_id)
+    analytics.wait_task(response['index'], response['taskID'])
+    response = analytics.get_ab_test(ab_test_id)
+    assert response['status'] == 'stopped'
+
+    response = analytics.delete_ab_test(ab_test_id)
+    analytics.wait_task(response['index'], response['taskID'])
+    try:
+        response = analytics.get_ab_test(ab_test_id)
+        assert False
+    except AlgoliaException as e:
+        assert True