oracle_test.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788
  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. "errors"
  21. "fmt"
  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. esv1alpha1 "github.com/external-secrets/external-secrets/apis/externalsecrets/v1alpha1"
  36. esv1beta1 "github.com/external-secrets/external-secrets/apis/externalsecrets/v1beta1"
  37. esmeta "github.com/external-secrets/external-secrets/apis/meta/v1"
  38. fakeoracle "github.com/external-secrets/external-secrets/pkg/provider/oracle/fake"
  39. testingfake "github.com/external-secrets/external-secrets/pkg/provider/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 *esv1beta1.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() *esv1beta1.ExternalSecretDataRemoteRef {
  75. return &esv1beta1.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. fmt.Println(*v.ref)
  134. out, err := sm.GetSecret(context.Background(), *v.ref)
  135. if !ErrorContains(err, v.expectError) {
  136. t.Errorf("[%d] unexpected error: %s, expected: '%s'", k, err.Error(), v.expectError)
  137. }
  138. if string(out) != v.expectedSecret {
  139. t.Errorf("[%d] unexpected secret: expected %s, got %s", k, v.expectedSecret, string(out))
  140. }
  141. }
  142. }
  143. func TestGetSecretMap(t *testing.T) {
  144. // good case: default version & deserialization
  145. setDeserialization := func(smtc *vaultTestCase) {
  146. smtc.apiOutput.SecretBundleContent = secrets.Base64SecretBundleContentDetails{
  147. Content: ptr.To(base64.StdEncoding.EncodeToString([]byte(`{"foo":"bar"}`))),
  148. }
  149. smtc.expectedData["foo"] = []byte("bar")
  150. }
  151. // bad case: invalid json
  152. setInvalidJSON := func(smtc *vaultTestCase) {
  153. smtc.apiOutput.SecretBundleContent = secrets.Base64SecretBundleContentDetails{
  154. Content: ptr.To(base64.StdEncoding.EncodeToString([]byte(`-----------------`))),
  155. }
  156. smtc.expectError = "unable to unmarshal secret"
  157. }
  158. successCases := []*vaultTestCase{
  159. makeValidVaultTestCaseCustom(setDeserialization),
  160. makeValidVaultTestCaseCustom(setInvalidJSON),
  161. makeValidVaultTestCaseCustom(setNilMockClient),
  162. makeValidVaultTestCaseCustom(setAPIErr),
  163. }
  164. sm := VaultManagementService{}
  165. for k, v := range successCases {
  166. sm.Client = v.mockClient
  167. out, err := sm.GetSecretMap(context.Background(), *v.ref)
  168. if !ErrorContains(err, v.expectError) {
  169. t.Errorf("[%d] unexpected error: %s, expected: '%s'", k, err.Error(), v.expectError)
  170. }
  171. if err == nil && !reflect.DeepEqual(out, v.expectedData) {
  172. t.Errorf("[%d] unexpected secret data: expected %#v, got %#v", k, v.expectedData, out)
  173. }
  174. }
  175. }
  176. func ErrorContains(out error, want string) bool {
  177. if out == nil {
  178. return want == ""
  179. }
  180. if want == "" {
  181. return false
  182. }
  183. return strings.Contains(out.Error(), want)
  184. }
  185. type storeModifier func(*esv1beta1.SecretStore) *esv1beta1.SecretStore
  186. func makeSecretStore(vault, region string, fn ...storeModifier) *esv1beta1.SecretStore {
  187. store := &esv1beta1.SecretStore{
  188. Spec: esv1beta1.SecretStoreSpec{
  189. Provider: &esv1beta1.SecretStoreProvider{
  190. Oracle: &esv1beta1.OracleProvider{
  191. Vault: vault,
  192. Region: region,
  193. },
  194. },
  195. },
  196. }
  197. for _, f := range fn {
  198. store = f(store)
  199. }
  200. return store
  201. }
  202. func withSecretAuth(user, tenancy string) storeModifier {
  203. return func(store *esv1beta1.SecretStore) *esv1beta1.SecretStore {
  204. store.Spec.Provider.Oracle.Auth = &esv1beta1.OracleAuth{
  205. User: user,
  206. Tenancy: tenancy,
  207. }
  208. return store
  209. }
  210. }
  211. func withPrivateKey(name, key string, namespace *string) storeModifier {
  212. return func(store *esv1beta1.SecretStore) *esv1beta1.SecretStore {
  213. store.Spec.Provider.Oracle.Auth.SecretRef.PrivateKey = esmeta.SecretKeySelector{
  214. Name: name,
  215. Key: key,
  216. Namespace: namespace,
  217. }
  218. return store
  219. }
  220. }
  221. func withFingerprint(name, key string, namespace *string) storeModifier {
  222. return func(store *esv1beta1.SecretStore) *esv1beta1.SecretStore {
  223. store.Spec.Provider.Oracle.Auth.SecretRef.Fingerprint = esmeta.SecretKeySelector{
  224. Name: name,
  225. Key: key,
  226. Namespace: namespace,
  227. }
  228. return store
  229. }
  230. }
  231. type ValidateStoreTestCase struct {
  232. store *esv1beta1.SecretStore
  233. err error
  234. }
  235. func TestValidateStore(t *testing.T) {
  236. namespace := "my-namespace"
  237. testCases := []ValidateStoreTestCase{
  238. {
  239. store: makeSecretStore("", region),
  240. err: errors.New("vault cannot be empty"),
  241. },
  242. {
  243. store: makeSecretStore(vaultOCID, ""),
  244. err: errors.New("region cannot be empty"),
  245. },
  246. {
  247. store: makeSecretStore(vaultOCID, region, withSecretAuth("", tenant)),
  248. err: errors.New("user cannot be empty"),
  249. },
  250. {
  251. store: makeSecretStore(vaultOCID, region, withSecretAuth(userOCID, "")),
  252. err: errors.New("tenant cannot be empty"),
  253. },
  254. {
  255. store: makeSecretStore(vaultOCID, region, withSecretAuth(userOCID, tenant), withPrivateKey("", secretKey, nil)),
  256. err: errors.New("privateKey.name cannot be empty"),
  257. },
  258. {
  259. store: makeSecretStore(vaultOCID, region, withSecretAuth(userOCID, tenant), withPrivateKey(secretName, secretKey, &namespace)),
  260. err: errors.New("namespace should either be empty or match the namespace of the SecretStore for a namespaced SecretStore"),
  261. },
  262. {
  263. store: makeSecretStore(vaultOCID, region, withSecretAuth(userOCID, tenant), withPrivateKey(secretName, "", nil)),
  264. err: errors.New("privateKey.key cannot be empty"),
  265. },
  266. {
  267. store: makeSecretStore(vaultOCID, region, withSecretAuth(userOCID, tenant), withPrivateKey(secretName, secretKey, nil), withFingerprint("", secretKey, nil)),
  268. err: errors.New("fingerprint.name cannot be empty"),
  269. },
  270. {
  271. store: makeSecretStore(vaultOCID, region, withSecretAuth(userOCID, tenant), withPrivateKey(secretName, secretKey, nil), withFingerprint(secretName, secretKey, &namespace)),
  272. err: errors.New("namespace should either be empty or match the namespace of the SecretStore for a namespaced SecretStore"),
  273. },
  274. {
  275. store: makeSecretStore(vaultOCID, region, withSecretAuth(userOCID, tenant), withPrivateKey(secretName, secretKey, nil), withFingerprint(secretName, "", nil)),
  276. err: errors.New("fingerprint.key cannot be empty"),
  277. },
  278. {
  279. store: makeSecretStore(vaultOCID, region),
  280. err: nil,
  281. },
  282. }
  283. p := VaultManagementService{}
  284. for _, tc := range testCases {
  285. _, err := p.ValidateStore(tc.store)
  286. if tc.err != nil && err != nil && err.Error() != tc.err.Error() {
  287. t.Errorf("test failed! want %v, got %v", tc.err, err)
  288. } else if tc.err == nil && err != nil {
  289. t.Errorf("want nil got err %v", err)
  290. } else if tc.err != nil && err == nil {
  291. t.Errorf("want err %v got nil", tc.err)
  292. }
  293. }
  294. }
  295. func TestVaultManagementServiceNewClient(t *testing.T) {
  296. t.Parallel()
  297. namespace := "default"
  298. authSecretName := "oracle-auth"
  299. auth := &esv1beta1.OracleAuth{
  300. User: "user",
  301. Tenancy: "tenancy",
  302. SecretRef: esv1beta1.OracleSecretRef{
  303. PrivateKey: esmeta.SecretKeySelector{
  304. Name: authSecretName,
  305. Key: "privateKey",
  306. },
  307. Fingerprint: esmeta.SecretKeySelector{
  308. Name: authSecretName,
  309. Key: "fingerprint",
  310. },
  311. },
  312. }
  313. tests := []struct {
  314. desc string
  315. secretStore *esv1beta1.SecretStore
  316. expectedErr string
  317. }{
  318. {
  319. desc: "no retry settings",
  320. secretStore: &esv1beta1.SecretStore{
  321. Spec: esv1beta1.SecretStoreSpec{
  322. Provider: &esv1beta1.SecretStoreProvider{
  323. Oracle: &esv1beta1.OracleProvider{
  324. Vault: vaultOCID,
  325. Region: region,
  326. Auth: auth,
  327. },
  328. },
  329. },
  330. },
  331. },
  332. {
  333. desc: "fill all the retry settings",
  334. secretStore: &esv1beta1.SecretStore{
  335. Spec: esv1beta1.SecretStoreSpec{
  336. Provider: &esv1beta1.SecretStoreProvider{
  337. Oracle: &esv1beta1.OracleProvider{
  338. Vault: vaultOCID,
  339. Region: region,
  340. Auth: auth,
  341. },
  342. },
  343. RetrySettings: &esv1beta1.SecretStoreRetrySettings{
  344. RetryInterval: ptr.To("1s"),
  345. MaxRetries: ptr.To(int32(5)),
  346. },
  347. },
  348. },
  349. },
  350. {
  351. desc: "partially configure the retry settings - retry interval",
  352. secretStore: &esv1beta1.SecretStore{
  353. Spec: esv1beta1.SecretStoreSpec{
  354. Provider: &esv1beta1.SecretStoreProvider{
  355. Oracle: &esv1beta1.OracleProvider{
  356. Vault: vaultOCID,
  357. Region: region,
  358. Auth: auth,
  359. },
  360. },
  361. RetrySettings: &esv1beta1.SecretStoreRetrySettings{
  362. RetryInterval: ptr.To("1s"),
  363. },
  364. },
  365. },
  366. },
  367. {
  368. desc: "partially configure the retry settings - max retries",
  369. secretStore: &esv1beta1.SecretStore{
  370. Spec: esv1beta1.SecretStoreSpec{
  371. Provider: &esv1beta1.SecretStoreProvider{
  372. Oracle: &esv1beta1.OracleProvider{
  373. Vault: vaultOCID,
  374. Region: region,
  375. Auth: auth,
  376. },
  377. },
  378. RetrySettings: &esv1beta1.SecretStoreRetrySettings{
  379. MaxRetries: ptr.To(int32(5)),
  380. },
  381. },
  382. },
  383. },
  384. {
  385. desc: "auth secret does not exist",
  386. secretStore: &esv1beta1.SecretStore{
  387. Spec: esv1beta1.SecretStoreSpec{
  388. Provider: &esv1beta1.SecretStoreProvider{
  389. Oracle: &esv1beta1.OracleProvider{
  390. Vault: vaultOCID,
  391. Region: region,
  392. Auth: &esv1beta1.OracleAuth{
  393. User: "user",
  394. Tenancy: "tenancy",
  395. SecretRef: esv1beta1.OracleSecretRef{
  396. PrivateKey: esmeta.SecretKeySelector{
  397. Name: "non-existing-secret",
  398. Key: "privateKey",
  399. },
  400. Fingerprint: esmeta.SecretKeySelector{
  401. Name: "non-existing-secret",
  402. Key: "fingerprint",
  403. },
  404. },
  405. },
  406. },
  407. },
  408. RetrySettings: &esv1beta1.SecretStoreRetrySettings{
  409. RetryInterval: ptr.To("invalid"),
  410. },
  411. },
  412. },
  413. expectedErr: `cannot get Kubernetes secret "non-existing-secret" from namespace "default": secrets "non-existing-secret" not found`,
  414. },
  415. {
  416. desc: "invalid retry interval",
  417. secretStore: &esv1beta1.SecretStore{
  418. Spec: esv1beta1.SecretStoreSpec{
  419. Provider: &esv1beta1.SecretStoreProvider{
  420. Oracle: &esv1beta1.OracleProvider{
  421. Vault: vaultOCID,
  422. Region: region,
  423. Auth: auth,
  424. },
  425. },
  426. RetrySettings: &esv1beta1.SecretStoreRetrySettings{
  427. RetryInterval: ptr.To("invalid"),
  428. },
  429. },
  430. },
  431. expectedErr: "cannot setup new oracle client: time: invalid duration",
  432. },
  433. }
  434. for _, tc := range tests {
  435. t.Run(tc.desc, func(t *testing.T) {
  436. provider := &VaultManagementService{
  437. Client: &fakeoracle.OracleMockClient{},
  438. KmsVaultClient: nil,
  439. vault: vaultOCID,
  440. }
  441. pk, err := rsa.GenerateKey(rand.Reader, 2048)
  442. if err != nil {
  443. t.Fatalf("failed to create a private key: %v", err)
  444. }
  445. schema := runtime.NewScheme()
  446. if err := corev1.AddToScheme(schema); err != nil {
  447. t.Fatalf("failed to add to schema: %v", err)
  448. }
  449. builder := clientfake.NewClientBuilder().WithRuntimeObjects(&corev1.Secret{
  450. ObjectMeta: metav1.ObjectMeta{
  451. Name: authSecretName,
  452. Namespace: namespace,
  453. },
  454. Data: map[string][]byte{
  455. "privateKey": pem.EncodeToMemory(&pem.Block{
  456. Type: "RSA PRIVATE KEY",
  457. Bytes: x509.MarshalPKCS1PrivateKey(pk),
  458. }),
  459. "fingerprint": []byte("fingerprint"),
  460. },
  461. })
  462. _, err = provider.NewClient(context.Background(), tc.secretStore, builder.Build(), namespace)
  463. if err != nil {
  464. if tc.expectedErr == "" {
  465. t.Fatalf("failed to call NewClient: %v", err)
  466. }
  467. if !strings.Contains(err.Error(), tc.expectedErr) {
  468. t.Fatalf("received an unexpected error: %q should have contained %q", err.Error(), tc.expectedErr)
  469. }
  470. return
  471. }
  472. if tc.expectedErr != "" {
  473. t.Fatalf("expeceted to receive an error but got nil")
  474. }
  475. })
  476. }
  477. }
  478. func TestOracleVaultGetAllSecrets(t *testing.T) {
  479. var testCases = map[string]struct {
  480. vms *VaultManagementService
  481. ref esv1beta1.ExternalSecretFind
  482. result map[string][]byte
  483. }{
  484. "filters secrets that don't match the pattern": {
  485. &VaultManagementService{
  486. Client: &fakeoracle.OracleMockClient{
  487. SecretBundles: map[string]secrets.SecretBundle{
  488. s1id: s1bundle,
  489. s2id: s2bundle,
  490. },
  491. },
  492. VaultClient: &fakeoracle.OracleMockVaultClient{
  493. SecretSummaries: []vault.SecretSummary{
  494. s1summary,
  495. s2summary,
  496. },
  497. },
  498. },
  499. esv1beta1.ExternalSecretFind{
  500. Name: &esv1beta1.FindName{
  501. RegExp: "^test.*",
  502. },
  503. },
  504. map[string][]byte{
  505. s1id: []byte(s1id),
  506. },
  507. },
  508. "filters secrets that are deleting": {
  509. &VaultManagementService{
  510. Client: &fakeoracle.OracleMockClient{
  511. SecretBundles: map[string]secrets.SecretBundle{
  512. s1id: s1bundle,
  513. s2id: s2bundle,
  514. s3id: s3bundle,
  515. },
  516. },
  517. VaultClient: &fakeoracle.OracleMockVaultClient{
  518. SecretSummaries: []vault.SecretSummary{
  519. s1summary,
  520. s2summary,
  521. s3summary,
  522. },
  523. },
  524. },
  525. esv1beta1.ExternalSecretFind{
  526. Name: &esv1beta1.FindName{
  527. RegExp: ".*",
  528. },
  529. },
  530. map[string][]byte{
  531. s1id: []byte(s1id),
  532. s2id: []byte(s2id),
  533. },
  534. },
  535. }
  536. for name, testCase := range testCases {
  537. t.Run(name, func(t *testing.T) {
  538. result, err := testCase.vms.GetAllSecrets(context.Background(), testCase.ref)
  539. assert.NoError(t, err)
  540. assert.EqualValues(t, testCase.result, result)
  541. })
  542. }
  543. }
  544. func TestOracleVaultPushSecret(t *testing.T) {
  545. testSecretKey := "test-secret-key"
  546. encryptionKey := "must-not-be-blank-for-push"
  547. var testCases = map[string]struct {
  548. vms *VaultManagementService
  549. data testingfake.PushSecretData
  550. validator func(service *VaultManagementService) bool
  551. content string
  552. }{
  553. "create a secret if not exists": {
  554. &VaultManagementService{
  555. encryptionKey: encryptionKey,
  556. Client: &fakeoracle.OracleMockClient{
  557. SecretBundles: map[string]secrets.SecretBundle{
  558. s2id: s2bundle,
  559. },
  560. },
  561. VaultClient: &fakeoracle.OracleMockVaultClient{},
  562. },
  563. testingfake.PushSecretData{
  564. SecretKey: testSecretKey,
  565. RemoteKey: s1id,
  566. },
  567. func(vms *VaultManagementService) bool {
  568. return vms.VaultClient.(*fakeoracle.OracleMockVaultClient).CreatedCount == 1
  569. },
  570. "created",
  571. },
  572. "create a json secret if not exists": {
  573. &VaultManagementService{
  574. encryptionKey: encryptionKey,
  575. Client: &fakeoracle.OracleMockClient{
  576. SecretBundles: map[string]secrets.SecretBundle{
  577. s2id: s2bundle,
  578. },
  579. },
  580. VaultClient: &fakeoracle.OracleMockVaultClient{},
  581. },
  582. testingfake.PushSecretData{
  583. SecretKey: testSecretKey,
  584. RemoteKey: s1id,
  585. },
  586. func(vms *VaultManagementService) bool {
  587. return vms.VaultClient.(*fakeoracle.OracleMockVaultClient).CreatedCount == 1
  588. },
  589. "{'key-a':'secret-a', 'key-b': 'secret-b'}",
  590. },
  591. "update a secret if exists": {
  592. &VaultManagementService{
  593. encryptionKey: encryptionKey,
  594. Client: &fakeoracle.OracleMockClient{
  595. SecretBundles: map[string]secrets.SecretBundle{
  596. s1id: s1bundle,
  597. s2id: s2bundle,
  598. },
  599. },
  600. VaultClient: &fakeoracle.OracleMockVaultClient{},
  601. },
  602. testingfake.PushSecretData{
  603. SecretKey: testSecretKey,
  604. RemoteKey: s1id,
  605. },
  606. func(vms *VaultManagementService) bool {
  607. return vms.VaultClient.(*fakeoracle.OracleMockVaultClient).UpdatedCount == 1
  608. },
  609. "updated",
  610. },
  611. "neither create nor update if secret content is unchanged": {
  612. &VaultManagementService{
  613. encryptionKey: encryptionKey,
  614. Client: &fakeoracle.OracleMockClient{
  615. SecretBundles: map[string]secrets.SecretBundle{
  616. s1id: s1bundle,
  617. s2id: s2bundle,
  618. },
  619. },
  620. VaultClient: &fakeoracle.OracleMockVaultClient{},
  621. },
  622. testingfake.PushSecretData{
  623. SecretKey: testSecretKey,
  624. RemoteKey: s1id,
  625. },
  626. func(vms *VaultManagementService) bool {
  627. return vms.VaultClient.(*fakeoracle.OracleMockVaultClient).UpdatedCount == 0 &&
  628. vms.VaultClient.(*fakeoracle.OracleMockVaultClient).CreatedCount == 0
  629. },
  630. s1id,
  631. },
  632. }
  633. for name, testCase := range testCases {
  634. t.Run(name, func(t *testing.T) {
  635. s := &corev1.Secret{Data: map[string][]byte{testSecretKey: []byte(testCase.content)}}
  636. err := testCase.vms.PushSecret(context.Background(), s, testCase.data)
  637. assert.NoError(t, err)
  638. assert.True(t, testCase.validator(testCase.vms))
  639. })
  640. }
  641. }
  642. func TestOracleVaultDeleteSecret(t *testing.T) {
  643. var testCases = map[string]struct {
  644. vms *VaultManagementService
  645. remoteRef esv1beta1.PushSecretRemoteRef
  646. validator func(service *VaultManagementService) bool
  647. }{
  648. "do not delete if secret not found": {
  649. &VaultManagementService{
  650. Client: &fakeoracle.OracleMockClient{
  651. SecretBundles: map[string]secrets.SecretBundle{
  652. s1id: s1bundle,
  653. },
  654. },
  655. VaultClient: &fakeoracle.OracleMockVaultClient{},
  656. },
  657. esv1alpha1.PushSecretRemoteRef{
  658. RemoteKey: s2id,
  659. },
  660. func(vms *VaultManagementService) bool {
  661. return vms.VaultClient.(*fakeoracle.OracleMockVaultClient).DeletedCount == 0
  662. },
  663. },
  664. "do not delete if secret os already deleting": {
  665. &VaultManagementService{
  666. Client: &fakeoracle.OracleMockClient{
  667. SecretBundles: map[string]secrets.SecretBundle{
  668. s1id: s1bundle,
  669. s3id: s3bundle,
  670. },
  671. },
  672. VaultClient: &fakeoracle.OracleMockVaultClient{},
  673. },
  674. esv1alpha1.PushSecretRemoteRef{
  675. RemoteKey: s3id,
  676. },
  677. func(vms *VaultManagementService) bool {
  678. return vms.VaultClient.(*fakeoracle.OracleMockVaultClient).DeletedCount == 0
  679. },
  680. },
  681. "delete existing secret": {
  682. &VaultManagementService{
  683. Client: &fakeoracle.OracleMockClient{
  684. SecretBundles: map[string]secrets.SecretBundle{
  685. s1id: s1bundle,
  686. s3id: s3bundle,
  687. },
  688. },
  689. VaultClient: &fakeoracle.OracleMockVaultClient{},
  690. },
  691. esv1alpha1.PushSecretRemoteRef{
  692. RemoteKey: s1id,
  693. },
  694. func(vms *VaultManagementService) bool {
  695. return vms.VaultClient.(*fakeoracle.OracleMockVaultClient).DeletedCount == 1
  696. },
  697. },
  698. }
  699. for name, testCase := range testCases {
  700. t.Run(name, func(t *testing.T) {
  701. err := testCase.vms.DeleteSecret(context.Background(), testCase.remoteRef)
  702. assert.NoError(t, err)
  703. assert.True(t, testCase.validator(testCase.vms))
  704. })
  705. }
  706. }
  707. var (
  708. s1id = "test1"
  709. s2id = "mysecret"
  710. s3id = "deleting"
  711. s1bundle = makeSecretBundle(s1id, false)
  712. s2bundle = makeSecretBundle(s2id, false)
  713. s3bundle = makeSecretBundle(s3id, true)
  714. s1summary = makeSecretSummary(s1id, false)
  715. s2summary = makeSecretSummary(s2id, false)
  716. s3summary = makeSecretSummary(s3id, true)
  717. )
  718. func makeSecretBundle(id string, deleting bool) secrets.SecretBundle {
  719. var deletionTime *common.SDKTime
  720. if deleting {
  721. deletionTime = &common.SDKTime{
  722. Time: time.Now(),
  723. }
  724. }
  725. return secrets.SecretBundle{
  726. SecretId: &id,
  727. SecretBundleContent: secrets.Base64SecretBundleContentDetails{
  728. Content: ptr.To(base64.StdEncoding.EncodeToString([]byte(id))),
  729. },
  730. TimeOfDeletion: deletionTime,
  731. }
  732. }
  733. func makeSecretSummary(id string, deleting bool) vault.SecretSummary {
  734. var deletionTime *common.SDKTime
  735. if deleting {
  736. deletionTime = &common.SDKTime{
  737. Time: time.Now(),
  738. }
  739. }
  740. return vault.SecretSummary{
  741. Id: &id,
  742. SecretName: &id,
  743. TimeOfDeletion: deletionTime,
  744. }
  745. }