From f7160b0dd89e4c748ddb6199e90bb3c7b25fe0e5 Mon Sep 17 00:00:00 2001
From: Joshua Klein <klein.joshua.a@gmail.com>
Date: Wed, 7 Sep 2022 17:04:29 -0400
Subject: [PATCH] added test

---
 .../microclients/attribute_microclient.py     |  4 +-
 .../tests/resources/test_traffic_type.py      | 48 +++++++++++++++++++
 2 files changed, 50 insertions(+), 2 deletions(-)

diff --git a/splitapiclient/microclients/attribute_microclient.py b/splitapiclient/microclients/attribute_microclient.py
index 52fc6bb..84cba81 100644
--- a/splitapiclient/microclients/attribute_microclient.py
+++ b/splitapiclient/microclients/attribute_microclient.py
@@ -31,7 +31,7 @@ class AttributeMicroClient:
             'query_string': [],
             'response': True,
         },
-        'import_from_json': {
+        'import_attributes_from_json': {
             'method': 'POST',
             'url_template': 'schema/ws/{workspaceId}/trafficTypes/{trafficTypeId}/attribute:bulk',
             'headers': [{
@@ -155,7 +155,7 @@ def import_attributes_from_json(self, workspaceId, trafficTypeId, data):
         :returns: bool
         '''
         res = self._http_client.make_request(
-            self._endpoint['import_from_json'],
+            self._endpoint['import_attributes_from_json'],
             body=data,
             workspaceId = workspaceId,
             trafficTypeId = trafficTypeId,
diff --git a/splitapiclient/tests/resources/test_traffic_type.py b/splitapiclient/tests/resources/test_traffic_type.py
index 2c1284e..83bfc84 100644
--- a/splitapiclient/tests/resources/test_traffic_type.py
+++ b/splitapiclient/tests/resources/test_traffic_type.py
@@ -395,3 +395,51 @@ def test_add_identities(self, mocker):
         assert [s.to_dict() for s in res3.successful] == data
         assert isinstance(res3.failed, list)
         assert isinstance(res3.metadata, dict)
+
+    def test_import_JSON(self, mocker):
+        '''
+        '''
+        data = {
+            'id': 'a1',
+            'trafficTypeId': '1',
+            'displayName': 'dn1',
+            'isSearchable': None,
+            'dataType': 'string',
+            'description': 'd1',
+            'workspaceId': 'ws_id'
+        }
+        http_client_mock = mocker.Mock(spec=BaseHttpClient)
+        http_client_mock.make_request.return_value = data
+        tt1 = TrafficType(
+            {
+                'id': '1',
+                'displayAttributeId': 'asd',
+                'name': 'n1',
+            },
+            'ws_id',
+            http_client_mock
+        )
+        attrib_data = [ {
+                            "id": "anAttribute2",
+                            "displayName": "An Attribute2",
+                            "description": "my description here",
+                            "dataType": "string",
+                            "suggestedValues": [
+                                "suggested",
+                                "values"
+                            ]
+                        } ]
+
+
+        tt1.import_attributes_from_json(attrib_data)
+
+
+        http_client_mock.make_request.assert_called_once_with(
+            AttributeMicroClient._endpoint['import_attributes_from_json'],
+            body=attrib_data,
+            workspaceId = data['workspaceId'],
+            trafficTypeId = data['trafficTypeId']
+            
+        )
+
+