803441 - fix cli unit test for org subscriptions

daviddavis · Mar 15, 2012 · 77ff0c0 · 77ff0c0
1 parent d93e13a
commit 77ff0c0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cli/test/organization_subscription_list_test.py b/cli/test/organization_subscription_list_test.py
@@ -37,6 +37,7 @@ def setUp(self):
         self.mock_options(self.OPTIONS)
         self.mock_printer()
 
+        self.mock(self.action.api, 'organization', test_data.ORGS[0])
         self.mock(self.action.api, 'pools', [test_data.POOL])
         self.mock(self.action.productApi, 'show', test_data.PRODUCTS[0])
 
@@ -45,7 +46,7 @@ def tearDown(self):
 
     def test_it_finds_all_pools_for_organization(self):
         self.action.run()
-        self.action.api.pools.assert_called_once_with(self.ORGANIZATION)
+        self.action.api.pools.assert_called_once_with(test_data.ORGS[0]["cp_key"])
 
     def test_extract_sla_from_product(self):
         self.assertEqual(test_data.SLA_VALUE, self.action.extract_sla_from_product(test_data.PRODUCTS[0]))