From 58a1c13883f815197c76487b382eddf2eed67207 Mon Sep 17 00:00:00 2001 From: Kiran Date: Tue, 17 Oct 2023 19:32:38 +0530 Subject: changes for update functionality for entities.. Issue-ID: TEST-412 Change-Id: I598c289f87fff80323ef979600df45d4fa4591dd Signed-off-by: Kiran --- tests/test_aai_customer.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) (limited to 'tests/test_aai_customer.py') diff --git a/tests/test_aai_customer.py b/tests/test_aai_customer.py index 260520c..1597ac2 100644 --- a/tests/test_aai_customer.py +++ b/tests/test_aai_customer.py @@ -392,6 +392,23 @@ def test_customer_delete(mock_send): customer.url ) +@mock.patch.object(Customer, "send_message") +@mock.patch.object(Customer, "send_message_json") +def test_customer_update(mock_send_json, mock_send): + """Test Customer's update method.""" + mock_send_json.return_value = SIMPLE_CUSTOMER_2 + customer = Customer.update("generic", "generic", "INFRA") + assert customer.global_customer_id == "generic" + assert customer.subscriber_name == "generic" + assert customer.subscriber_type == "INFRA" + assert customer.resource_version is not None + + customer = Customer.create("generic", "generic", "INFRA", service_subscriptions=["test-service-type"]) + assert customer.global_customer_id == "generic" + assert customer.subscriber_name == "generic" + assert customer.subscriber_type == "INFRA" + assert customer.resource_version is not None + @mock.patch.object(Customer, "send_message") def test_delete_subscribed_service(mock_send): -- cgit 1.2.3-korg