def test_get_environment_id_multi_available_bad_name(self, mock_sys_exit): def env_list(*args, **kwargs): return [{ "id": "1234", "name": "somename" }, { "id": "5678", "name": "othername" }] # Avoid writing to stderr mock_sys_exit.side_effect = SystemExit(-1) StubUEP.getEnvironmentList = env_list cp = StubUEP() cp.supports_resource = Mock(return_value=True) rc = RegisterCommand() rc.options = Mock() rc.options.activation_keys = None rc._prompt_for_environment = Mock(return_value="not_an_env") try: rc._get_environment_id(cp, 'owner', None) self.fail("No Exception Raised") except SystemExit: pass
def test_get_environment_id_one_available(self): def env_list(*args, **kwargs): return [{"id": "1234", "name": "somename"}] StubUEP.getEnvironmentList = env_list cp = StubUEP() cp.supports_resource = Mock(return_value=True) rc = RegisterCommand() rc.options = Mock() rc.options.activation_keys = None env_id = rc._get_environment_id(cp, "owner", None) expected = "1234" self.assertEquals(expected, env_id)
def test_get_environment_id_one_available(self): def env_list(*args, **kwargs): return [{"id": "1234", "name": "somename"}] StubUEP.getEnvironmentList = env_list cp = StubUEP() cp.supports_resource = Mock(return_value=True) rc = RegisterCommand() rc.options = Mock() rc.options.activation_keys = None env_id = rc._get_environment_id(cp, 'owner', None) expected = "1234" self.assertEquals(expected, env_id)
def setUp(self, mock_update): SubManFixture.setUp(self) # Setup mock product and entitlement certs: self.prod_dir = StubProductDirectory( pids=[INST_PID_1, INST_PID_2, INST_PID_3, INST_PID_4]) self.ent_dir = StubEntitlementDirectory([ StubEntitlementCertificate(PROD_2, ent_id=ENT_ID_2), StubEntitlementCertificate(PROD_1, ent_id=ENT_ID_1), StubEntitlementCertificate(product=PROD_4, stacking_id=STACK_1, ent_id=ENT_ID_4), # entitled, but not installed StubEntitlementCertificate(StubProduct('not_installed_product', name="Some Product"), ent_id="SomeSubId"), ]) self.mock_uep = StubUEP() self.status_mgr = EntitlementStatusCache() self.status_mgr.load_status = Mock(return_value=SAMPLE_COMPLIANCE_JSON) self.status_mgr.write_cache = Mock() inj.provide(inj.ENTITLEMENT_STATUS_CACHE, self.status_mgr) inj.provide(inj.PROD_DIR, self.prod_dir) inj.provide(inj.ENT_DIR, self.ent_dir) self.sorter = CertSorter() self.sorter.is_registered = Mock(return_value=True)
def test_get_environment_id_multi_available_bad_name(self, mock_sys_exit): def env_list(*args, **kwargs): return [{"id": "1234", "name": "somename"}, {"id": "5678", "name": "othername"}] # Avoid writing to stderr mock_sys_exit.side_effect = SystemExit(-1) StubUEP.getEnvironmentList = env_list cp = StubUEP() cp.supports_resource = Mock(return_value=True) rc = RegisterCommand() rc.options = Mock() rc.options.activation_keys = None rc._prompt_for_environment = Mock(return_value="not_an_env") try: rc._get_environment_id(cp, "owner", None) self.fail("No Exception Raised") except SystemExit: pass
def test_library_no_longer_filtered(self): self.cc.cp = StubUEP() environments = [] environments.append({'name': 'JarJar'}) environments.append({'name': 'Library'}) environments.append({'name': 'library'}) environments.append({'name': 'Binks'}) self.cc.cp.setEnvironmentList(environments) results = self.cc._get_enviornments("Anikan") self.assertTrue(len(results) == 4)
def test_no_library(self): self.cc.cp = StubUEP() environments = [] environments.append({'name': 'JarJar'}) environments.append({'name': 'Library'}) environments.append({'name': 'library'}) environments.append({'name': 'Binks'}) self.cc.cp.setEnvironmentList(environments) results = self.cc._get_enviornments("Anikan") self.assertTrue(len(results) == 2) self.assertTrue(results[0]['name'] == 'JarJar') self.assertTrue(results[1]['name'] == 'Binks')
def setUp(self): expected_facts = {'fact1': 'one', 'fact2': 'two', 'system': 'Unknown', 'system.uuid': 'MOCKUUID'} self.expected_facts = expected_facts self.stub_facts = StubFacts(expected_facts) self.uep = StubUEP() self.consumer = Mock() self.consumer.uuid = "MOCKUUID" self.consumer.name = "MOCK CONSUMER" sys.stderr = MockStderr sys.stdout = MockStdout
def setUp(self): TestCliProxyCommand.setUp(self) self.cc.consumerIdentity = StubConsumerIdentity self.cc.cp = StubUEP()