oracle_test.go 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833
  1. /*
  2. Copyright © 2025 ESO Maintainer Team
  3. Licensed under the Apache License, Version 2.0 (the "License");
  4. you may not use this file except in compliance with the License.
  5. You may obtain a copy of the License at
  6. https://www.apache.org/licenses/LICENSE-2.0
  7. Unless required by applicable law or agreed to in writing, software
  8. distributed under the License is distributed on an "AS IS" BASIS,
  9. WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  10. See the License for the specific language governing permissions and
  11. limitations under the License.
  12. */
  13. package oracle
  14. import (
  15. "context"
  16. "crypto/rand"
  17. "crypto/rsa"
  18. "crypto/x509"
  19. "encoding/base64"
  20. "encoding/pem"
  21. "errors"
  22. "reflect"
  23. "strings"
  24. "testing"
  25. "time"
  26. "github.com/oracle/oci-go-sdk/v65/common"
  27. "github.com/oracle/oci-go-sdk/v65/secrets"
  28. "github.com/oracle/oci-go-sdk/v65/vault"
  29. "github.com/stretchr/testify/assert"
  30. corev1 "k8s.io/api/core/v1"
  31. metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
  32. "k8s.io/apimachinery/pkg/runtime"
  33. "k8s.io/utils/ptr"
  34. clientfake "sigs.k8s.io/controller-runtime/pkg/client/fake"
  35. esv1 "github.com/external-secrets/external-secrets/apis/externalsecrets/v1"
  36. esv1alpha1 "github.com/external-secrets/external-secrets/apis/externalsecrets/v1alpha1"
  37. esmeta "github.com/external-secrets/external-secrets/apis/meta/v1"
  38. fakeoracle "github.com/external-secrets/external-secrets/providers/v1/oracle/fake"
  39. testingfake "github.com/external-secrets/external-secrets/runtime/testing/fake"
  40. )
  41. const (
  42. vaultOCID = "vault-OCID"
  43. region = "some-region"
  44. tenant = "a-tenant"
  45. userOCID = "user-OCID"
  46. secretKey = "key"
  47. secretName = "name"
  48. )
  49. type vaultTestCase struct {
  50. mockClient *fakeoracle.OracleMockClient
  51. apiInput *secrets.GetSecretBundleByNameRequest
  52. apiOutput *secrets.GetSecretBundleByNameResponse
  53. ref *esv1.ExternalSecretDataRemoteRef
  54. apiErr error
  55. expectError string
  56. expectedSecret string
  57. // for testing secretmap
  58. expectedData map[string][]byte
  59. }
  60. func makeValidVaultTestCase() *vaultTestCase {
  61. smtc := vaultTestCase{
  62. mockClient: &fakeoracle.OracleMockClient{},
  63. apiInput: makeValidAPIInput(),
  64. ref: makeValidRef(),
  65. apiOutput: makeValidAPIOutput(),
  66. apiErr: nil,
  67. expectError: "",
  68. expectedSecret: "",
  69. expectedData: map[string][]byte{},
  70. }
  71. smtc.mockClient.WithValue(*smtc.apiInput, *smtc.apiOutput, smtc.apiErr)
  72. return &smtc
  73. }
  74. func makeValidRef() *esv1.ExternalSecretDataRemoteRef {
  75. return &esv1.ExternalSecretDataRemoteRef{
  76. Key: "test-secret",
  77. Version: "default",
  78. }
  79. }
  80. func makeValidAPIInput() *secrets.GetSecretBundleByNameRequest {
  81. return &secrets.GetSecretBundleByNameRequest{
  82. SecretName: ptr.To("test-secret"),
  83. VaultId: ptr.To("test-vault"),
  84. }
  85. }
  86. func makeValidAPIOutput() *secrets.GetSecretBundleByNameResponse {
  87. return &secrets.GetSecretBundleByNameResponse{
  88. SecretBundle: secrets.SecretBundle{},
  89. }
  90. }
  91. func makeValidVaultTestCaseCustom(tweaks ...func(smtc *vaultTestCase)) *vaultTestCase {
  92. smtc := makeValidVaultTestCase()
  93. for _, fn := range tweaks {
  94. fn(smtc)
  95. }
  96. smtc.mockClient.WithValue(*smtc.apiInput, *smtc.apiOutput, smtc.apiErr)
  97. return smtc
  98. }
  99. // This case can be shared by both GetSecret and GetSecretMap tests.
  100. // bad case: set apiErr.
  101. var setAPIErr = func(smtc *vaultTestCase) {
  102. smtc.apiErr = errors.New("oh no")
  103. smtc.expectError = "oh no"
  104. }
  105. var setNilMockClient = func(smtc *vaultTestCase) {
  106. smtc.mockClient = nil
  107. smtc.expectError = errUninitalizedOracleProvider
  108. }
  109. func TestOracleVaultGetSecret(t *testing.T) {
  110. secretValue := "changedvalue"
  111. // good case: default version is set
  112. // key is passed in, output is sent back
  113. setSecretString := func(smtc *vaultTestCase) {
  114. smtc.apiOutput = &secrets.GetSecretBundleByNameResponse{
  115. SecretBundle: secrets.SecretBundle{
  116. SecretId: ptr.To("test-id"),
  117. VersionNumber: ptr.To(int64(1)),
  118. SecretBundleContent: secrets.Base64SecretBundleContentDetails{
  119. Content: ptr.To(base64.StdEncoding.EncodeToString([]byte(secretValue))),
  120. },
  121. },
  122. }
  123. smtc.expectedSecret = secretValue
  124. }
  125. successCases := []*vaultTestCase{
  126. makeValidVaultTestCaseCustom(setAPIErr),
  127. makeValidVaultTestCaseCustom(setNilMockClient),
  128. makeValidVaultTestCaseCustom(setSecretString),
  129. }
  130. sm := VaultManagementService{}
  131. for k, v := range successCases {
  132. sm.Client = v.mockClient
  133. out, err := sm.GetSecret(context.Background(), *v.ref)
  134. if !ErrorContains(err, v.expectError) {
  135. t.Errorf("[%d] unexpected error: %s, expected: '%s'", k, err.Error(), v.expectError)
  136. }
  137. if string(out) != v.expectedSecret {
  138. t.Errorf("[%d] unexpected secret: expected %s, got %s", k, v.expectedSecret, string(out))
  139. }
  140. }
  141. }
  142. func TestGetSecretMap(t *testing.T) {
  143. // good case: default version & deserialization
  144. setDeserialization := func(smtc *vaultTestCase) {
  145. smtc.apiOutput.SecretBundleContent = secrets.Base64SecretBundleContentDetails{
  146. Content: ptr.To(base64.StdEncoding.EncodeToString([]byte(`{"foo":"bar"}`))),
  147. }
  148. smtc.expectedData["foo"] = []byte("bar")
  149. }
  150. // bad case: invalid json
  151. setInvalidJSON := func(smtc *vaultTestCase) {
  152. smtc.apiOutput.SecretBundleContent = secrets.Base64SecretBundleContentDetails{
  153. Content: ptr.To(base64.StdEncoding.EncodeToString([]byte(`-----------------`))),
  154. }
  155. smtc.expectError = "unable to unmarshal secret"
  156. }
  157. successCases := []*vaultTestCase{
  158. makeValidVaultTestCaseCustom(setDeserialization),
  159. makeValidVaultTestCaseCustom(setInvalidJSON),
  160. makeValidVaultTestCaseCustom(setNilMockClient),
  161. makeValidVaultTestCaseCustom(setAPIErr),
  162. }
  163. sm := VaultManagementService{}
  164. for k, v := range successCases {
  165. sm.Client = v.mockClient
  166. out, err := sm.GetSecretMap(context.Background(), *v.ref)
  167. if !ErrorContains(err, v.expectError) {
  168. t.Errorf("[%d] unexpected error: %s, expected: '%s'", k, err.Error(), v.expectError)
  169. }
  170. if err == nil && !reflect.DeepEqual(out, v.expectedData) {
  171. t.Errorf("[%d] unexpected secret data: expected %#v, got %#v", k, v.expectedData, out)
  172. }
  173. }
  174. }
  175. func ErrorContains(out error, want string) bool {
  176. if out == nil {
  177. return want == ""
  178. }
  179. if want == "" {
  180. return false
  181. }
  182. return strings.Contains(out.Error(), want)
  183. }
  184. type storeModifier func(*esv1.SecretStore) *esv1.SecretStore
  185. func makeSecretStore(vault, region string, fn ...storeModifier) *esv1.SecretStore {
  186. store := &esv1.SecretStore{
  187. Spec: esv1.SecretStoreSpec{
  188. Provider: &esv1.SecretStoreProvider{
  189. Oracle: &esv1.OracleProvider{
  190. Vault: vault,
  191. Region: region,
  192. },
  193. },
  194. },
  195. }
  196. for _, f := range fn {
  197. store = f(store)
  198. }
  199. return store
  200. }
  201. func withSecretAuth(user, tenancy string) storeModifier {
  202. return func(store *esv1.SecretStore) *esv1.SecretStore {
  203. store.Spec.Provider.Oracle.Auth = &esv1.OracleAuth{
  204. User: user,
  205. Tenancy: tenancy,
  206. }
  207. return store
  208. }
  209. }
  210. func withPrivateKey(name, key string, namespace *string) storeModifier {
  211. return func(store *esv1.SecretStore) *esv1.SecretStore {
  212. store.Spec.Provider.Oracle.Auth.SecretRef.PrivateKey = esmeta.SecretKeySelector{
  213. Name: name,
  214. Key: key,
  215. Namespace: namespace,
  216. }
  217. return store
  218. }
  219. }
  220. func withFingerprint(name, key string, namespace *string) storeModifier {
  221. return func(store *esv1.SecretStore) *esv1.SecretStore {
  222. store.Spec.Provider.Oracle.Auth.SecretRef.Fingerprint = esmeta.SecretKeySelector{
  223. Name: name,
  224. Key: key,
  225. Namespace: namespace,
  226. }
  227. return store
  228. }
  229. }
  230. type ValidateStoreTestCase struct {
  231. store *esv1.SecretStore
  232. err error
  233. }
  234. func TestValidateStore(t *testing.T) {
  235. namespace := "my-namespace"
  236. testCases := []ValidateStoreTestCase{
  237. {
  238. store: makeSecretStore("", region),
  239. err: errors.New("vault cannot be empty"),
  240. },
  241. {
  242. store: makeSecretStore(vaultOCID, ""),
  243. err: errors.New("region cannot be empty"),
  244. },
  245. {
  246. store: makeSecretStore(vaultOCID, region, withSecretAuth("", tenant)),
  247. err: errors.New("user cannot be empty"),
  248. },
  249. {
  250. store: makeSecretStore(vaultOCID, region, withSecretAuth(userOCID, "")),
  251. err: errors.New("tenant cannot be empty"),
  252. },
  253. {
  254. store: makeSecretStore(vaultOCID, region, withSecretAuth(userOCID, tenant), withPrivateKey("", secretKey, nil)),
  255. err: errors.New("privateKey.name cannot be empty"),
  256. },
  257. {
  258. store: makeSecretStore(vaultOCID, region, withSecretAuth(userOCID, tenant), withPrivateKey(secretName, secretKey, &namespace)),
  259. err: errors.New("namespace should either be empty or match the namespace of the SecretStore for a namespaced SecretStore"),
  260. },
  261. {
  262. store: makeSecretStore(vaultOCID, region, withSecretAuth(userOCID, tenant), withPrivateKey(secretName, "", nil)),
  263. err: errors.New("privateKey.key cannot be empty"),
  264. },
  265. {
  266. store: makeSecretStore(vaultOCID, region, withSecretAuth(userOCID, tenant), withPrivateKey(secretName, secretKey, nil), withFingerprint("", secretKey, nil)),
  267. err: errors.New("fingerprint.name cannot be empty"),
  268. },
  269. {
  270. store: makeSecretStore(vaultOCID, region, withSecretAuth(userOCID, tenant), withPrivateKey(secretName, secretKey, nil), withFingerprint(secretName, secretKey, &namespace)),
  271. err: errors.New("namespace should either be empty or match the namespace of the SecretStore for a namespaced SecretStore"),
  272. },
  273. {
  274. store: makeSecretStore(vaultOCID, region, withSecretAuth(userOCID, tenant), withPrivateKey(secretName, secretKey, nil), withFingerprint(secretName, "", nil)),
  275. err: errors.New("fingerprint.key cannot be empty"),
  276. },
  277. {
  278. store: makeSecretStore(vaultOCID, region),
  279. err: nil,
  280. },
  281. }
  282. p := VaultManagementService{}
  283. for _, tc := range testCases {
  284. _, err := p.ValidateStore(tc.store)
  285. if tc.err != nil && err != nil && err.Error() != tc.err.Error() {
  286. t.Errorf("test failed! want %v, got %v", tc.err, err)
  287. } else if tc.err == nil && err != nil {
  288. t.Errorf("want nil got err %v", err)
  289. } else if tc.err != nil && err == nil {
  290. t.Errorf("want err %v got nil", tc.err)
  291. }
  292. }
  293. }
  294. func TestVaultManagementServiceNewClient(t *testing.T) {
  295. t.Parallel()
  296. namespace := "default"
  297. authSecretName := "oracle-auth"
  298. auth := &esv1.OracleAuth{
  299. User: "user",
  300. Tenancy: "tenancy",
  301. SecretRef: esv1.OracleSecretRef{
  302. PrivateKey: esmeta.SecretKeySelector{
  303. Name: authSecretName,
  304. Key: "privateKey",
  305. },
  306. Fingerprint: esmeta.SecretKeySelector{
  307. Name: authSecretName,
  308. Key: "fingerprint",
  309. },
  310. },
  311. }
  312. tests := []struct {
  313. desc string
  314. secretStore *esv1.SecretStore
  315. expectedErr string
  316. }{
  317. {
  318. desc: "no retry settings",
  319. secretStore: &esv1.SecretStore{
  320. Spec: esv1.SecretStoreSpec{
  321. Provider: &esv1.SecretStoreProvider{
  322. Oracle: &esv1.OracleProvider{
  323. Vault: vaultOCID,
  324. Region: region,
  325. Auth: auth,
  326. },
  327. },
  328. },
  329. },
  330. },
  331. {
  332. desc: "fill all the retry settings",
  333. secretStore: &esv1.SecretStore{
  334. Spec: esv1.SecretStoreSpec{
  335. Provider: &esv1.SecretStoreProvider{
  336. Oracle: &esv1.OracleProvider{
  337. Vault: vaultOCID,
  338. Region: region,
  339. Auth: auth,
  340. },
  341. },
  342. RetrySettings: &esv1.SecretStoreRetrySettings{
  343. RetryInterval: ptr.To("1s"),
  344. MaxRetries: ptr.To(int32(5)),
  345. },
  346. },
  347. },
  348. },
  349. {
  350. desc: "partially configure the retry settings - retry interval",
  351. secretStore: &esv1.SecretStore{
  352. Spec: esv1.SecretStoreSpec{
  353. Provider: &esv1.SecretStoreProvider{
  354. Oracle: &esv1.OracleProvider{
  355. Vault: vaultOCID,
  356. Region: region,
  357. Auth: auth,
  358. },
  359. },
  360. RetrySettings: &esv1.SecretStoreRetrySettings{
  361. RetryInterval: ptr.To("1s"),
  362. },
  363. },
  364. },
  365. },
  366. {
  367. desc: "partially configure the retry settings - max retries",
  368. secretStore: &esv1.SecretStore{
  369. Spec: esv1.SecretStoreSpec{
  370. Provider: &esv1.SecretStoreProvider{
  371. Oracle: &esv1.OracleProvider{
  372. Vault: vaultOCID,
  373. Region: region,
  374. Auth: auth,
  375. },
  376. },
  377. RetrySettings: &esv1.SecretStoreRetrySettings{
  378. MaxRetries: ptr.To(int32(5)),
  379. },
  380. },
  381. },
  382. },
  383. {
  384. desc: "auth secret does not exist",
  385. secretStore: &esv1.SecretStore{
  386. Spec: esv1.SecretStoreSpec{
  387. Provider: &esv1.SecretStoreProvider{
  388. Oracle: &esv1.OracleProvider{
  389. Vault: vaultOCID,
  390. Region: region,
  391. Auth: &esv1.OracleAuth{
  392. User: "user",
  393. Tenancy: "tenancy",
  394. SecretRef: esv1.OracleSecretRef{
  395. PrivateKey: esmeta.SecretKeySelector{
  396. Name: "non-existing-secret",
  397. Key: "privateKey",
  398. },
  399. Fingerprint: esmeta.SecretKeySelector{
  400. Name: "non-existing-secret",
  401. Key: "fingerprint",
  402. },
  403. },
  404. },
  405. },
  406. },
  407. RetrySettings: &esv1.SecretStoreRetrySettings{
  408. RetryInterval: ptr.To("invalid"),
  409. },
  410. },
  411. },
  412. expectedErr: `cannot get Kubernetes secret "non-existing-secret" from namespace "default": secrets "non-existing-secret" not found`,
  413. },
  414. {
  415. desc: "invalid retry interval",
  416. secretStore: &esv1.SecretStore{
  417. Spec: esv1.SecretStoreSpec{
  418. Provider: &esv1.SecretStoreProvider{
  419. Oracle: &esv1.OracleProvider{
  420. Vault: vaultOCID,
  421. Region: region,
  422. Auth: auth,
  423. },
  424. },
  425. RetrySettings: &esv1.SecretStoreRetrySettings{
  426. RetryInterval: ptr.To("invalid"),
  427. },
  428. },
  429. },
  430. expectedErr: "cannot setup new oracle client: time: invalid duration",
  431. },
  432. }
  433. for _, tc := range tests {
  434. t.Run(tc.desc, func(t *testing.T) {
  435. provider := &VaultManagementService{
  436. Client: &fakeoracle.OracleMockClient{},
  437. KmsVaultClient: nil,
  438. vault: vaultOCID,
  439. }
  440. pk, err := rsa.GenerateKey(rand.Reader, 2048)
  441. if err != nil {
  442. t.Fatalf("failed to create a private key: %v", err)
  443. }
  444. schema := runtime.NewScheme()
  445. if err := corev1.AddToScheme(schema); err != nil {
  446. t.Fatalf("failed to add to schema: %v", err)
  447. }
  448. builder := clientfake.NewClientBuilder().WithRuntimeObjects(&corev1.Secret{
  449. ObjectMeta: metav1.ObjectMeta{
  450. Name: authSecretName,
  451. Namespace: namespace,
  452. },
  453. Data: map[string][]byte{
  454. "privateKey": pem.EncodeToMemory(&pem.Block{
  455. Type: "RSA PRIVATE KEY",
  456. Bytes: x509.MarshalPKCS1PrivateKey(pk),
  457. }),
  458. "fingerprint": []byte("fingerprint"),
  459. },
  460. })
  461. _, err = provider.NewClient(context.Background(), tc.secretStore, builder.Build(), namespace)
  462. if err != nil {
  463. if tc.expectedErr == "" {
  464. t.Fatalf("failed to call NewClient: %v", err)
  465. }
  466. if !strings.Contains(err.Error(), tc.expectedErr) {
  467. t.Fatalf("received an unexpected error: %q should have contained %q", err.Error(), tc.expectedErr)
  468. }
  469. return
  470. }
  471. if tc.expectedErr != "" {
  472. t.Fatalf("expeceted to receive an error but got nil")
  473. }
  474. })
  475. }
  476. }
  477. func TestOracleVaultGetAllSecrets(t *testing.T) {
  478. var testCases = map[string]struct {
  479. vms *VaultManagementService
  480. ref esv1.ExternalSecretFind
  481. result map[string][]byte
  482. }{
  483. "filters secrets that don't match the pattern": {
  484. &VaultManagementService{
  485. Client: &fakeoracle.OracleMockClient{
  486. SecretBundles: map[string]secrets.SecretBundle{
  487. s1id: s1bundle,
  488. s2id: s2bundle,
  489. },
  490. },
  491. VaultClient: &fakeoracle.OracleMockVaultClient{
  492. SecretSummaries: []vault.SecretSummary{
  493. s1summary,
  494. s2summary,
  495. },
  496. },
  497. },
  498. esv1.ExternalSecretFind{
  499. Name: &esv1.FindName{
  500. RegExp: "^test.*",
  501. },
  502. },
  503. map[string][]byte{
  504. s1id: []byte(s1id),
  505. },
  506. },
  507. "filters secrets that are deleting": {
  508. &VaultManagementService{
  509. Client: &fakeoracle.OracleMockClient{
  510. SecretBundles: map[string]secrets.SecretBundle{
  511. s1id: s1bundle,
  512. s2id: s2bundle,
  513. s3id: s3bundle,
  514. },
  515. },
  516. VaultClient: &fakeoracle.OracleMockVaultClient{
  517. SecretSummaries: []vault.SecretSummary{
  518. s1summary,
  519. s2summary,
  520. s3summary,
  521. },
  522. },
  523. },
  524. esv1.ExternalSecretFind{
  525. Name: &esv1.FindName{
  526. RegExp: ".*",
  527. },
  528. },
  529. map[string][]byte{
  530. s1id: []byte(s1id),
  531. s2id: []byte(s2id),
  532. },
  533. },
  534. }
  535. for name, testCase := range testCases {
  536. t.Run(name, func(t *testing.T) {
  537. result, err := testCase.vms.GetAllSecrets(context.Background(), testCase.ref)
  538. assert.NoError(t, err)
  539. assert.EqualValues(t, testCase.result, result)
  540. })
  541. }
  542. }
  543. func TestOracleVaultPushSecret(t *testing.T) {
  544. testSecretKey := "test-secret-key"
  545. encryptionKey := "must-not-be-blank-for-push"
  546. var testCases = map[string]struct {
  547. vms *VaultManagementService
  548. data testingfake.PushSecretData
  549. validator func(service *VaultManagementService) bool
  550. content string
  551. }{
  552. "create a secret if not exists": {
  553. &VaultManagementService{
  554. encryptionKey: encryptionKey,
  555. Client: &fakeoracle.OracleMockClient{
  556. SecretBundles: map[string]secrets.SecretBundle{
  557. s2id: s2bundle,
  558. },
  559. },
  560. VaultClient: &fakeoracle.OracleMockVaultClient{},
  561. },
  562. testingfake.PushSecretData{
  563. SecretKey: testSecretKey,
  564. RemoteKey: s1id,
  565. },
  566. func(vms *VaultManagementService) bool {
  567. return vms.VaultClient.(*fakeoracle.OracleMockVaultClient).CreatedCount == 1
  568. },
  569. "created",
  570. },
  571. "create a json secret if not exists": {
  572. &VaultManagementService{
  573. encryptionKey: encryptionKey,
  574. Client: &fakeoracle.OracleMockClient{
  575. SecretBundles: map[string]secrets.SecretBundle{
  576. s2id: s2bundle,
  577. },
  578. },
  579. VaultClient: &fakeoracle.OracleMockVaultClient{},
  580. },
  581. testingfake.PushSecretData{
  582. SecretKey: testSecretKey,
  583. RemoteKey: s1id,
  584. },
  585. func(vms *VaultManagementService) bool {
  586. return vms.VaultClient.(*fakeoracle.OracleMockVaultClient).CreatedCount == 1
  587. },
  588. "{'key-a':'secret-a', 'key-b': 'secret-b'}",
  589. },
  590. "update a secret if exists": {
  591. &VaultManagementService{
  592. encryptionKey: encryptionKey,
  593. Client: &fakeoracle.OracleMockClient{
  594. SecretBundles: map[string]secrets.SecretBundle{
  595. s1id: s1bundle,
  596. s2id: s2bundle,
  597. },
  598. },
  599. VaultClient: &fakeoracle.OracleMockVaultClient{},
  600. },
  601. testingfake.PushSecretData{
  602. SecretKey: testSecretKey,
  603. RemoteKey: s1id,
  604. },
  605. func(vms *VaultManagementService) bool {
  606. return vms.VaultClient.(*fakeoracle.OracleMockVaultClient).UpdatedCount == 1
  607. },
  608. "updated",
  609. },
  610. "neither create nor update if secret content is unchanged": {
  611. &VaultManagementService{
  612. encryptionKey: encryptionKey,
  613. Client: &fakeoracle.OracleMockClient{
  614. SecretBundles: map[string]secrets.SecretBundle{
  615. s1id: s1bundle,
  616. s2id: s2bundle,
  617. },
  618. },
  619. VaultClient: &fakeoracle.OracleMockVaultClient{},
  620. },
  621. testingfake.PushSecretData{
  622. SecretKey: testSecretKey,
  623. RemoteKey: s1id,
  624. },
  625. func(vms *VaultManagementService) bool {
  626. return vms.VaultClient.(*fakeoracle.OracleMockVaultClient).UpdatedCount == 0 &&
  627. vms.VaultClient.(*fakeoracle.OracleMockVaultClient).CreatedCount == 0
  628. },
  629. s1id,
  630. },
  631. }
  632. for name, testCase := range testCases {
  633. t.Run(name, func(t *testing.T) {
  634. s := &corev1.Secret{Data: map[string][]byte{testSecretKey: []byte(testCase.content)}}
  635. err := testCase.vms.PushSecret(context.Background(), s, testCase.data)
  636. assert.NoError(t, err)
  637. assert.True(t, testCase.validator(testCase.vms))
  638. })
  639. }
  640. }
  641. func TestOracleVaultDeleteSecret(t *testing.T) {
  642. var testCases = map[string]struct {
  643. vms *VaultManagementService
  644. remoteRef esv1.PushSecretRemoteRef
  645. validator func(service *VaultManagementService) bool
  646. }{
  647. "do not delete if secret not found": {
  648. &VaultManagementService{
  649. Client: &fakeoracle.OracleMockClient{
  650. SecretBundles: map[string]secrets.SecretBundle{
  651. s1id: s1bundle,
  652. },
  653. },
  654. VaultClient: &fakeoracle.OracleMockVaultClient{},
  655. },
  656. esv1alpha1.PushSecretRemoteRef{
  657. RemoteKey: s2id,
  658. },
  659. func(vms *VaultManagementService) bool {
  660. return vms.VaultClient.(*fakeoracle.OracleMockVaultClient).DeletedCount == 0
  661. },
  662. },
  663. "do not delete if secret os already deleting": {
  664. &VaultManagementService{
  665. Client: &fakeoracle.OracleMockClient{
  666. SecretBundles: map[string]secrets.SecretBundle{
  667. s1id: s1bundle,
  668. s3id: s3bundle,
  669. },
  670. },
  671. VaultClient: &fakeoracle.OracleMockVaultClient{},
  672. },
  673. esv1alpha1.PushSecretRemoteRef{
  674. RemoteKey: s3id,
  675. },
  676. func(vms *VaultManagementService) bool {
  677. return vms.VaultClient.(*fakeoracle.OracleMockVaultClient).DeletedCount == 0
  678. },
  679. },
  680. "delete existing secret": {
  681. &VaultManagementService{
  682. Client: &fakeoracle.OracleMockClient{
  683. SecretBundles: map[string]secrets.SecretBundle{
  684. s1id: s1bundle,
  685. s3id: s3bundle,
  686. },
  687. },
  688. VaultClient: &fakeoracle.OracleMockVaultClient{},
  689. },
  690. esv1alpha1.PushSecretRemoteRef{
  691. RemoteKey: s1id,
  692. },
  693. func(vms *VaultManagementService) bool {
  694. return vms.VaultClient.(*fakeoracle.OracleMockVaultClient).DeletedCount == 1
  695. },
  696. },
  697. }
  698. for name, testCase := range testCases {
  699. t.Run(name, func(t *testing.T) {
  700. err := testCase.vms.DeleteSecret(context.Background(), testCase.remoteRef)
  701. assert.NoError(t, err)
  702. assert.True(t, testCase.validator(testCase.vms))
  703. })
  704. }
  705. }
  706. func TestOracleVaultSecretExists(t *testing.T) {
  707. var testCases = map[string]struct {
  708. vms *VaultManagementService
  709. remoteRef esv1.PushSecretRemoteRef
  710. ok bool
  711. }{
  712. "ok if secret exists": {
  713. &VaultManagementService{
  714. VaultClient: &fakeoracle.OracleMockVaultClient{},
  715. Client: &fakeoracle.OracleMockClient{
  716. SecretBundles: map[string]secrets.SecretBundle{
  717. s1id: s1bundle,
  718. },
  719. },
  720. },
  721. esv1alpha1.PushSecretRemoteRef{
  722. RemoteKey: s1id,
  723. },
  724. true,
  725. },
  726. "not ok if secret doesn't exist": {
  727. &VaultManagementService{
  728. VaultClient: &fakeoracle.OracleMockVaultClient{},
  729. Client: &fakeoracle.OracleMockClient{
  730. SecretBundles: map[string]secrets.SecretBundle{
  731. s1id: s1bundle,
  732. },
  733. },
  734. },
  735. esv1alpha1.PushSecretRemoteRef{
  736. RemoteKey: s2id,
  737. },
  738. false,
  739. },
  740. }
  741. for name, testCase := range testCases {
  742. t.Run(name, func(t *testing.T) {
  743. ok, err := testCase.vms.SecretExists(context.Background(), testCase.remoteRef)
  744. assert.NoError(t, err)
  745. assert.Equal(t, testCase.ok, ok)
  746. })
  747. }
  748. }
  749. var (
  750. s1id = "test1"
  751. s2id = "mysecret"
  752. s3id = "deleting"
  753. s1bundle = makeSecretBundle(s1id, false)
  754. s2bundle = makeSecretBundle(s2id, false)
  755. s3bundle = makeSecretBundle(s3id, true)
  756. s1summary = makeSecretSummary(s1id, false)
  757. s2summary = makeSecretSummary(s2id, false)
  758. s3summary = makeSecretSummary(s3id, true)
  759. )
  760. func makeSecretBundle(id string, deleting bool) secrets.SecretBundle {
  761. var deletionTime *common.SDKTime
  762. if deleting {
  763. deletionTime = &common.SDKTime{
  764. Time: time.Now(),
  765. }
  766. }
  767. return secrets.SecretBundle{
  768. SecretId: &id,
  769. SecretBundleContent: secrets.Base64SecretBundleContentDetails{
  770. Content: ptr.To(base64.StdEncoding.EncodeToString([]byte(id))),
  771. },
  772. TimeOfDeletion: deletionTime,
  773. }
  774. }
  775. func makeSecretSummary(id string, deleting bool) vault.SecretSummary {
  776. var deletionTime *common.SDKTime
  777. if deleting {
  778. deletionTime = &common.SDKTime{
  779. Time: time.Now(),
  780. }
  781. }
  782. return vault.SecretSummary{
  783. Id: &id,
  784. SecretName: &id,
  785. TimeOfDeletion: deletionTime,
  786. }
  787. }