tokens_controller_test.go 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666
  1. /*
  2. Copyright 2014 The Kubernetes Authors.
  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. http://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 serviceaccount
  14. import (
  15. "errors"
  16. "reflect"
  17. "testing"
  18. "time"
  19. "github.com/golang/glog"
  20. "k8s.io/kubernetes/pkg/api"
  21. apierrors "k8s.io/kubernetes/pkg/api/errors"
  22. "k8s.io/kubernetes/pkg/api/unversioned"
  23. "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/fake"
  24. "k8s.io/kubernetes/pkg/client/testing/core"
  25. "k8s.io/kubernetes/pkg/runtime"
  26. utilrand "k8s.io/kubernetes/pkg/util/rand"
  27. )
  28. type testGenerator struct {
  29. GeneratedServiceAccounts []api.ServiceAccount
  30. GeneratedSecrets []api.Secret
  31. Token string
  32. Err error
  33. }
  34. func (t *testGenerator) GenerateToken(serviceAccount api.ServiceAccount, secret api.Secret) (string, error) {
  35. t.GeneratedSecrets = append(t.GeneratedSecrets, secret)
  36. t.GeneratedServiceAccounts = append(t.GeneratedServiceAccounts, serviceAccount)
  37. return t.Token, t.Err
  38. }
  39. // emptySecretReferences is used by a service account without any secrets
  40. func emptySecretReferences() []api.ObjectReference {
  41. return []api.ObjectReference{}
  42. }
  43. // missingSecretReferences is used by a service account that references secrets which do no exist
  44. func missingSecretReferences() []api.ObjectReference {
  45. return []api.ObjectReference{{Name: "missing-secret-1"}}
  46. }
  47. // regularSecretReferences is used by a service account that references secrets which are not ServiceAccountTokens
  48. func regularSecretReferences() []api.ObjectReference {
  49. return []api.ObjectReference{{Name: "regular-secret-1"}}
  50. }
  51. // tokenSecretReferences is used by a service account that references a ServiceAccountToken secret
  52. func tokenSecretReferences() []api.ObjectReference {
  53. return []api.ObjectReference{{Name: "token-secret-1"}}
  54. }
  55. // addTokenSecretReference adds a reference to the ServiceAccountToken that will be created
  56. func addTokenSecretReference(refs []api.ObjectReference) []api.ObjectReference {
  57. return addNamedTokenSecretReference(refs, "default-token-fplln")
  58. }
  59. // addNamedTokenSecretReference adds a reference to the named ServiceAccountToken
  60. func addNamedTokenSecretReference(refs []api.ObjectReference, name string) []api.ObjectReference {
  61. return append(refs, api.ObjectReference{Name: name})
  62. }
  63. // serviceAccount returns a service account with the given secret refs
  64. func serviceAccount(secretRefs []api.ObjectReference) *api.ServiceAccount {
  65. return &api.ServiceAccount{
  66. ObjectMeta: api.ObjectMeta{
  67. Name: "default",
  68. UID: "12345",
  69. Namespace: "default",
  70. ResourceVersion: "1",
  71. },
  72. Secrets: secretRefs,
  73. }
  74. }
  75. // updatedServiceAccount returns a service account with the resource version modified
  76. func updatedServiceAccount(secretRefs []api.ObjectReference) *api.ServiceAccount {
  77. sa := serviceAccount(secretRefs)
  78. sa.ResourceVersion = "2"
  79. return sa
  80. }
  81. // opaqueSecret returns a persisted non-ServiceAccountToken secret named "regular-secret-1"
  82. func opaqueSecret() *api.Secret {
  83. return &api.Secret{
  84. ObjectMeta: api.ObjectMeta{
  85. Name: "regular-secret-1",
  86. Namespace: "default",
  87. UID: "23456",
  88. ResourceVersion: "1",
  89. },
  90. Type: "Opaque",
  91. Data: map[string][]byte{
  92. "mykey": []byte("mydata"),
  93. },
  94. }
  95. }
  96. // createdTokenSecret returns the ServiceAccountToken secret posted when creating a new token secret.
  97. // Named "default-token-fplln", since that is the first generated name after rand.Seed(1)
  98. func createdTokenSecret(overrideName ...string) *api.Secret {
  99. return namedCreatedTokenSecret("default-token-fplln")
  100. }
  101. // namedTokenSecret returns the ServiceAccountToken secret posted when creating a new token secret with the given name.
  102. func namedCreatedTokenSecret(name string) *api.Secret {
  103. return &api.Secret{
  104. ObjectMeta: api.ObjectMeta{
  105. Name: name,
  106. Namespace: "default",
  107. Annotations: map[string]string{
  108. api.ServiceAccountNameKey: "default",
  109. api.ServiceAccountUIDKey: "12345",
  110. },
  111. },
  112. Type: api.SecretTypeServiceAccountToken,
  113. Data: map[string][]byte{
  114. "token": []byte("ABC"),
  115. "ca.crt": []byte("CA Data"),
  116. "namespace": []byte("default"),
  117. },
  118. }
  119. }
  120. // serviceAccountTokenSecret returns an existing ServiceAccountToken secret named "token-secret-1"
  121. func serviceAccountTokenSecret() *api.Secret {
  122. return &api.Secret{
  123. ObjectMeta: api.ObjectMeta{
  124. Name: "token-secret-1",
  125. Namespace: "default",
  126. UID: "23456",
  127. ResourceVersion: "1",
  128. Annotations: map[string]string{
  129. api.ServiceAccountNameKey: "default",
  130. api.ServiceAccountUIDKey: "12345",
  131. },
  132. },
  133. Type: api.SecretTypeServiceAccountToken,
  134. Data: map[string][]byte{
  135. "token": []byte("ABC"),
  136. "ca.crt": []byte("CA Data"),
  137. "namespace": []byte("default"),
  138. },
  139. }
  140. }
  141. // serviceAccountTokenSecretWithoutTokenData returns an existing ServiceAccountToken secret that lacks token data
  142. func serviceAccountTokenSecretWithoutTokenData() *api.Secret {
  143. secret := serviceAccountTokenSecret()
  144. delete(secret.Data, api.ServiceAccountTokenKey)
  145. return secret
  146. }
  147. // serviceAccountTokenSecretWithoutCAData returns an existing ServiceAccountToken secret that lacks ca data
  148. func serviceAccountTokenSecretWithoutCAData() *api.Secret {
  149. secret := serviceAccountTokenSecret()
  150. delete(secret.Data, api.ServiceAccountRootCAKey)
  151. return secret
  152. }
  153. // serviceAccountTokenSecretWithCAData returns an existing ServiceAccountToken secret with the specified ca data
  154. func serviceAccountTokenSecretWithCAData(data []byte) *api.Secret {
  155. secret := serviceAccountTokenSecret()
  156. secret.Data[api.ServiceAccountRootCAKey] = data
  157. return secret
  158. }
  159. // serviceAccountTokenSecretWithoutNamespaceData returns an existing ServiceAccountToken secret that lacks namespace data
  160. func serviceAccountTokenSecretWithoutNamespaceData() *api.Secret {
  161. secret := serviceAccountTokenSecret()
  162. delete(secret.Data, api.ServiceAccountNamespaceKey)
  163. return secret
  164. }
  165. // serviceAccountTokenSecretWithNamespaceData returns an existing ServiceAccountToken secret with the specified namespace data
  166. func serviceAccountTokenSecretWithNamespaceData(data []byte) *api.Secret {
  167. secret := serviceAccountTokenSecret()
  168. secret.Data[api.ServiceAccountNamespaceKey] = data
  169. return secret
  170. }
  171. type reaction struct {
  172. verb string
  173. resource string
  174. reactor func(t *testing.T) core.ReactionFunc
  175. }
  176. func TestTokenCreation(t *testing.T) {
  177. testcases := map[string]struct {
  178. ClientObjects []runtime.Object
  179. IsAsync bool
  180. MaxRetries int
  181. Reactors []reaction
  182. ExistingServiceAccount *api.ServiceAccount
  183. ExistingSecrets []*api.Secret
  184. AddedServiceAccount *api.ServiceAccount
  185. UpdatedServiceAccount *api.ServiceAccount
  186. DeletedServiceAccount *api.ServiceAccount
  187. AddedSecret *api.Secret
  188. UpdatedSecret *api.Secret
  189. DeletedSecret *api.Secret
  190. ExpectedActions []core.Action
  191. }{
  192. "new serviceaccount with no secrets": {
  193. ClientObjects: []runtime.Object{serviceAccount(emptySecretReferences())},
  194. AddedServiceAccount: serviceAccount(emptySecretReferences()),
  195. ExpectedActions: []core.Action{
  196. core.NewGetAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, "default"),
  197. core.NewCreateAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, createdTokenSecret()),
  198. core.NewUpdateAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, serviceAccount(addTokenSecretReference(emptySecretReferences()))),
  199. },
  200. },
  201. "new serviceaccount with no secrets encountering create error": {
  202. ClientObjects: []runtime.Object{serviceAccount(emptySecretReferences())},
  203. MaxRetries: 10,
  204. IsAsync: true,
  205. Reactors: []reaction{{
  206. verb: "create",
  207. resource: "secrets",
  208. reactor: func(t *testing.T) core.ReactionFunc {
  209. i := 0
  210. return func(core.Action) (bool, runtime.Object, error) {
  211. i++
  212. if i < 3 {
  213. return true, nil, apierrors.NewForbidden(api.Resource("secrets"), "foo", errors.New("No can do"))
  214. }
  215. return false, nil, nil
  216. }
  217. },
  218. }},
  219. AddedServiceAccount: serviceAccount(emptySecretReferences()),
  220. ExpectedActions: []core.Action{
  221. // Attempt 1
  222. core.NewGetAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, "default"),
  223. core.NewCreateAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, createdTokenSecret()),
  224. // Attempt 2
  225. core.NewGetAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, "default"),
  226. core.NewCreateAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, namedCreatedTokenSecret("default-token-gziey")),
  227. // Attempt 3
  228. core.NewGetAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, "default"),
  229. core.NewCreateAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, namedCreatedTokenSecret("default-token-oh43e")),
  230. core.NewUpdateAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, serviceAccount(addNamedTokenSecretReference(emptySecretReferences(), "default-token-oh43e"))),
  231. },
  232. },
  233. "new serviceaccount with no secrets encountering unending create error": {
  234. ClientObjects: []runtime.Object{serviceAccount(emptySecretReferences()), createdTokenSecret()},
  235. MaxRetries: 2,
  236. IsAsync: true,
  237. Reactors: []reaction{{
  238. verb: "create",
  239. resource: "secrets",
  240. reactor: func(t *testing.T) core.ReactionFunc {
  241. return func(core.Action) (bool, runtime.Object, error) {
  242. return true, nil, apierrors.NewForbidden(api.Resource("secrets"), "foo", errors.New("No can do"))
  243. }
  244. },
  245. }},
  246. AddedServiceAccount: serviceAccount(emptySecretReferences()),
  247. ExpectedActions: []core.Action{
  248. // Attempt
  249. core.NewGetAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, "default"),
  250. core.NewCreateAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, createdTokenSecret()),
  251. // Retry 1
  252. core.NewGetAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, "default"),
  253. core.NewCreateAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, namedCreatedTokenSecret("default-token-gziey")),
  254. // Retry 2
  255. core.NewGetAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, "default"),
  256. core.NewCreateAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, namedCreatedTokenSecret("default-token-oh43e")),
  257. },
  258. },
  259. "new serviceaccount with missing secrets": {
  260. ClientObjects: []runtime.Object{serviceAccount(missingSecretReferences())},
  261. AddedServiceAccount: serviceAccount(missingSecretReferences()),
  262. ExpectedActions: []core.Action{
  263. core.NewGetAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, "default"),
  264. core.NewCreateAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, createdTokenSecret()),
  265. core.NewUpdateAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, serviceAccount(addTokenSecretReference(missingSecretReferences()))),
  266. },
  267. },
  268. "new serviceaccount with non-token secrets": {
  269. ClientObjects: []runtime.Object{serviceAccount(regularSecretReferences()), opaqueSecret()},
  270. AddedServiceAccount: serviceAccount(regularSecretReferences()),
  271. ExpectedActions: []core.Action{
  272. core.NewGetAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, "default"),
  273. core.NewCreateAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, createdTokenSecret()),
  274. core.NewUpdateAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, serviceAccount(addTokenSecretReference(regularSecretReferences()))),
  275. },
  276. },
  277. "new serviceaccount with token secrets": {
  278. ClientObjects: []runtime.Object{serviceAccount(tokenSecretReferences()), serviceAccountTokenSecret()},
  279. ExistingSecrets: []*api.Secret{serviceAccountTokenSecret()},
  280. AddedServiceAccount: serviceAccount(tokenSecretReferences()),
  281. ExpectedActions: []core.Action{},
  282. },
  283. "new serviceaccount with no secrets with resource conflict": {
  284. ClientObjects: []runtime.Object{updatedServiceAccount(emptySecretReferences()), createdTokenSecret()},
  285. AddedServiceAccount: serviceAccount(emptySecretReferences()),
  286. ExpectedActions: []core.Action{
  287. core.NewGetAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, "default"),
  288. },
  289. },
  290. "updated serviceaccount with no secrets": {
  291. ClientObjects: []runtime.Object{serviceAccount(emptySecretReferences())},
  292. UpdatedServiceAccount: serviceAccount(emptySecretReferences()),
  293. ExpectedActions: []core.Action{
  294. core.NewGetAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, "default"),
  295. core.NewCreateAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, createdTokenSecret()),
  296. core.NewUpdateAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, serviceAccount(addTokenSecretReference(emptySecretReferences()))),
  297. },
  298. },
  299. "updated serviceaccount with missing secrets": {
  300. ClientObjects: []runtime.Object{serviceAccount(missingSecretReferences())},
  301. UpdatedServiceAccount: serviceAccount(missingSecretReferences()),
  302. ExpectedActions: []core.Action{
  303. core.NewGetAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, "default"),
  304. core.NewCreateAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, createdTokenSecret()),
  305. core.NewUpdateAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, serviceAccount(addTokenSecretReference(missingSecretReferences()))),
  306. },
  307. },
  308. "updated serviceaccount with non-token secrets": {
  309. ClientObjects: []runtime.Object{serviceAccount(regularSecretReferences()), opaqueSecret()},
  310. UpdatedServiceAccount: serviceAccount(regularSecretReferences()),
  311. ExpectedActions: []core.Action{
  312. core.NewGetAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, "default"),
  313. core.NewCreateAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, createdTokenSecret()),
  314. core.NewUpdateAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, serviceAccount(addTokenSecretReference(regularSecretReferences()))),
  315. },
  316. },
  317. "updated serviceaccount with token secrets": {
  318. ExistingSecrets: []*api.Secret{serviceAccountTokenSecret()},
  319. UpdatedServiceAccount: serviceAccount(tokenSecretReferences()),
  320. ExpectedActions: []core.Action{},
  321. },
  322. "updated serviceaccount with no secrets with resource conflict": {
  323. ClientObjects: []runtime.Object{updatedServiceAccount(emptySecretReferences())},
  324. UpdatedServiceAccount: serviceAccount(emptySecretReferences()),
  325. ExpectedActions: []core.Action{
  326. core.NewGetAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, "default"),
  327. },
  328. },
  329. "deleted serviceaccount with no secrets": {
  330. DeletedServiceAccount: serviceAccount(emptySecretReferences()),
  331. ExpectedActions: []core.Action{},
  332. },
  333. "deleted serviceaccount with missing secrets": {
  334. DeletedServiceAccount: serviceAccount(missingSecretReferences()),
  335. ExpectedActions: []core.Action{},
  336. },
  337. "deleted serviceaccount with non-token secrets": {
  338. ClientObjects: []runtime.Object{opaqueSecret()},
  339. DeletedServiceAccount: serviceAccount(regularSecretReferences()),
  340. ExpectedActions: []core.Action{},
  341. },
  342. "deleted serviceaccount with token secrets": {
  343. ClientObjects: []runtime.Object{serviceAccountTokenSecret()},
  344. ExistingSecrets: []*api.Secret{serviceAccountTokenSecret()},
  345. DeletedServiceAccount: serviceAccount(tokenSecretReferences()),
  346. ExpectedActions: []core.Action{
  347. core.NewDeleteAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, "token-secret-1"),
  348. },
  349. },
  350. "added secret without serviceaccount": {
  351. ClientObjects: []runtime.Object{serviceAccountTokenSecret()},
  352. AddedSecret: serviceAccountTokenSecret(),
  353. ExpectedActions: []core.Action{
  354. core.NewGetAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, "default"),
  355. core.NewDeleteAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, "token-secret-1"),
  356. },
  357. },
  358. "added secret with serviceaccount": {
  359. ExistingServiceAccount: serviceAccount(tokenSecretReferences()),
  360. AddedSecret: serviceAccountTokenSecret(),
  361. ExpectedActions: []core.Action{},
  362. },
  363. "added token secret without token data": {
  364. ClientObjects: []runtime.Object{serviceAccountTokenSecretWithoutTokenData()},
  365. ExistingServiceAccount: serviceAccount(tokenSecretReferences()),
  366. AddedSecret: serviceAccountTokenSecretWithoutTokenData(),
  367. ExpectedActions: []core.Action{
  368. core.NewGetAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, "token-secret-1"),
  369. core.NewUpdateAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, serviceAccountTokenSecret()),
  370. },
  371. },
  372. "added token secret without ca data": {
  373. ClientObjects: []runtime.Object{serviceAccountTokenSecretWithoutCAData()},
  374. ExistingServiceAccount: serviceAccount(tokenSecretReferences()),
  375. AddedSecret: serviceAccountTokenSecretWithoutCAData(),
  376. ExpectedActions: []core.Action{
  377. core.NewGetAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, "token-secret-1"),
  378. core.NewUpdateAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, serviceAccountTokenSecret()),
  379. },
  380. },
  381. "added token secret with mismatched ca data": {
  382. ClientObjects: []runtime.Object{serviceAccountTokenSecretWithCAData([]byte("mismatched"))},
  383. ExistingServiceAccount: serviceAccount(tokenSecretReferences()),
  384. AddedSecret: serviceAccountTokenSecretWithCAData([]byte("mismatched")),
  385. ExpectedActions: []core.Action{
  386. core.NewGetAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, "token-secret-1"),
  387. core.NewUpdateAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, serviceAccountTokenSecret()),
  388. },
  389. },
  390. "added token secret without namespace data": {
  391. ClientObjects: []runtime.Object{serviceAccountTokenSecretWithoutNamespaceData()},
  392. ExistingServiceAccount: serviceAccount(tokenSecretReferences()),
  393. AddedSecret: serviceAccountTokenSecretWithoutNamespaceData(),
  394. ExpectedActions: []core.Action{
  395. core.NewGetAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, "token-secret-1"),
  396. core.NewUpdateAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, serviceAccountTokenSecret()),
  397. },
  398. },
  399. "added token secret with custom namespace data": {
  400. ClientObjects: []runtime.Object{serviceAccountTokenSecretWithNamespaceData([]byte("custom"))},
  401. ExistingServiceAccount: serviceAccount(tokenSecretReferences()),
  402. AddedSecret: serviceAccountTokenSecretWithNamespaceData([]byte("custom")),
  403. ExpectedActions: []core.Action{
  404. // no update is performed... the custom namespace is preserved
  405. },
  406. },
  407. "updated secret without serviceaccount": {
  408. ClientObjects: []runtime.Object{serviceAccountTokenSecret()},
  409. UpdatedSecret: serviceAccountTokenSecret(),
  410. ExpectedActions: []core.Action{
  411. core.NewGetAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, "default"),
  412. core.NewDeleteAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, "token-secret-1"),
  413. },
  414. },
  415. "updated secret with serviceaccount": {
  416. ExistingServiceAccount: serviceAccount(tokenSecretReferences()),
  417. UpdatedSecret: serviceAccountTokenSecret(),
  418. ExpectedActions: []core.Action{},
  419. },
  420. "updated token secret without token data": {
  421. ClientObjects: []runtime.Object{serviceAccountTokenSecretWithoutTokenData()},
  422. ExistingServiceAccount: serviceAccount(tokenSecretReferences()),
  423. UpdatedSecret: serviceAccountTokenSecretWithoutTokenData(),
  424. ExpectedActions: []core.Action{
  425. core.NewGetAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, "token-secret-1"),
  426. core.NewUpdateAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, serviceAccountTokenSecret()),
  427. },
  428. },
  429. "updated token secret without ca data": {
  430. ClientObjects: []runtime.Object{serviceAccountTokenSecretWithoutCAData()},
  431. ExistingServiceAccount: serviceAccount(tokenSecretReferences()),
  432. UpdatedSecret: serviceAccountTokenSecretWithoutCAData(),
  433. ExpectedActions: []core.Action{
  434. core.NewGetAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, "token-secret-1"),
  435. core.NewUpdateAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, serviceAccountTokenSecret()),
  436. },
  437. },
  438. "updated token secret with mismatched ca data": {
  439. ClientObjects: []runtime.Object{serviceAccountTokenSecretWithCAData([]byte("mismatched"))},
  440. ExistingServiceAccount: serviceAccount(tokenSecretReferences()),
  441. UpdatedSecret: serviceAccountTokenSecretWithCAData([]byte("mismatched")),
  442. ExpectedActions: []core.Action{
  443. core.NewGetAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, "token-secret-1"),
  444. core.NewUpdateAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, serviceAccountTokenSecret()),
  445. },
  446. },
  447. "updated token secret without namespace data": {
  448. ClientObjects: []runtime.Object{serviceAccountTokenSecretWithoutNamespaceData()},
  449. ExistingServiceAccount: serviceAccount(tokenSecretReferences()),
  450. UpdatedSecret: serviceAccountTokenSecretWithoutNamespaceData(),
  451. ExpectedActions: []core.Action{
  452. core.NewGetAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, "token-secret-1"),
  453. core.NewUpdateAction(unversioned.GroupVersionResource{Resource: "secrets"}, api.NamespaceDefault, serviceAccountTokenSecret()),
  454. },
  455. },
  456. "updated token secret with custom namespace data": {
  457. ClientObjects: []runtime.Object{serviceAccountTokenSecretWithNamespaceData([]byte("custom"))},
  458. ExistingServiceAccount: serviceAccount(tokenSecretReferences()),
  459. UpdatedSecret: serviceAccountTokenSecretWithNamespaceData([]byte("custom")),
  460. ExpectedActions: []core.Action{
  461. // no update is performed... the custom namespace is preserved
  462. },
  463. },
  464. "deleted secret without serviceaccount": {
  465. DeletedSecret: serviceAccountTokenSecret(),
  466. ExpectedActions: []core.Action{},
  467. },
  468. "deleted secret with serviceaccount with reference": {
  469. ClientObjects: []runtime.Object{serviceAccount(tokenSecretReferences())},
  470. ExistingServiceAccount: serviceAccount(tokenSecretReferences()),
  471. DeletedSecret: serviceAccountTokenSecret(),
  472. ExpectedActions: []core.Action{
  473. core.NewGetAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, "default"),
  474. core.NewUpdateAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, serviceAccount(emptySecretReferences())),
  475. },
  476. },
  477. "deleted secret with serviceaccount without reference": {
  478. ExistingServiceAccount: serviceAccount(emptySecretReferences()),
  479. DeletedSecret: serviceAccountTokenSecret(),
  480. ExpectedActions: []core.Action{
  481. core.NewGetAction(unversioned.GroupVersionResource{Resource: "serviceaccounts"}, api.NamespaceDefault, "default"),
  482. },
  483. },
  484. }
  485. for k, tc := range testcases {
  486. glog.Infof(k)
  487. // Re-seed to reset name generation
  488. utilrand.Seed(1)
  489. generator := &testGenerator{Token: "ABC"}
  490. client := fake.NewSimpleClientset(tc.ClientObjects...)
  491. for _, reactor := range tc.Reactors {
  492. client.Fake.PrependReactor(reactor.verb, reactor.resource, reactor.reactor(t))
  493. }
  494. controller := NewTokensController(client, TokensControllerOptions{TokenGenerator: generator, RootCA: []byte("CA Data"), MaxRetries: tc.MaxRetries})
  495. if tc.ExistingServiceAccount != nil {
  496. controller.serviceAccounts.Add(tc.ExistingServiceAccount)
  497. }
  498. for _, s := range tc.ExistingSecrets {
  499. controller.secrets.Add(s)
  500. }
  501. if tc.AddedServiceAccount != nil {
  502. controller.serviceAccounts.Add(tc.AddedServiceAccount)
  503. controller.queueServiceAccountSync(tc.AddedServiceAccount)
  504. }
  505. if tc.UpdatedServiceAccount != nil {
  506. controller.serviceAccounts.Add(tc.UpdatedServiceAccount)
  507. controller.queueServiceAccountUpdateSync(nil, tc.UpdatedServiceAccount)
  508. }
  509. if tc.DeletedServiceAccount != nil {
  510. controller.serviceAccounts.Delete(tc.DeletedServiceAccount)
  511. controller.queueServiceAccountSync(tc.DeletedServiceAccount)
  512. }
  513. if tc.AddedSecret != nil {
  514. controller.secrets.Add(tc.AddedSecret)
  515. controller.queueSecretSync(tc.AddedSecret)
  516. }
  517. if tc.UpdatedSecret != nil {
  518. controller.secrets.Add(tc.UpdatedSecret)
  519. controller.queueSecretUpdateSync(nil, tc.UpdatedSecret)
  520. }
  521. if tc.DeletedSecret != nil {
  522. controller.secrets.Delete(tc.DeletedSecret)
  523. controller.queueSecretSync(tc.DeletedSecret)
  524. }
  525. // This is the longest we'll wait for async tests
  526. timeout := time.Now().Add(30 * time.Second)
  527. waitedForAdditionalActions := false
  528. for {
  529. if controller.syncServiceAccountQueue.Len() > 0 {
  530. controller.syncServiceAccount()
  531. }
  532. if controller.syncSecretQueue.Len() > 0 {
  533. controller.syncSecret()
  534. }
  535. // The queues still have things to work on
  536. if controller.syncServiceAccountQueue.Len() > 0 || controller.syncSecretQueue.Len() > 0 {
  537. continue
  538. }
  539. // If we expect this test to work asynchronously...
  540. if tc.IsAsync {
  541. // if we're still missing expected actions within our test timeout
  542. if len(client.Actions()) < len(tc.ExpectedActions) && time.Now().Before(timeout) {
  543. // wait for the expected actions (without hotlooping)
  544. time.Sleep(time.Millisecond)
  545. continue
  546. }
  547. // if we exactly match our expected actions, wait a bit to make sure no other additional actions show up
  548. if len(client.Actions()) == len(tc.ExpectedActions) && !waitedForAdditionalActions {
  549. time.Sleep(time.Second)
  550. waitedForAdditionalActions = true
  551. continue
  552. }
  553. }
  554. break
  555. }
  556. if controller.syncServiceAccountQueue.Len() > 0 {
  557. t.Errorf("%s: unexpected items in service account queue: %d", k, controller.syncServiceAccountQueue.Len())
  558. }
  559. if controller.syncSecretQueue.Len() > 0 {
  560. t.Errorf("%s: unexpected items in secret queue: %d", k, controller.syncSecretQueue.Len())
  561. }
  562. actions := client.Actions()
  563. for i, action := range actions {
  564. if len(tc.ExpectedActions) < i+1 {
  565. t.Errorf("%s: %d unexpected actions: %+v", k, len(actions)-len(tc.ExpectedActions), actions[i:])
  566. break
  567. }
  568. expectedAction := tc.ExpectedActions[i]
  569. if !reflect.DeepEqual(expectedAction, action) {
  570. t.Errorf("%s: Expected\n\t%#v\ngot\n\t%#v", k, expectedAction, action)
  571. continue
  572. }
  573. }
  574. if len(tc.ExpectedActions) > len(actions) {
  575. t.Errorf("%s: %d additional expected actions", k, len(tc.ExpectedActions)-len(actions))
  576. for _, a := range tc.ExpectedActions[len(actions):] {
  577. t.Logf(" %+v", a)
  578. }
  579. }
  580. }
  581. }