1 files changed, 82 insertions, 0 deletions
diff --git a/tests/test_aai_line_of_business.py b/tests/test_aai_line_of_business.py
new file mode 100644
index 0000000..1bf4672
--- /dev/null
+++ b/tests/test_aai_line_of_business.py
@@ -0,0 +1,82 @@
+#   Copyright 2022 Orange, Deutsche Telekom AG
+#
+#   Licensed under the Apache License, Version 2.0 (the "License");
+#   you may not use this file except in compliance with the License.
+#   You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+#   Unless required by applicable law or agreed to in writing, software
+#   distributed under the License is distributed on an "AS IS" BASIS,
+#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#   See the License for the specific language governing permissions and
+#   limitations under the License.
+from unittest import mock
+
+from onapsdk.aai.business.line_of_business import LineOfBusiness
+
+
+LINES_OF_BUSINESS = {
+    "line-of-business": [
+        {
+            "line-of-business-name": "test-name",
+            "resource-version": "1234"
+        },
+        {
+            "line-of-business-name": "test-name2",
+            "resource-version": "4321"
+        }
+    ]
+}
+
+
+COUNT = {
+    "results":[
+        {
+            "line-of-business":1
+        }
+    ]
+}
+
+
+@mock.patch("onapsdk.aai.business.line_of_business.LineOfBusiness.send_message_json")
+def test_line_of_business_get_all(mock_send_message_json):
+    mock_send_message_json.return_value = {}
+    assert len(list(LineOfBusiness.get_all())) == 0
+
+    mock_send_message_json.return_value = LINES_OF_BUSINESS
+    lines_of_business = list(LineOfBusiness.get_all())
+    assert len(lines_of_business) == 2
+    lob1, lob2 = lines_of_business
+    assert lob1.name == "test-name"
+    assert lob1.resource_version == "1234"
+    assert lob2.name == "test-name2"
+    assert lob2.resource_version == "4321"
+
+
+@mock.patch("onapsdk.aai.business.line_of_business.LineOfBusiness.send_message_json")
+def test_line_of_business_get_by_name(mock_send):
+    LineOfBusiness.get_by_name(name="test-name")
+    mock_send.assert_called_once_with("GET",
+                                      "Get test-name line of business",
+                                      "https://aai.api.sparky.simpledemo.onap.org:30233/aai/v23/business/lines-of-business/line-of-business/test-name")
+
+
+@mock.patch("onapsdk.aai.business.line_of_business.LineOfBusiness.send_message")
+@mock.patch("onapsdk.aai.business.line_of_business.LineOfBusiness.get_by_name")
+def test_line_of_business_create(_, mock_send):
+    LineOfBusiness.create(name="test-name")
+    mock_send.assert_called_once_with("PUT",
+                                      "Declare A&AI line of business",
+                                      "https://aai.api.sparky.simpledemo.onap.org:30233/aai/v23/business/lines-of-business/line-of-business/test-name",
+                                      data='{\n    "line-of-business-name": "test-name"\n}')
+
+
+@mock.patch("onapsdk.aai.business.line_of_business.LineOfBusiness.send_message_json")
+def test_line_of_business_count(mock_send_message_json):
+    mock_send_message_json.return_value = COUNT
+    assert LineOfBusiness.count() == 1
+
+def test_line_of_business_url():
+    line_of_business = LineOfBusiness(name="test-lob", resource_version="123")
+    assert line_of_business.name in line_of_business.url