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()
