@@ -1797,15 +1797,18 @@ def test_cert_store_save_key(self, _set_store, cephadm_module: CephadmOrchestrat
17971797 grafana_host1_key = 'fake-grafana-host1-key'
17981798 nvmeof_client_key = 'nvmeof-client-key'
17991799 nvmeof_server_key = 'nvmeof-server-key'
1800+ nvmeof_encryption_key = 'nvmeof-encryption-key'
18001801 grafana_host1_key = 'fake-grafana-host1-cert'
18011802 cephadm_module .cert_key_store .save_key ('grafana_key' , grafana_host1_key , host = 'host1' )
18021803 cephadm_module .cert_key_store .save_key ('nvmeof_client_key' , nvmeof_client_key , service_name = 'nvmeof.foo' )
18031804 cephadm_module .cert_key_store .save_key ('nvmeof_server_key' , nvmeof_server_key , service_name = 'nvmeof.foo' )
1805+ cephadm_module .cert_key_store .save_key ('nvmeof_encryption_key' , nvmeof_encryption_key , service_name = 'nvmeof.foo' )
18041806
18051807 expected_calls = [
18061808 mock .call (f'{ CERT_STORE_KEY_PREFIX } grafana_key' , json .dumps ({'host1' : PrivKey (grafana_host1_key ).to_json ()})),
18071809 mock .call (f'{ CERT_STORE_KEY_PREFIX } nvmeof_client_key' , json .dumps ({'nvmeof.foo' : PrivKey (nvmeof_client_key ).to_json ()})),
18081810 mock .call (f'{ CERT_STORE_KEY_PREFIX } nvmeof_server_key' , json .dumps ({'nvmeof.foo' : PrivKey (nvmeof_server_key ).to_json ()})),
1811+ mock .call (f'{ CERT_STORE_KEY_PREFIX } nvmeof_encryption_key' , json .dumps ({'nvmeof.foo' : PrivKey (nvmeof_encryption_key ).to_json ()})),
18091812 ]
18101813 _set_store .assert_has_calls (expected_calls )
18111814
@@ -1822,15 +1825,19 @@ def test_cert_store_key_ls(self, _set_store, cephadm_module: CephadmOrchestrator
18221825 'ingress_ssl_key' : False ,
18231826 'nvmeof_client_key' : False ,
18241827 'nvmeof_server_key' : False ,
1828+ 'nvmeof_encryption_key' : False ,
18251829 }
18261830 assert cephadm_module .cert_key_store .key_ls () == expected_ls
18271831
18281832 cephadm_module .cert_key_store .save_key ('nvmeof_client_key' , 'xxx' , service_name = 'nvmeof.foo' )
18291833 cephadm_module .cert_key_store .save_key ('nvmeof_server_key' , 'xxx' , service_name = 'nvmeof.foo' )
1834+ cephadm_module .cert_key_store .save_key ('nvmeof_encryption_key' , 'xxx' , service_name = 'nvmeof.foo' )
18301835 expected_ls ['nvmeof_server_key' ] = {}
18311836 expected_ls ['nvmeof_server_key' ]['nvmeof.foo' ] = True
18321837 expected_ls ['nvmeof_client_key' ] = {}
18331838 expected_ls ['nvmeof_client_key' ]['nvmeof.foo' ] = True
1839+ expected_ls ['nvmeof_encryption_key' ] = {}
1840+ expected_ls ['nvmeof_encryption_key' ]['nvmeof.foo' ] = True
18341841 assert cephadm_module .cert_key_store .key_ls () == expected_ls
18351842
18361843 @mock .patch ("cephadm.module.CephadmOrchestrator.get_store_prefix" )
@@ -1844,6 +1851,7 @@ def test_cert_store_load(self, _get_store_prefix, cephadm_module: CephadmOrchest
18441851 nvmeof_root_ca_cert = 'nvmeof-root-ca-cert'
18451852 nvmeof_server_key = 'nvmeof-server-key'
18461853 nvmeof_client_key = 'nvmeof-client-key'
1854+ nvmeof_encryption_key = 'nvmeof-encryption-key'
18471855
18481856 def _fake_prefix_store (key ):
18491857 if key == 'cert_store.cert.' :
@@ -1858,6 +1866,7 @@ def _fake_prefix_store(key):
18581866 f'{ CERT_STORE_KEY_PREFIX } grafana_key' : json .dumps ({'host1' : PrivKey (grafana_host1_key ).to_json ()}),
18591867 f'{ CERT_STORE_KEY_PREFIX } nvmeof_server_key' : json .dumps ({'nvmeof.foo' : PrivKey (nvmeof_server_key ).to_json ()}),
18601868 f'{ CERT_STORE_KEY_PREFIX } nvmeof_client_key' : json .dumps ({'nvmeof.foo' : PrivKey (nvmeof_client_key ).to_json ()}),
1869+ f'{ CERT_STORE_KEY_PREFIX } nvmeof_encryption_key' : json .dumps ({'nvmeof.foo' : PrivKey (nvmeof_encryption_key ).to_json ()}),
18611870 }
18621871 else :
18631872 raise Exception (f'Get store with unexpected value { key } ' )
@@ -1871,6 +1880,7 @@ def _fake_prefix_store(key):
18711880 assert cephadm_module .cert_key_store .known_keys ['grafana_key' ]['host1' ] == PrivKey (grafana_host1_key )
18721881 assert cephadm_module .cert_key_store .known_keys ['nvmeof_server_key' ]['nvmeof.foo' ] == PrivKey (nvmeof_server_key )
18731882 assert cephadm_module .cert_key_store .known_keys ['nvmeof_client_key' ]['nvmeof.foo' ] == PrivKey (nvmeof_client_key )
1883+ assert cephadm_module .cert_key_store .known_keys ['nvmeof_encryption_key' ]['nvmeof.foo' ] == PrivKey (nvmeof_encryption_key )
18741884
18751885 def test_cert_store_get_cert_key (self , cephadm_module : CephadmOrchestrator ):
18761886 cephadm_module .cert_key_store ._init_known_cert_key_dicts ()
@@ -1898,13 +1908,16 @@ def test_cert_store_get_cert_key(self, cephadm_module: CephadmOrchestrator):
18981908
18991909 grafana_host1_key = 'fake-grafana-host1-cert'
19001910 nvmeof_server_key = 'nvmeof-server-key'
1911+ nvmeof_encryption_key = 'nvmeof-encryption-key'
19011912 cephadm_module .cert_key_store .save_key ('grafana_key' , grafana_host1_key , host = 'host1' )
19021913 cephadm_module .cert_key_store .save_key ('grafana_key' , grafana_host1_key , host = 'host1' )
19031914 cephadm_module .cert_key_store .save_key ('nvmeof_server_key' , nvmeof_server_key , service_name = 'nvmeof.foo' )
1915+ cephadm_module .cert_key_store .save_key ('nvmeof_encryption_key' , nvmeof_encryption_key , service_name = 'nvmeof.foo' )
19041916
19051917 assert cephadm_module .cert_key_store .get_key ('grafana_key' , host = 'host1' ) == grafana_host1_key
19061918 assert cephadm_module .cert_key_store .get_key ('nvmeof_server_key' , service_name = 'nvmeof.foo' ) == nvmeof_server_key
19071919 assert cephadm_module .cert_key_store .get_key ('nvmeof_client_key' , service_name = 'nvmeof.foo' ) == ''
1920+ assert cephadm_module .cert_key_store .get_key ('nvmeof_encryption_key' , service_name = 'nvmeof.foo' ) == nvmeof_encryption_key
19081921
19091922 with pytest .raises (OrchestratorError , match = 'Attempted to access priv key for unknown entity' ):
19101923 cephadm_module .cert_key_store .get_key ('unknown_entity' )
0 commit comments