def main():
v_cassandra_storage = cassandra_storage.CassandraSanInfoStorage(CONF)
sni_attribute_default_list = {
'enrollmentId': None,
}
sni_info_dict = dict()
for sni_cert_name in CONF[driver.AKAMAI_GROUP].sni_cert_cnames:
sni_info_dict[sni_cert_name] = {}
print("Insert SNI info for: {0}".format(sni_cert_name))
for attr in sni_attribute_default_list:
user_input = None
while ((user_input or "").strip() or user_input) in ["", None]:
user_input = input(
'Please input value for attr: {0}, SNI cert: {1}, '
'default value: {2} (if default is None, '
'that means a real value has to be input): '.format(
attr, sni_cert_name, sni_attribute_default_list[attr]))
# enrollmentId is required
if user_input in ["", None] and attr == "enrollmentId":
break
if sni_attribute_default_list[attr] is None:
continue
else:
user_input = sni_attribute_default_list[attr]
break
sni_info_dict[sni_cert_name][attr] = user_input
v_cassandra_storage.update_san_info(sni_info_dict, info_type='sni')
def test__get_akamai_provider_info(self):
self.cassandra_storage = cassandra_storage.CassandraSanInfoStorage(
self.conf)
mock_execute = self.cassandra_storage.session.execute
mock_execute.return_value = self.get_returned_value
res = self.cassandra_storage._get_akamai_provider_info()
mock_execute.assert_called()
self.assertTrue(res == self.get_returned_value[0])
def test_save_cert_last_spsid(self, san_cert_name):
self.cassandra_storage = cassandra_storage.CassandraSanInfoStorage(
self.conf)
mock_execute = self.cassandra_storage.session.execute
mock_execute.return_value = self.get_returned_value
self.cassandra_storage.save_cert_last_ids(san_cert_name, '1234')
self.assertTrue(mock_execute.call_count == 3)
def test_usage_non_shared_cassandra_cluster(self):
with mock.patch('poppy.storage.cassandra.driver.CassandraStorageDriver'
) as mock_driver:
mock_d = mock.Mock()
mock_d.cassandra_conf.consistency_level = 'ONE'
mock_driver.return_value = mock_d
cass = cassandra_storage.CassandraSanInfoStorage(self.conf)
self.assertTrue(cass.storage.change_config_group.called)
def test_list_all_san_cert_names(self):
self.cassandra_storage = cassandra_storage.CassandraSanInfoStorage(
self.conf)
mock_execute = self.cassandra_storage.session.execute
mock_execute.return_value = self.get_returned_value
res = self.cassandra_storage.list_all_san_cert_names()
mock_execute.assert_called()
self.assertTrue(res == json.loads(self.get_returned_value[0]['info']
['san_info']).keys())
def test_set_san_cert_hostname_limit(self):
self.cassandra_storage = cassandra_storage.CassandraSanInfoStorage(
self.conf
)
mock_execute = self.cassandra_storage.session.execute
mock_execute.return_value = self.get_returned_value
self.cassandra_storage.set_san_cert_hostname_limit(99)
mock_execute.assert_called()
def test__get_akamai_sni_certs_info(self):
self.cassandra_storage = cassandra_storage.CassandraSanInfoStorage(
self.conf)
mock_execute = self.cassandra_storage.session.execute
mock_execute.return_value = self.get_returned_value
res = self.cassandra_storage._get_akamai_sni_certs_info()
mock_execute.assert_called()
self.assertTrue(
res == json.loads(self.get_returned_value[0]['info']['sni_info'])
)
def test_update_cert_config(self):
self.cassandra_storage = cassandra_storage.CassandraSanInfoStorage(
self.conf)
mock_execute = self.cassandra_storage.session.execute
mock_execute.return_value = self.get_returned_value
cert_name = "secure1.san1.test-cdn.com"
new_spsId = 3456
self.cassandra_storage.update_cert_config(cert_name,
{'spsId': new_spsId})
mock_execute.assert_called()
def test_get_cert_last_spsid(self):
self.cassandra_storage = cassandra_storage.CassandraSanInfoStorage(
self.conf)
mock_execute = self.cassandra_storage.session.execute
mock_execute.return_value = self.get_returned_value
cert_name = "secure1.san1.test-cdn.com"
res = self.cassandra_storage.get_cert_last_spsid(cert_name)
mock_execute.assert_called()
self.assertTrue(res == json.loads(self.get_returned_value[0]['info']
['san_info'])[cert_name]['spsId'])
def test_get_san_cert_hostname_limit(self):
self.cassandra_storage = cassandra_storage.CassandraSanInfoStorage(
self.conf
)
mock_execute = self.cassandra_storage.session.execute
mock_execute.return_value = self.get_returned_value
res = self.cassandra_storage.get_san_cert_hostname_limit()
mock_execute.assert_called()
self.assertEqual(res, self.default_limit)
def test_update_sni_cert_config(self):
self.cassandra_storage = cassandra_storage.CassandraSanInfoStorage(
self.conf)
mock_execute = self.cassandra_storage.session.execute
mock_execute.return_value = self.get_returned_value
cert_name = "secured2.sni1.test-cdn.com"
new_enrollment_id = 9898
self.cassandra_storage.update_sni_cert_config(
cert_name, {'enrollmentId': new_enrollment_id}
)
mock_execute.assert_called()
def main():
v_cassandra_storage = cassandra_storage.CassandraSanInfoStorage(CONF)
all_san_cert_names = v_cassandra_storage.list_all_san_cert_names()
if not all_san_cert_names:
print("Currently no SAN cert info has been initialized")
for san_cert_name in all_san_cert_names:
print("%s:%s" %
(san_cert_name,
str(v_cassandra_storage.get_cert_info(san_cert_name))))
def test_get_sni_cert_info(self):
self.cassandra_storage = cassandra_storage.CassandraSanInfoStorage(
self.conf)
mock_execute = self.cassandra_storage.session.execute
mock_execute.return_value = self.get_returned_value
cert_name = "secured2.sni1.test-cdn.com"
res = self.cassandra_storage.get_sni_cert_info(
cert_name
)
mock_execute.assert_called()
self.assertTrue(
res['enrollmentId'] == str(json.loads(
self.get_returned_value[0]['info']['sni_info']
)[cert_name]['enrollmentId'])
)
def main():
v_cassandra_storage = cassandra_storage.CassandraSanInfoStorage(CONF)
sni_attribute_default_list = {
'enrollmentId': None,
}
sni_info_dict = dict()
cnames = CONF[driver.AKAMAI_GROUP].sni_cert_cnames
enrollment_ids = CONF[driver.AKAMAI_GROUP].sni_cert_enrollment_ids
if enrollment_ids and len(cnames) == len(enrollment_ids):
print("Found valid 'sni_cert_enrollment_ids' setting in poppy.conf")
print("{:-^50}\n".format(" Starting upsert "))
for cname, eid in zip(cnames, enrollment_ids):
sni_info_dict[cname] = {}
sni_info_dict[cname]["enrollmentId"] = eid
print("{} -> {}".format(cname, eid))
print("\n{:-^50}".format(" Finished "))
else:
for sni_cert_name in cnames:
sni_info_dict[sni_cert_name] = {}
print("Insert SNI info for: {0}".format(sni_cert_name))
for attr in sni_attribute_default_list:
user_input = None
while ((user_input or "").strip() or user_input) in ["", None]:
user_input = input(
'Please input value for attr: {0}, SNI cert: {1}, '
'default value: {2} (if default is None, '
'that means a real value has to be input): '.format(
attr, sni_cert_name,
sni_attribute_default_list[attr]))
# enrollmentId is required
if user_input in ["", None] and attr == "enrollmentId":
break
if sni_attribute_default_list[attr] is None:
continue
else:
user_input = sni_attribute_default_list[attr]
break
sni_info_dict[sni_cert_name][attr] = user_input
v_cassandra_storage.update_san_info(sni_info_dict, info_type='sni')
def main():
v_cassandra_storage = cassandra_storage.CassandraSanInfoStorage(CONF)
san_attribute_default_list = {
'issuer': 'symantec',
'ipVersion': 'ipv4',
'slot_deployment_klass': 'esslType',
'spsId': None,
'jobId': None
}
san_info_dict = {}
for san_cert_name in CONF[driver.AKAMAI_GROUP].san_cert_cnames:
san_info_dict[san_cert_name] = {}
print("Insert SAN info for :%s" % (san_cert_name))
for attr in san_attribute_default_list:
user_input = None
while ((user_input or "").strip() or user_input) in ["", None]:
user_input = raw_input(
'Please input value for attr: %s, '
'San cert: %s,'
'default value: %s'
' (if default is None, '
'that means a real value has to'
' be input, except spsId): ' %
(attr, san_cert_name, san_attribute_default_list[attr]))
# We allow not inputing spsId, however if it is in
# transitioning from the old manual SAN process we NEED
# to put in spsId
if user_input in ["", None] and attr == "spsId":
break
if san_attribute_default_list[attr] is None:
continue
else:
user_input = san_attribute_default_list[attr]
break
san_info_dict[san_cert_name][attr] = user_input
v_cassandra_storage.update_san_info(san_info_dict)
def test_update_san_info(self):
self.cassandra_storage = cassandra_storage.CassandraSanInfoStorage(
self.conf)
mock_execute = self.cassandra_storage.session.execute
self.cassandra_storage.update_san_info({})
mock_execute.assert_called()
