#19 Get usage report business unit

Signed-off-by: Uilian Ries <uilianries@gmail.com>

Author: uilianries

bintray/bintray.py

@@ -3000,3 +3000,30 @@ def get_usage_report_for_package(self, subject, repo, package=None, start_pos=50
         response = self._requester.post(url, json=json_data, params=params)
         self._logger.info("Search successfully")
         return response
+
+    def get_usage_report_grouped_by_business_unit(self, subject, business_unit=None, from_date=None,
+                                                  to_date=None):
+        """ Get monthly download and storage usage report, according to the specified date range
+            and grouped by business unit. Report can be requested for a subject or for a specific
+            subject business unit.
+
+            Security: Authenticated user with 'admin' permission.
+
+        :param subject: repository owner
+        :param business_unit: business unit name
+        :param from_date: initial date range ISO8601 (yyyy-MM-dd'T'HH:mm:ss.SSSZ)
+        :param to_date: end date range ISO8601 (yyyy-MM-dd'T'HH:mm:ss.SSSZ)
+        :return: download details
+        """
+        url = "{}/usage/business_unit_usage/{}".format(Bintray.BINTRAY_URL, subject)
+        if business_unit:
+            url += "/{}".format(business_unit)
+        json_data = {}
+        if from_date:
+            json_data["from"] = from_date
+        if to_date:
+            json_data["to"] = to_date
+
+        response = self._requester.post(url, json=json_data)
+        self._logger.info("Search successfully")
+        return response

tests/test_statistics.py

@@ -65,3 +65,14 @@ def test_get_usage_report_for_package():
         error_message = str(error)
     assert "Could not POST (403): This resource is only available for Premium repositories" \
            == error_message
+
+
+def test_get_usage_report_grouped_by_business_unit():
+    bintray = Bintray()
+    error_message = ""
+    try:
+        bintray.get_usage_report_grouped_by_business_unit("uilianries", "generic")
+    except Exception as error:
+        error_message = str(error)
+    assert "Could not POST (403): This resource is only available for Enterprise users" \
+           == error_message