diff options
Diffstat (limited to 'tests/test_aai_site_resource.py')
-rw-r--r-- | tests/test_aai_site_resource.py | 35 |
1 files changed, 34 insertions, 1 deletions
diff --git a/tests/test_aai_site_resource.py b/tests/test_aai_site_resource.py index 68dd556..a0d4a12 100644 --- a/tests/test_aai_site_resource.py +++ b/tests/test_aai_site_resource.py @@ -11,7 +11,7 @@ # 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.mock import patch +from unittest.mock import patch, MagicMock from onapsdk.aai.network.site_resource import SiteResource @@ -55,3 +55,36 @@ def test_site_resource_create(mock_get_by_site_resource_id, mock_send_message): SiteResource.create("123") mock_send_message.assert_called_once() assert mock_get_by_site_resource_id.called_once_with("123") + +@patch("onapsdk.aai.network.site_resource.SiteResource.add_relationship") +def test_site_resource_link_to_complex(mock_add_relationship): + cmplx = MagicMock(physical_location_id="test-complex-physical-location-id", + url="test-complex-url") + site_resource = SiteResource("test-site-resource") + site_resource.link_to_complex(cmplx) + mock_add_relationship.assert_called_once() + relationship = mock_add_relationship.call_args[0][0] + assert relationship.related_to == "complex" + assert relationship.related_link == "test-complex-url" + assert relationship.relationship_label == "org.onap.relationships.inventory.Uses" + assert relationship.relationship_data == [{ + "relationship-key": "complex.physical-location-id", + "relationship-value": "test-complex-physical-location-id", + }] + + +@patch("onapsdk.aai.network.site_resource.SiteResource.add_relationship") +def test_site_resource_link_to_site_resource(mock_add_relationship): + site_resource_rel = MagicMock(site_resource_id="test-site-resource-id", + url="test-site-resource-url") + site_resource = SiteResource("test-site-resource") + site_resource.link_to_site_resource(site_resource_rel) + mock_add_relationship.assert_called_once() + relationship = mock_add_relationship.call_args[0][0] + assert relationship.related_to == "site-resource" + assert relationship.related_link == "test-site-resource-url" + assert relationship.relationship_label == "org.onap.relationships.inventory.Supports" + assert relationship.relationship_data == [{ + "relationship-key": "site_resource.site-resource-id", + "relationship-value": "test-site-resource-id", + }] |