oracle_test.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763
  1. /*
  2. Licensed under the Apache License, Version 2.0 (the "License");
  3. you may not use this file except in compliance with the License.
  4. You may obtain a copy of the License at
  5. http://www.apache.org/licenses/LICENSE-2.0
  6. Unless required by applicable law or agreed to in writing, software
  7. distributed under the License is distributed on an "AS IS" BASIS,
  8. WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  9. See the License for the specific language governing permissions and
  10. limitations under the License.
  11. */
  12. package oracle
  13. import (
  14. "context"
  15. "crypto/rand"
  16. "crypto/rsa"
  17. "crypto/x509"
  18. "encoding/base64"
  19. "encoding/pem"
  20. "fmt"
  21. "reflect"
  22. "strings"
  23. "testing"
  24. "time"
  25. "github.com/oracle/oci-go-sdk/v65/common"
  26. "github.com/oracle/oci-go-sdk/v65/secrets"
  27. "github.com/oracle/oci-go-sdk/v65/vault"
  28. "github.com/stretchr/testify/assert"
  29. corev1 "k8s.io/api/core/v1"
  30. metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
  31. "k8s.io/apimachinery/pkg/runtime"
  32. "k8s.io/utils/ptr"
  33. clientfake "sigs.k8s.io/controller-runtime/pkg/client/fake"
  34. esv1alpha1 "github.com/external-secrets/external-secrets/apis/externalsecrets/v1alpha1"
  35. esv1beta1 "github.com/external-secrets/external-secrets/apis/externalsecrets/v1beta1"
  36. esmeta "github.com/external-secrets/external-secrets/apis/meta/v1"
  37. fakeoracle "github.com/external-secrets/external-secrets/pkg/provider/oracle/fake"
  38. testingfake "github.com/external-secrets/external-secrets/pkg/provider/testing/fake"
  39. )
  40. const (
  41. vaultOCID = "vault-OCID"
  42. region = "some-region"
  43. tenant = "a-tenant"
  44. userOCID = "user-OCID"
  45. secretKey = "key"
  46. secretName = "name"
  47. )
  48. type vaultTestCase struct {
  49. mockClient *fakeoracle.OracleMockClient
  50. apiInput *secrets.GetSecretBundleByNameRequest
  51. apiOutput *secrets.GetSecretBundleByNameResponse
  52. ref *esv1beta1.ExternalSecretDataRemoteRef
  53. apiErr error
  54. expectError string
  55. expectedSecret string
  56. // for testing secretmap
  57. expectedData map[string][]byte
  58. }
  59. func makeValidVaultTestCase() *vaultTestCase {
  60. smtc := vaultTestCase{
  61. mockClient: &fakeoracle.OracleMockClient{},
  62. apiInput: makeValidAPIInput(),
  63. ref: makeValidRef(),
  64. apiOutput: makeValidAPIOutput(),
  65. apiErr: nil,
  66. expectError: "",
  67. expectedSecret: "",
  68. expectedData: map[string][]byte{},
  69. }
  70. smtc.mockClient.WithValue(*smtc.apiInput, *smtc.apiOutput, smtc.apiErr)
  71. return &smtc
  72. }
  73. func makeValidRef() *esv1beta1.ExternalSecretDataRemoteRef {
  74. return &esv1beta1.ExternalSecretDataRemoteRef{
  75. Key: "test-secret",
  76. Version: "default",
  77. }
  78. }
  79. func makeValidAPIInput() *secrets.GetSecretBundleByNameRequest {
  80. return &secrets.GetSecretBundleByNameRequest{
  81. SecretName: ptr.To("test-secret"),
  82. VaultId: ptr.To("test-vault"),
  83. }
  84. }
  85. func makeValidAPIOutput() *secrets.GetSecretBundleByNameResponse {
  86. return &secrets.GetSecretBundleByNameResponse{
  87. SecretBundle: secrets.SecretBundle{},
  88. }
  89. }
  90. func makeValidVaultTestCaseCustom(tweaks ...func(smtc *vaultTestCase)) *vaultTestCase {
  91. smtc := makeValidVaultTestCase()
  92. for _, fn := range tweaks {
  93. fn(smtc)
  94. }
  95. smtc.mockClient.WithValue(*smtc.apiInput, *smtc.apiOutput, smtc.apiErr)
  96. return smtc
  97. }
  98. // This case can be shared by both GetSecret and GetSecretMap tests.
  99. // bad case: set apiErr.
  100. var setAPIErr = func(smtc *vaultTestCase) {
  101. smtc.apiErr = fmt.Errorf("oh no")
  102. smtc.expectError = "oh no"
  103. }
  104. var setNilMockClient = func(smtc *vaultTestCase) {
  105. smtc.mockClient = nil
  106. smtc.expectError = errUninitalizedOracleProvider
  107. }
  108. func TestOracleVaultGetSecret(t *testing.T) {
  109. secretValue := "changedvalue"
  110. // good case: default version is set
  111. // key is passed in, output is sent back
  112. setSecretString := func(smtc *vaultTestCase) {
  113. smtc.apiOutput = &secrets.GetSecretBundleByNameResponse{
  114. SecretBundle: secrets.SecretBundle{
  115. SecretId: ptr.To("test-id"),
  116. VersionNumber: ptr.To(int64(1)),
  117. SecretBundleContent: secrets.Base64SecretBundleContentDetails{
  118. Content: ptr.To(base64.StdEncoding.EncodeToString([]byte(secretValue))),
  119. },
  120. },
  121. }
  122. smtc.expectedSecret = secretValue
  123. }
  124. successCases := []*vaultTestCase{
  125. makeValidVaultTestCaseCustom(setAPIErr),
  126. makeValidVaultTestCaseCustom(setNilMockClient),
  127. makeValidVaultTestCaseCustom(setSecretString),
  128. }
  129. sm := VaultManagementService{}
  130. for k, v := range successCases {
  131. sm.Client = v.mockClient
  132. fmt.Println(*v.ref)
  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(*esv1beta1.SecretStore) *esv1beta1.SecretStore
  185. func makeSecretStore(vault, region string, fn ...storeModifier) *esv1beta1.SecretStore {
  186. store := &esv1beta1.SecretStore{
  187. Spec: esv1beta1.SecretStoreSpec{
  188. Provider: &esv1beta1.SecretStoreProvider{
  189. Oracle: &esv1beta1.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 *esv1beta1.SecretStore) *esv1beta1.SecretStore {
  203. store.Spec.Provider.Oracle.Auth = &esv1beta1.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 *esv1beta1.SecretStore) *esv1beta1.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 *esv1beta1.SecretStore) *esv1beta1.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 *esv1beta1.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: fmt.Errorf("vault cannot be empty"),
  240. },
  241. {
  242. store: makeSecretStore(vaultOCID, ""),
  243. err: fmt.Errorf("region cannot be empty"),
  244. },
  245. {
  246. store: makeSecretStore(vaultOCID, region, withSecretAuth("", tenant)),
  247. err: fmt.Errorf("user cannot be empty"),
  248. },
  249. {
  250. store: makeSecretStore(vaultOCID, region, withSecretAuth(userOCID, "")),
  251. err: fmt.Errorf("tenant cannot be empty"),
  252. },
  253. {
  254. store: makeSecretStore(vaultOCID, region, withSecretAuth(userOCID, tenant), withPrivateKey("", secretKey, nil)),
  255. err: fmt.Errorf("privateKey.name cannot be empty"),
  256. },
  257. {
  258. store: makeSecretStore(vaultOCID, region, withSecretAuth(userOCID, tenant), withPrivateKey(secretName, secretKey, &namespace)),
  259. err: fmt.Errorf("namespace not allowed with namespaced SecretStore"),
  260. },
  261. {
  262. store: makeSecretStore(vaultOCID, region, withSecretAuth(userOCID, tenant), withPrivateKey(secretName, "", nil)),
  263. err: fmt.Errorf("privateKey.key cannot be empty"),
  264. },
  265. {
  266. store: makeSecretStore(vaultOCID, region, withSecretAuth(userOCID, tenant), withPrivateKey(secretName, secretKey, nil), withFingerprint("", secretKey, nil)),
  267. err: fmt.Errorf("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: fmt.Errorf("namespace not allowed with namespaced SecretStore"),
  272. },
  273. {
  274. store: makeSecretStore(vaultOCID, region, withSecretAuth(userOCID, tenant), withPrivateKey(secretName, secretKey, nil), withFingerprint(secretName, "", nil)),
  275. err: fmt.Errorf("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 TestVaultManagementService_NewClient(t *testing.T) {
  295. t.Parallel()
  296. namespace := "default"
  297. authSecretName := "oracle-auth"
  298. auth := &esv1beta1.OracleAuth{
  299. User: "user",
  300. Tenancy: "tenancy",
  301. SecretRef: esv1beta1.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 *esv1beta1.SecretStore
  315. expectedErr string
  316. }{
  317. {
  318. desc: "no retry settings",
  319. secretStore: &esv1beta1.SecretStore{
  320. Spec: esv1beta1.SecretStoreSpec{
  321. Provider: &esv1beta1.SecretStoreProvider{
  322. Oracle: &esv1beta1.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: &esv1beta1.SecretStore{
  334. Spec: esv1beta1.SecretStoreSpec{
  335. Provider: &esv1beta1.SecretStoreProvider{
  336. Oracle: &esv1beta1.OracleProvider{
  337. Vault: vaultOCID,
  338. Region: region,
  339. Auth: auth,
  340. },
  341. },
  342. RetrySettings: &esv1beta1.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: &esv1beta1.SecretStore{
  352. Spec: esv1beta1.SecretStoreSpec{
  353. Provider: &esv1beta1.SecretStoreProvider{
  354. Oracle: &esv1beta1.OracleProvider{
  355. Vault: vaultOCID,
  356. Region: region,
  357. Auth: auth,
  358. },
  359. },
  360. RetrySettings: &esv1beta1.SecretStoreRetrySettings{
  361. RetryInterval: ptr.To("1s"),
  362. },
  363. },
  364. },
  365. },
  366. {
  367. desc: "partially configure the retry settings - max retries",
  368. secretStore: &esv1beta1.SecretStore{
  369. Spec: esv1beta1.SecretStoreSpec{
  370. Provider: &esv1beta1.SecretStoreProvider{
  371. Oracle: &esv1beta1.OracleProvider{
  372. Vault: vaultOCID,
  373. Region: region,
  374. Auth: auth,
  375. },
  376. },
  377. RetrySettings: &esv1beta1.SecretStoreRetrySettings{
  378. MaxRetries: ptr.To(int32(5)),
  379. },
  380. },
  381. },
  382. },
  383. {
  384. desc: "auth secret does not exist",
  385. secretStore: &esv1beta1.SecretStore{
  386. Spec: esv1beta1.SecretStoreSpec{
  387. Provider: &esv1beta1.SecretStoreProvider{
  388. Oracle: &esv1beta1.OracleProvider{
  389. Vault: vaultOCID,
  390. Region: region,
  391. Auth: &esv1beta1.OracleAuth{
  392. User: "user",
  393. Tenancy: "tenancy",
  394. SecretRef: esv1beta1.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: &esv1beta1.SecretStoreRetrySettings{
  408. RetryInterval: ptr.To("invalid"),
  409. },
  410. },
  411. },
  412. expectedErr: `could not fetch SecretAccessKey secret: secrets "non-existing-secret"`,
  413. },
  414. {
  415. desc: "invalid retry interval",
  416. secretStore: &esv1beta1.SecretStore{
  417. Spec: esv1beta1.SecretStoreSpec{
  418. Provider: &esv1beta1.SecretStoreProvider{
  419. Oracle: &esv1beta1.OracleProvider{
  420. Vault: vaultOCID,
  421. Region: region,
  422. Auth: auth,
  423. },
  424. },
  425. RetrySettings: &esv1beta1.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 esv1beta1.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. esv1beta1.ExternalSecretFind{
  499. Name: &esv1beta1.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. esv1beta1.ExternalSecretFind{
  525. Name: &esv1beta1.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. var testCases = map[string]struct {
  546. vms *VaultManagementService
  547. data testingfake.PushSecretData
  548. validator func(service *VaultManagementService) bool
  549. content string
  550. }{
  551. "create a secret if not exists": {
  552. &VaultManagementService{
  553. Client: &fakeoracle.OracleMockClient{
  554. SecretBundles: map[string]secrets.SecretBundle{
  555. s2id: s2bundle,
  556. },
  557. },
  558. VaultClient: &fakeoracle.OracleMockVaultClient{},
  559. },
  560. testingfake.PushSecretData{
  561. SecretKey: testSecretKey,
  562. RemoteKey: s1id,
  563. },
  564. func(vms *VaultManagementService) bool {
  565. return vms.VaultClient.(*fakeoracle.OracleMockVaultClient).CreatedCount == 1
  566. },
  567. "created",
  568. },
  569. "update a secret if exists": {
  570. &VaultManagementService{
  571. Client: &fakeoracle.OracleMockClient{
  572. SecretBundles: map[string]secrets.SecretBundle{
  573. s1id: s1bundle,
  574. s2id: s2bundle,
  575. },
  576. },
  577. VaultClient: &fakeoracle.OracleMockVaultClient{},
  578. },
  579. testingfake.PushSecretData{
  580. SecretKey: testSecretKey,
  581. RemoteKey: s1id,
  582. },
  583. func(vms *VaultManagementService) bool {
  584. return vms.VaultClient.(*fakeoracle.OracleMockVaultClient).UpdatedCount == 1
  585. },
  586. "updated",
  587. },
  588. "neither create nor update if secret content is unchanged": {
  589. &VaultManagementService{
  590. Client: &fakeoracle.OracleMockClient{
  591. SecretBundles: map[string]secrets.SecretBundle{
  592. s1id: s1bundle,
  593. s2id: s2bundle,
  594. },
  595. },
  596. VaultClient: &fakeoracle.OracleMockVaultClient{},
  597. },
  598. testingfake.PushSecretData{
  599. SecretKey: testSecretKey,
  600. RemoteKey: s1id,
  601. },
  602. func(vms *VaultManagementService) bool {
  603. return vms.VaultClient.(*fakeoracle.OracleMockVaultClient).UpdatedCount == 0 &&
  604. vms.VaultClient.(*fakeoracle.OracleMockVaultClient).CreatedCount == 0
  605. },
  606. s1id,
  607. },
  608. }
  609. for name, testCase := range testCases {
  610. t.Run(name, func(t *testing.T) {
  611. s := &corev1.Secret{Data: map[string][]byte{testSecretKey: []byte(testCase.content)}}
  612. err := testCase.vms.PushSecret(context.Background(), s, testCase.data)
  613. assert.NoError(t, err)
  614. assert.True(t, testCase.validator(testCase.vms))
  615. })
  616. }
  617. }
  618. func TestOracleVaultDeleteSecret(t *testing.T) {
  619. var testCases = map[string]struct {
  620. vms *VaultManagementService
  621. remoteRef esv1beta1.PushSecretRemoteRef
  622. validator func(service *VaultManagementService) bool
  623. }{
  624. "do not delete if secret not found": {
  625. &VaultManagementService{
  626. Client: &fakeoracle.OracleMockClient{
  627. SecretBundles: map[string]secrets.SecretBundle{
  628. s1id: s1bundle,
  629. },
  630. },
  631. VaultClient: &fakeoracle.OracleMockVaultClient{},
  632. },
  633. esv1alpha1.PushSecretRemoteRef{
  634. RemoteKey: s2id,
  635. },
  636. func(vms *VaultManagementService) bool {
  637. return vms.VaultClient.(*fakeoracle.OracleMockVaultClient).DeletedCount == 0
  638. },
  639. },
  640. "do not delete if secret os already deleting": {
  641. &VaultManagementService{
  642. Client: &fakeoracle.OracleMockClient{
  643. SecretBundles: map[string]secrets.SecretBundle{
  644. s1id: s1bundle,
  645. s3id: s3bundle,
  646. },
  647. },
  648. VaultClient: &fakeoracle.OracleMockVaultClient{},
  649. },
  650. esv1alpha1.PushSecretRemoteRef{
  651. RemoteKey: s3id,
  652. },
  653. func(vms *VaultManagementService) bool {
  654. return vms.VaultClient.(*fakeoracle.OracleMockVaultClient).DeletedCount == 0
  655. },
  656. },
  657. "delete existing secret": {
  658. &VaultManagementService{
  659. Client: &fakeoracle.OracleMockClient{
  660. SecretBundles: map[string]secrets.SecretBundle{
  661. s1id: s1bundle,
  662. s3id: s3bundle,
  663. },
  664. },
  665. VaultClient: &fakeoracle.OracleMockVaultClient{},
  666. },
  667. esv1alpha1.PushSecretRemoteRef{
  668. RemoteKey: s1id,
  669. },
  670. func(vms *VaultManagementService) bool {
  671. return vms.VaultClient.(*fakeoracle.OracleMockVaultClient).DeletedCount == 1
  672. },
  673. },
  674. }
  675. for name, testCase := range testCases {
  676. t.Run(name, func(t *testing.T) {
  677. err := testCase.vms.DeleteSecret(context.Background(), testCase.remoteRef)
  678. assert.NoError(t, err)
  679. assert.True(t, testCase.validator(testCase.vms))
  680. })
  681. }
  682. }
  683. var (
  684. s1id = "test1"
  685. s2id = "mysecret"
  686. s3id = "deleting"
  687. s1bundle = makeSecretBundle(s1id, false)
  688. s2bundle = makeSecretBundle(s2id, false)
  689. s3bundle = makeSecretBundle(s3id, true)
  690. s1summary = makeSecretSummary(s1id, false)
  691. s2summary = makeSecretSummary(s2id, false)
  692. s3summary = makeSecretSummary(s3id, true)
  693. )
  694. func makeSecretBundle(id string, deleting bool) secrets.SecretBundle {
  695. var deletionTime *common.SDKTime
  696. if deleting {
  697. deletionTime = &common.SDKTime{
  698. Time: time.Now(),
  699. }
  700. }
  701. return secrets.SecretBundle{
  702. SecretId: &id,
  703. SecretBundleContent: secrets.Base64SecretBundleContentDetails{
  704. Content: ptr.To(base64.StdEncoding.EncodeToString([]byte(id))),
  705. },
  706. TimeOfDeletion: deletionTime,
  707. }
  708. }
  709. func makeSecretSummary(id string, deleting bool) vault.SecretSummary {
  710. var deletionTime *common.SDKTime
  711. if deleting {
  712. deletionTime = &common.SDKTime{
  713. Time: time.Now(),
  714. }
  715. }
  716. return vault.SecretSummary{
  717. Id: &id,
  718. SecretName: &id,
  719. TimeOfDeletion: deletionTime,
  720. }
  721. }