def test_product_cert_output(self):
with Capture() as cap:
command = CatCertCommandStub(certdata.PRODUCT_CERT_V1_0)
command.main(['will_use_stub'])
cert_output = cap.out
self.assert_string_equals(certdata.PRODUCT_CERT_V1_0_OUTPUT,
cert_output)
def test_omit_product_list(self):
with Capture() as cap:
command = CatCertCommandStub(certdata.ENTITLEMENT_CERT_V1_0)
command.main(["not_used.pem", "--no-products"])
cert_output = cap.out
self.assertTrue(cert_output.find("Product:\n") == -1,
"Products were not excluded from the output.")
def _main_help(self, args):
with Capture() as cap:
try:
self.cc.main(args)
except SystemExit, e:
# --help/-h returns 0
self.assertEquals(e.code, 0)
def test_consumerid_with_distributor_id(self):
def get_consumer(self, *args, **kwargs):
pass
with patch('rhsm.connection.UEPConnection',
new_callable=StubUEP) as mock_uep:
mock_uep.getConsumer = get_consumer
self.stub_cp_provider.basic_auth_cp = mock_uep
self._inject_mock_invalid_consumer()
cmd = RegisterCommand()
mock_uep.getConsumer = Mock(
return_value={
'uuid': '123123',
'type': {
'manifest': True
},
'idCert': {
'key': ''
}
})
self._inject_ipm()
with nested(Capture(silent=True),
self.assertRaises(SystemExit)) as e:
cmd.main([
'register', '--consumerid=TaylorSwift',
'--username=testuser1', '--password=password',
'--org=test_org'
])
self.assertEquals(e.code, os.EX_USAGE)
def test_already_registered_to_sat_6(self):
self._inject_mock_valid_consumer()
self.engine.options = self.create_options(five_to_six=True)
with Capture() as c:
self.assertRaises(SystemExit, self.engine.check_ok_to_proceed,
"some_name")
self.assertTrue("Satellite 6" in c.err)
def test_cert_v3_cat(self):
with Capture() as cap:
command = CatCertCommandStub(certdata.ENTITLEMENT_CERT_V3_0)
command.main(['will_use_stub'])
cert_output = cap.out
self.assert_string_equals(certdata.ENTITLEMENT_CERT_V3_0_OUTPUT,
cert_output)
def test_none_wrap_available_pool_id(self, mget_ents):
list_command = managercli.ListCommand()
def create_pool_list(*args, **kwargs):
return [{
'productName': 'dummy-name',
'productId': 'dummy-id',
'providedProducts': [],
'id': '888888888888',
'attributes': [{
'name': 'is_virt_only',
'value': 'false'
}],
'pool_type': 'Some Type',
'quantity': '4',
'service_level': '',
'service_type': '',
'contractNumber': '5',
'multi-entitlement': 'false',
'endDate': '',
'suggested': '2'
}]
mget_ents.return_value = create_pool_list()
with Capture() as cap:
list_command.main(['list', '--available'])
self.assertTrue('888888888888' in cap.out)
def test_a_msg(self):
msg = "some message"
with Capture() as cap:
try:
managercli.system_exit(1, msg)
except SystemExit:
pass
self.assertEquals("%s\n" % msg, cap.err)
def test_list_nonexistant_repos(self):
data = [Override('x', 'hello', 'world')]
with Capture() as cap:
self.cc._list(data, ['x', 'z'])
output = cap.out
self.assertTrue(re.search("Nothing is known about 'z'", output))
self.assertTrue(re.search('Repository: x', output))
self.assertTrue(re.search('\s+hello:\s+world', output))
def test_already_registered_to_rhsm(self):
self._inject_mock_valid_consumer()
self.engine.options = self.create_options(five_to_six=False)
with Capture() as c:
self.assertRaises(SystemExit, self.engine.check_ok_to_proceed,
"some_name")
self.assertTrue("Red Hat Subscription Management" in c.err)
self.assertTrue("access.redhat.com" in c.err)
def test_msgs(self):
msgs = ["a", "b", "c"]
with Capture() as cap:
try:
managercli.system_exit(1, msgs)
except SystemExit:
pass
self.assertEquals("%s\n" % ("\n".join(msgs)), cap.err)
def test_msg_unicode(self):
msgs = [u"\u2620 \u2603 \u203D"]
with Capture() as cap:
try:
managercli.system_exit(1, msgs)
except SystemExit:
pass
self.assertEquals("%s\n" % msgs[0].encode("utf8"), cap.err)
def test_msg_and_exception(self):
msgs = ["a", ValueError()]
with Capture() as cap:
try:
managercli.system_exit(1, msgs)
except SystemExit:
pass
self.assertEquals("%s\n\n" % msgs[0], cap.err)
def test_cat_manifest(self):
catman = CatManifestCommand()
catman.args = [_build_valid_manifest()]
with Capture() as cap:
catman._do_command()
self.assertEquals("", cap.err)
self.assert_string_equals(manifestdata.correct_manifest_output, cap.out)
def test_msg_and_exception_no_str(self):
class NoStrException(Exception):
pass
msgs = ["a", NoStrException()]
with Capture() as cap:
try:
managercli.system_exit(1, msgs)
except SystemExit:
pass
self.assertEquals("%s\n\n" % msgs[0], cap.err)
def test_list_specific_repos(self):
data = [
Override('x', 'hello', 'world'),
Override('z', 'greetings', 'mars')
]
with Capture() as cap:
self.cc._list(data, ['x'])
output = cap.out
self.assertTrue(re.search('Repository: x', output))
self.assertTrue(re.search('\s+hello:\s+world', output))
self.assertFalse(re.search('Repository: z', output))
def test_handle_collisions(self):
cmap = {
'1': {'cert-a-1.pem': ['chan1', 'chan2'], 'cert-b-1.pem': ['chan3']},
'2': {'cert-x-2.pem': ['chan4', 'chan5']},
'3': {'cert-m-3.pem': ['chanA'], 'cert-n-3.pem': ['chanB'], 'cert-o-3.pem': ['chanC']}
}
with Capture() as cap:
try:
self.engine.handle_collisions(cmap)
except SystemExit, e:
self.assertEquals(e.code, 1)
else:
def test_cat_manifest(self):
catman = CatManifestCommand()
parser = OptionParser()
parser.add_option("--no-content")
(options, args) = parser.parse_args([])
catman.options = options
catman.args = [_build_valid_manifest()]
with Capture() as cap:
catman._do_command()
self.assertEquals("", cap.err)
self.assert_string_equals(manifestdata.correct_manifest_output, cap.out)
def test_list_function(self):
data = [
Override('x', 'hello', 'world'),
Override('x', 'blast-off', 'space'),
Override('y', 'goodbye', 'earth'),
Override('z', 'greetings', 'mars')
]
with Capture() as cap:
self.cc._list(data, None)
output = cap.out
self.assertTrue(re.search('Repository: x', output))
self.assertTrue(re.search('\s+hello:\s+world', output))
self.assertTrue(re.search('\s+blast-off:\s+space', output))
self.assertTrue(re.search('Repository: y', output))
self.assertTrue(re.search('\s+goodbye:\s+earth', output))
self.assertTrue(re.search('Repository: z', output))
self.assertTrue(re.search('\s+greetings:\s+mars', output))
def test_get_environment_id_multi_available_bad_name(self):
def env_list(*args, **kwargs):
return [{
"id": "1234",
"name": "somename"
}, {
"id": "5678",
"name": "othername"
}]
with patch('rhsm.connection.UEPConnection',
new_callable=StubUEP) as mock_uep:
mock_uep.getEnvironmentList = env_list
mock_uep.supports_resource = Mock(return_value=True)
self.stub_cp_provider.basic_auth_cp = mock_uep
rc = RegisterCommand()
rc.options = Mock()
rc.options.activation_keys = None
rc._prompt_for_environment = Mock(return_value="not_an_env")
with nested(Capture(silent=True), self.assertRaises(SystemExit)):
rc._get_environment_id(mock_uep, 'owner', None)
def test_date_validate_supported_locales_12_29_2020(self):
with Capture(silent=True):
d = datetime(2020, 12, 29, tzinfo=tzlocal())
self.__date_validate_supported_locales(d)
def test_product_cert_with_os_name_output(self):
with Capture() as cap:
command = StatCertCommandStub(certdata.PRODUCT_CERT_WITH_OS_NAME_V1_0)
command.main(['will_use_stub'])
cert_output = cap.out
self.assert_string_equals(certdata.PRODUCT_CERT_WITH_OS_NAME_V1_0_STAT_OUTPUT, cert_output)
def test_identity_cert_output(self):
with Capture() as cap:
command = CatCertCommandStub(certdata.IDENTITY_CERT)
command.main(['will_use_stub'])
cert_output = cap.out
self.assert_string_equals(certdata.IDENTITY_CERT_OUTPUT, cert_output)
def test_entitlement_cert_output_includes_content_sets(self):
with Capture() as cap:
command = StatCertCommandStub(certdata.ENTITLEMENT_CERT_V3_0)
command.main(['will_use_stub'])
cert_output = cap.out
self.assert_string_equals(certdata.ENTITLEMENT_CERT_V3_0_STAT_OUTPUT, cert_output)
def test_remove_config_default(self):
with Capture() as cap:
self.cc._do_command = self._orig_do_command
self.cc.main(['--remove', 'rhsm.baseurl'])
self.assertTrue('The default value for' in cap.out)