X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=blobdiff_plain;f=yardstick%2Ftests%2Funit%2Fnetwork_services%2Fvnf_generic%2Fvnf%2Ftest_base.py;h=ce8205a4702f8ea475356695a2fd612c86656b6d;hb=639287faab56f702806d5bff09e38788d165ac9b;hp=43e5ac839b61b19abb3c2abd0fb6d2ea5b6fd91c;hpb=438f7b5100e05ed668a0ea90674008a391eb5784;p=yardstick.git

diff --git a/yardstick/tests/unit/network_services/vnf_generic/vnf/test_base.py b/yardstick/tests/unit/network_services/vnf_generic/vnf/test_base.py
index 43e5ac839..ce8205a47 100644
--- a/yardstick/tests/unit/network_services/vnf_generic/vnf/test_base.py
+++ b/yardstick/tests/unit/network_services/vnf_generic/vnf/test_base.py
@@ -264,7 +264,7 @@ class GenericTrafficGenTestCase(unittest.TestCase):
                 }
         tg = _DummyGenericTrafficGen('name', vnfd)
         tg._mq_producer = mock.Mock()
-        tg._mq_producer.get_id.return_value = 'fake_id'
+        tg._mq_producer.id = 'fake_id'
         self.assertEqual('fake_id', tg.get_mq_producer_id())