func init() { … } type controllerTest … const annSkipLocalStore … type testCall … const testNamespace … const mockPluginName … var novolumes … var noclaims … var noevents … var noerrors … type volumeReactor … func newVolumeReactor(ctx context.Context, client *fake.Clientset, ctrl *PersistentVolumeController, fakeVolumeWatch, fakeClaimWatch *watch.FakeWatcher, errors []pvtesting.ReactorError) *volumeReactor { … } // waitForIdle waits until all tests, controllers and other goroutines do their // job and no new actions are registered for 10 milliseconds. func (r *volumeReactor) waitForIdle() { … } // waitTest waits until all tests, controllers and other goroutines do their // job and list of current volumes/claims is equal to list of expected // volumes/claims (with ~10 second timeout). func (r *volumeReactor) waitTest(test controllerTest) error { … } // checkEvents compares all expectedEvents with events generated during the test // and reports differences. func checkEvents(t *testing.T, ctx context.Context, expectedEvents []string, ctrl *PersistentVolumeController) error { … } func alwaysReady() bool { … } func newTestController(ctx context.Context, kubeClient clientset.Interface, informerFactory informers.SharedInformerFactory, enableDynamicProvisioning bool) (*PersistentVolumeController, error) { … } // newVolume returns a new volume with given attributes func newVolume(name, capacity, boundToClaimUID, boundToClaimName string, phase v1.PersistentVolumePhase, reclaimPolicy v1.PersistentVolumeReclaimPolicy, class string, annotations ...string) *v1.PersistentVolume { … } // newExternalProvisionedVolume returns a new volume with given attributes func newExternalProvisionedVolume(name, capacity, boundToClaimUID, boundToClaimName string, phase v1.PersistentVolumePhase, reclaimPolicy v1.PersistentVolumeReclaimPolicy, class string, driverName string, finalizers []string, annotations ...string) *v1.PersistentVolume { … } // newVolume returns a new volume with given attributes func newVolumeWithFinalizers(name, capacity, boundToClaimUID, boundToClaimName string, phase v1.PersistentVolumePhase, reclaimPolicy v1.PersistentVolumeReclaimPolicy, class string, finalizers []string, annotations ...string) *v1.PersistentVolume { … } // withLabels applies the given labels to the first volume in the array and // returns the array. Meant to be used to compose volumes specified inline in // a test. func withLabels(labels map[string]string, volumes []*v1.PersistentVolume) []*v1.PersistentVolume { … } // withLabelSelector sets the label selector of the first claim in the array // to be MatchLabels of the given label set and returns the array. Meant // to be used to compose claims specified inline in a test. func withLabelSelector(labels map[string]string, claims []*v1.PersistentVolumeClaim) []*v1.PersistentVolumeClaim { … } // withVolumeVolumeMode applies the given VolumeMode to the first volume in the array and // returns the array. Meant to be used to compose volumes specified inline in // a test. func withVolumeVolumeMode(mode *v1.PersistentVolumeMode, volumes []*v1.PersistentVolume) []*v1.PersistentVolume { … } // withClaimVolumeMode applies the given VolumeMode to the first claim in the array and // returns the array. Meant to be used to compose volumes specified inline in // a test. func withClaimVolumeMode(mode *v1.PersistentVolumeMode, claims []*v1.PersistentVolumeClaim) []*v1.PersistentVolumeClaim { … } // withExpectedCapacity sets the claim.Spec.Capacity of the first claim in the // array to given value and returns the array. Meant to be used to compose // claims specified inline in a test. func withExpectedCapacity(capacity string, claims []*v1.PersistentVolumeClaim) []*v1.PersistentVolumeClaim { … } // withExpectedVAC sets the claim.Status.CurrentVolumeAttributesClassName of the first claim in the // array to given value and returns the array. Meant to be used to compose // claims specified inline in a test. func withExpectedVAC(vacName *string, claims []*v1.PersistentVolumeClaim) []*v1.PersistentVolumeClaim { … } // withMessage saves given message into volume.Status.Message of the first // volume in the array and returns the array. Meant to be used to compose // volumes specified inline in a test. func withMessage(message string, volumes []*v1.PersistentVolume) []*v1.PersistentVolume { … } // newVolumeArray returns array with a single volume that would be returned by // newVolume() with the same parameters. // TODO: make the newVolumeArray function accept volume attributes class name as an input parameter func newVolumeArray(name, capacity, boundToClaimUID, boundToClaimName string, phase v1.PersistentVolumePhase, reclaimPolicy v1.PersistentVolumeReclaimPolicy, class string, annotations ...string) []*v1.PersistentVolume { … } func withVolumeDeletionTimestamp(pvs []*v1.PersistentVolume) []*v1.PersistentVolume { … } func volumesWithFinalizers(pvs []*v1.PersistentVolume, finalizers []string) []*v1.PersistentVolume { … } // newClaim returns a new claim with given attributes func newClaim(name, claimUID, capacity, boundToVolume string, phase v1.PersistentVolumeClaimPhase, class *string, annotations ...string) *v1.PersistentVolumeClaim { … } // newClaimArray returns array with a single claim that would be returned by // newClaim() with the same parameters. // TODO: make the newClaimArray function accept volume attributes class name as an input parameter func newClaimArray(name, claimUID, capacity, boundToVolume string, phase v1.PersistentVolumeClaimPhase, class *string, annotations ...string) []*v1.PersistentVolumeClaim { … } func claimWithVAC(vacName *string, claims []*v1.PersistentVolumeClaim) []*v1.PersistentVolumeClaim { … } // claimWithAnnotation saves given annotation into given claims. Meant to be // used to compose claims specified inline in a test. // TODO(refactor): This helper function (and other helpers related to claim // arrays) could use some cleaning up (most assume an array size of one)- // replace with annotateClaim at all callsites. The tests require claimArrays // but mostly operate on single claims func claimWithAnnotation(name, value string, claims []*v1.PersistentVolumeClaim) []*v1.PersistentVolumeClaim { … } func claimWithDataSource(name, kind, apiGroup string, claims []*v1.PersistentVolumeClaim) []*v1.PersistentVolumeClaim { … } func annotateClaim(claim *v1.PersistentVolumeClaim, ann map[string]string) *v1.PersistentVolumeClaim { … } // volumeWithVAC saves given vac into given volume. // Meant to be used to compose volume specified inline in a test. func volumeWithVAC(vacName string, volume *v1.PersistentVolume) *v1.PersistentVolume { … } // volumesWithVAC saves given vac into given volumes. // Meant to be used to compose volumes specified inline in a test. func volumesWithVAC(vacName string, volumes []*v1.PersistentVolume) []*v1.PersistentVolume { … } // volumeWithAnnotation saves given annotation into given volume. // Meant to be used to compose volume specified inline in a test. func volumeWithAnnotation(name, value string, volume *v1.PersistentVolume) *v1.PersistentVolume { … } // volumesWithAnnotation saves given annotation into given volumes. // Meant to be used to compose volumes specified inline in a test. func volumesWithAnnotation(name, value string, volumes []*v1.PersistentVolume) []*v1.PersistentVolume { … } // claimWithAccessMode saves given access into given claims. // Meant to be used to compose claims specified inline in a test. func claimWithAccessMode(modes []v1.PersistentVolumeAccessMode, claims []*v1.PersistentVolumeClaim) []*v1.PersistentVolumeClaim { … } func testSyncClaim(ctrl *PersistentVolumeController, reactor *pvtesting.VolumeReactor, test controllerTest) error { … } func testSyncClaimError(ctrl *PersistentVolumeController, reactor *pvtesting.VolumeReactor, test controllerTest) error { … } func testSyncVolume(ctrl *PersistentVolumeController, reactor *pvtesting.VolumeReactor, test controllerTest) error { … } type operationType … const operationDelete … const operationRecycle … var classGold … var classSilver … var classCopper … var classEmpty … var classNonExisting … var classExternal … var classExternalWait … var classUnknownInternal … var classUnsupportedMountOptions … var classLarge … var classWait … var classCSI … var modeWait … // wrapTestWithPluginCalls returns a testCall that: // - configures controller with a volume plugin that implements recycler, // deleter and provisioner. The plugin returns provided errors when a volume // is deleted, recycled or provisioned. // - calls given testCall func wrapTestWithPluginCalls(expectedRecycleCalls, expectedDeleteCalls []error, expectedProvisionCalls []provisionCall, toWrap testCall) testCall { … } // wrapTestWithReclaimCalls returns a testCall that: // - configures controller with recycler or deleter which will return provided // errors when a volume is deleted or recycled // - calls given testCall func wrapTestWithReclaimCalls(operation operationType, expectedOperationCalls []error, toWrap testCall) testCall { … } // wrapTestWithProvisionCalls returns a testCall that: // - configures controller with a provisioner which will return provided errors // when a claim is provisioned // - calls given testCall func wrapTestWithProvisionCalls(expectedProvisionCalls []provisionCall, toWrap testCall) testCall { … } type fakeCSINameTranslator … func (t fakeCSINameTranslator) GetCSINameFromInTreeName(pluginName string) (string, error) { … } type fakeCSIMigratedPluginManager … func (t fakeCSIMigratedPluginManager) IsMigrationEnabledForPlugin(pluginName string) bool { … } // wrapTestWithCSIMigrationProvisionCalls returns a testCall that: // - configures controller with a volume plugin that emulates CSI migration // - calls given testCall func wrapTestWithCSIMigrationProvisionCalls(toWrap testCall) testCall { … } // wrapTestWithInjectedOperation returns a testCall that: // - starts the controller and lets it run original testCall until // scheduleOperation() call. It blocks the controller there and calls the // injected function to simulate that something is happening when the // controller waits for the operation lock. Controller is then resumed and we // check how it behaves. func wrapTestWithInjectedOperation(ctx context.Context, toWrap testCall, injectBeforeOperation func(ctrl *PersistentVolumeController, reactor *pvtesting.VolumeReactor)) testCall { … } func evaluateTestResults(ctx context.Context, ctrl *PersistentVolumeController, reactor *pvtesting.VolumeReactor, test controllerTest, t *testing.T) { … } // Test single call to syncClaim and syncVolume methods. // For all tests: // 1. Fill in the controller with initial data // 2. Call the tested function (syncClaim/syncVolume) via // controllerTest.testCall *once*. // 3. Compare resulting volumes and claims with expected volumes and claims. func runSyncTests(t *testing.T, ctx context.Context, tests []controllerTest, storageClasses []*storage.StorageClass, pods []*v1.Pod) { … } // Test multiple calls to syncClaim/syncVolume and periodic sync of all // volume/claims. For all tests, the test follows this pattern: // 0. Load the controller with initial data. // 1. Call controllerTest.testCall() once as in TestSync() // 2. For all volumes/claims changed by previous syncVolume/syncClaim calls, // call appropriate syncVolume/syncClaim (simulating "volume/claim changed" // events). Go to 2. if these calls change anything. // 3. When all changes are processed and no new changes were made, call // syncVolume/syncClaim on all volumes/claims (simulating "periodic sync"). // 4. If some changes were done by step 3., go to 2. (simulation of // "volume/claim updated" events, eventually performing step 3. again) // 5. When 3. does not do any changes, finish the tests and compare final set // of volumes/claims with expected claims/volumes and report differences. // // Some limit of calls in enforced to prevent endless loops. func runMultisyncTests(t *testing.T, ctx context.Context, tests []controllerTest, storageClasses []*storage.StorageClass, defaultStorageClass string) { … } type mockVolumePlugin … type provisionCall … var _ … var _ … var _ … var _ … func (plugin *mockVolumePlugin) Init(host volume.VolumeHost) error { … } func (plugin *mockVolumePlugin) GetPluginName() string { … } func (plugin *mockVolumePlugin) GetVolumeName(spec *volume.Spec) (string, error) { … } func (plugin *mockVolumePlugin) CanSupport(spec *volume.Spec) bool { … } func (plugin *mockVolumePlugin) RequiresRemount(spec *volume.Spec) bool { … } func (plugin *mockVolumePlugin) SupportsMountOption() bool { … } func (plugin *mockVolumePlugin) SupportsBulkVolumeVerification() bool { … } func (plugin *mockVolumePlugin) ConstructVolumeSpec(volumeName, mountPath string) (volume.ReconstructedVolume, error) { … } func (plugin *mockVolumePlugin) SupportsSELinuxContextMount(spec *volume.Spec) (bool, error) { … } func (plugin *mockVolumePlugin) NewMounter(spec *volume.Spec, podRef *v1.Pod) (volume.Mounter, error) { … } func (plugin *mockVolumePlugin) NewUnmounter(name string, podUID types.UID) (volume.Unmounter, error) { … } func (plugin *mockVolumePlugin) NewProvisioner(logger klog.Logger, options volume.VolumeOptions) (volume.Provisioner, error) { … } func (plugin *mockVolumePlugin) Provision(selectedNode *v1.Node, allowedTopologies []v1.TopologySelectorTerm) (*v1.PersistentVolume, error) { … } func (plugin *mockVolumePlugin) NewDeleter(logger klog.Logger, spec *volume.Spec) (volume.Deleter, error) { … } func (plugin *mockVolumePlugin) Delete() error { … } func (plugin *mockVolumePlugin) GetPath() string { … } func (plugin *mockVolumePlugin) GetMetrics() (*volume.Metrics, error) { … } func (plugin *mockVolumePlugin) Recycle(pvName string, spec *volume.Spec, eventRecorder recyclerclient.RecycleEventRecorder) error { … }
Codebase Browser
kubernetes