1 files changed, 51 insertions, 0 deletions
diff --git a/tests/test_aai_service_resource.py b/tests/test_aai_service_resource.py
new file mode 100644
index 0000000..9056299
--- /dev/null
+++ b/tests/test_aai_service_resource.py
@@ -0,0 +1,51 @@
+from unittest.mock import patch, PropertyMock
+
+from onap_data_provider.resources.aai_service_resource import AaiService, AaiServiceResource, ResourceNotFound
+
+
+AAI_SERVICE_DATA = {
+    "service-id": "123",
+    "service-description": "123"
+}
+
+
+@patch("onap_data_provider.resources.aai_service_resource.AaiService.get_all")
+def test_aai_service_resource_aai_resource(mock_aai_service_get_all):
+    mock_aai_service_get_all.side_effect = ResourceNotFound
+    mock_aai_service_get_all.return_value = iter([])
+    aai_service_resource = AaiServiceResource(AAI_SERVICE_DATA)
+    assert aai_service_resource.aai_service is None
+    mock_aai_service_get_all.side_effect = None
+    mock_aai_service_get_all.return_value = iter([AaiService(service_id="123", service_description="123", resource_version="123")])
+    assert aai_service_resource.aai_service is not None
+
+
+@patch(
+    "onap_data_provider.resources.aai_service_resource.AaiServiceResource.aai_service",
+    new_callable=PropertyMock,
+)
+def test_aai_service_resource_exists(mock_aai_service):
+    mock_aai_service.return_value = None
+    aai_service_resource = AaiServiceResource(AAI_SERVICE_DATA)
+    assert aai_service_resource.exists is False
+    mock_aai_service.return_value = 1  # Anything but not None
+    assert aai_service_resource.exists is True
+
+
+@patch(
+    "onap_data_provider.resources.aai_service_resource.AaiServiceResource.exists",
+    new_callable=PropertyMock,
+)
+@patch("onap_data_provider.resources.aai_service_resource.AaiService.create")
+def test_aai_service_resource_create(mock_aai_service_create, mock_exists):
+    mock_exists.return_value = True
+    aai_service_resource = AaiServiceResource(AAI_SERVICE_DATA)
+    aai_service_resource.create()
+    mock_aai_service_create.assert_not_called()
+
+    mock_exists.return_value = False
+    aai_service_resource.create()
+    mock_aai_service_create.assert_called_once_with(
+        service_id="123",
+        service_description="123"
+    )