1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796379737983799380038013802380338043805380638073808380938103811381238133814381538163817381838193820382138223823382438253826382738283829383038313832383338343835383638373838383938403841384238433844384538463847384838493850385138523853385438553856385738583859386038613862386338643865386638673868386938703871387238733874387538763877387838793880388138823883388438853886388738883889389038913892389338943895389638973898389939003901390239033904390539063907390839093910391139123913391439153916391739183919392039213922392339243925392639273928392939303931393239333934393539363937393839393940394139423943394439453946394739483949395039513952395339543955395639573958395939603961396239633964396539663967396839693970397139723973397439753976397739783979398039813982398339843985398639873988398939903991399239933994399539963997399839994000400140024003400440054006400740084009401040114012401340144015401640174018401940204021402240234024402540264027402840294030403140324033403440354036403740384039404040414042404340444045404640474048 |
- /*
- Copyright 2015 The Kubernetes Authors.
- Licensed under the Apache License, Version 2.0 (the "License");
- you may not use this file except in compliance with the License.
- You may obtain a copy of the License at
- http://www.apache.org/licenses/LICENSE-2.0
- Unless required by applicable law or agreed to in writing, software
- distributed under the License is distributed on an "AS IS" BASIS,
- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- See the License for the specific language governing permissions and
- limitations under the License.
- */
- package podautoscaler
- import (
- "encoding/json"
- "fmt"
- "math"
- "sync"
- "testing"
- "time"
- autoscalingv1 "k8s.io/api/autoscaling/v1"
- autoscalingv2 "k8s.io/api/autoscaling/v2beta2"
- v1 "k8s.io/api/core/v1"
- "k8s.io/apimachinery/pkg/api/meta/testrestmapper"
- "k8s.io/apimachinery/pkg/api/resource"
- metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
- "k8s.io/apimachinery/pkg/labels"
- "k8s.io/apimachinery/pkg/runtime"
- "k8s.io/apimachinery/pkg/runtime/schema"
- "k8s.io/apimachinery/pkg/watch"
- "k8s.io/client-go/informers"
- "k8s.io/client-go/kubernetes/fake"
- scalefake "k8s.io/client-go/scale/fake"
- core "k8s.io/client-go/testing"
- "k8s.io/kubernetes/pkg/api/legacyscheme"
- "k8s.io/kubernetes/pkg/apis/autoscaling"
- autoscalingapiv2beta2 "k8s.io/kubernetes/pkg/apis/autoscaling/v2beta2"
- "k8s.io/kubernetes/pkg/controller"
- "k8s.io/kubernetes/pkg/controller/podautoscaler/metrics"
- cmapi "k8s.io/metrics/pkg/apis/custom_metrics/v1beta2"
- emapi "k8s.io/metrics/pkg/apis/external_metrics/v1beta1"
- metricsapi "k8s.io/metrics/pkg/apis/metrics/v1beta1"
- metricsfake "k8s.io/metrics/pkg/client/clientset/versioned/fake"
- cmfake "k8s.io/metrics/pkg/client/custom_metrics/fake"
- emfake "k8s.io/metrics/pkg/client/external_metrics/fake"
- utilpointer "k8s.io/utils/pointer"
- "github.com/stretchr/testify/assert"
- _ "k8s.io/kubernetes/pkg/apis/apps/install"
- _ "k8s.io/kubernetes/pkg/apis/autoscaling/install"
- )
- // From now on, the HPA controller does have history in it (scaleUpEvents, scaleDownEvents)
- // Hence the second HPA controller reconcile cycle might return different result (comparing with the first run).
- // Current test infrastructure has a race condition, when several reconcile cycles will be performed
- // while it should be stopped right after the first one. And the second will raise an exception
- // because of different result.
- // This comment has more info: https://github.com/kubernetes/kubernetes/pull/74525#issuecomment-502653106
- // We need to rework this infrastructure: https://github.com/kubernetes/kubernetes/issues/79222
- var statusOk = []autoscalingv2.HorizontalPodAutoscalerCondition{
- {Type: autoscalingv2.AbleToScale, Status: v1.ConditionTrue, Reason: "SucceededRescale"},
- {Type: autoscalingv2.ScalingActive, Status: v1.ConditionTrue, Reason: "ValidMetricFound"},
- {Type: autoscalingv2.ScalingLimited, Status: v1.ConditionFalse, Reason: "DesiredWithinRange"},
- }
- // statusOkWithOverrides returns the "ok" status with the given conditions as overridden
- func statusOkWithOverrides(overrides ...autoscalingv2.HorizontalPodAutoscalerCondition) []autoscalingv1.HorizontalPodAutoscalerCondition {
- resv2 := make([]autoscalingv2.HorizontalPodAutoscalerCondition, len(statusOk))
- copy(resv2, statusOk)
- for _, override := range overrides {
- resv2 = setConditionInList(resv2, override.Type, override.Status, override.Reason, override.Message)
- }
- // copy to a v1 slice
- resv1 := make([]autoscalingv1.HorizontalPodAutoscalerCondition, len(resv2))
- for i, cond := range resv2 {
- resv1[i] = autoscalingv1.HorizontalPodAutoscalerCondition{
- Type: autoscalingv1.HorizontalPodAutoscalerConditionType(cond.Type),
- Status: cond.Status,
- Reason: cond.Reason,
- }
- }
- return resv1
- }
- func alwaysReady() bool { return true }
- type fakeResource struct {
- name string
- apiVersion string
- kind string
- }
- type testCase struct {
- sync.Mutex
- minReplicas int32
- maxReplicas int32
- specReplicas int32
- statusReplicas int32
- initialReplicas int32
- scaleUpRules *autoscalingv2.HPAScalingRules
- scaleDownRules *autoscalingv2.HPAScalingRules
- // CPU target utilization as a percentage of the requested resources.
- CPUTarget int32
- CPUCurrent int32
- verifyCPUCurrent bool
- reportedLevels []uint64
- reportedCPURequests []resource.Quantity
- reportedPodReadiness []v1.ConditionStatus
- reportedPodStartTime []metav1.Time
- reportedPodPhase []v1.PodPhase
- reportedPodDeletionTimestamp []bool
- scaleUpdated bool
- statusUpdated bool
- eventCreated bool
- verifyEvents bool
- useMetricsAPI bool
- metricsTarget []autoscalingv2.MetricSpec
- expectedDesiredReplicas int32
- expectedConditions []autoscalingv1.HorizontalPodAutoscalerCondition
- // Channel with names of HPA objects which we have reconciled.
- processed chan string
- // Target resource information.
- resource *fakeResource
- // Last scale time
- lastScaleTime *metav1.Time
- // override the test clients
- testClient *fake.Clientset
- testMetricsClient *metricsfake.Clientset
- testCMClient *cmfake.FakeCustomMetricsClient
- testEMClient *emfake.FakeExternalMetricsClient
- testScaleClient *scalefake.FakeScaleClient
- recommendations []timestampedRecommendation
- }
- // Needs to be called under a lock.
- func (tc *testCase) computeCPUCurrent() {
- if len(tc.reportedLevels) != len(tc.reportedCPURequests) || len(tc.reportedLevels) == 0 {
- return
- }
- reported := 0
- for _, r := range tc.reportedLevels {
- reported += int(r)
- }
- requested := 0
- for _, req := range tc.reportedCPURequests {
- requested += int(req.MilliValue())
- }
- tc.CPUCurrent = int32(100 * reported / requested)
- }
- func init() {
- // set this high so we don't accidentally run into it when testing
- scaleUpLimitFactor = 8
- }
- func (tc *testCase) prepareTestClient(t *testing.T) (*fake.Clientset, *metricsfake.Clientset, *cmfake.FakeCustomMetricsClient, *emfake.FakeExternalMetricsClient, *scalefake.FakeScaleClient) {
- namespace := "test-namespace"
- hpaName := "test-hpa"
- podNamePrefix := "test-pod"
- labelSet := map[string]string{"name": podNamePrefix}
- selector := labels.SelectorFromSet(labelSet).String()
- tc.Lock()
- tc.scaleUpdated = false
- tc.statusUpdated = false
- tc.eventCreated = false
- tc.processed = make(chan string, 100)
- if tc.CPUCurrent == 0 {
- tc.computeCPUCurrent()
- }
- if tc.resource == nil {
- tc.resource = &fakeResource{
- name: "test-rc",
- apiVersion: "v1",
- kind: "ReplicationController",
- }
- }
- tc.Unlock()
- fakeClient := &fake.Clientset{}
- fakeClient.AddReactor("list", "horizontalpodautoscalers", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- tc.Lock()
- defer tc.Unlock()
- var behavior *autoscalingv2.HorizontalPodAutoscalerBehavior
- if tc.scaleUpRules != nil || tc.scaleDownRules != nil {
- behavior = &autoscalingv2.HorizontalPodAutoscalerBehavior{
- ScaleUp: tc.scaleUpRules,
- ScaleDown: tc.scaleDownRules,
- }
- }
- hpa := autoscalingv2.HorizontalPodAutoscaler{
- ObjectMeta: metav1.ObjectMeta{
- Name: hpaName,
- Namespace: namespace,
- SelfLink: "experimental/v1/namespaces/" + namespace + "/horizontalpodautoscalers/" + hpaName,
- },
- Spec: autoscalingv2.HorizontalPodAutoscalerSpec{
- ScaleTargetRef: autoscalingv2.CrossVersionObjectReference{
- Kind: tc.resource.kind,
- Name: tc.resource.name,
- APIVersion: tc.resource.apiVersion,
- },
- MinReplicas: &tc.minReplicas,
- MaxReplicas: tc.maxReplicas,
- Behavior: behavior,
- },
- Status: autoscalingv2.HorizontalPodAutoscalerStatus{
- CurrentReplicas: tc.specReplicas,
- DesiredReplicas: tc.specReplicas,
- LastScaleTime: tc.lastScaleTime,
- },
- }
- // Initialize default values
- autoscalingapiv2beta2.SetDefaults_HorizontalPodAutoscalerBehavior(&hpa)
- obj := &autoscalingv2.HorizontalPodAutoscalerList{
- Items: []autoscalingv2.HorizontalPodAutoscaler{hpa},
- }
- if tc.CPUTarget > 0 {
- obj.Items[0].Spec.Metrics = []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.ResourceMetricSourceType,
- Resource: &autoscalingv2.ResourceMetricSource{
- Name: v1.ResourceCPU,
- Target: autoscalingv2.MetricTarget{
- AverageUtilization: &tc.CPUTarget,
- },
- },
- },
- }
- }
- if len(tc.metricsTarget) > 0 {
- obj.Items[0].Spec.Metrics = append(obj.Items[0].Spec.Metrics, tc.metricsTarget...)
- }
- if len(obj.Items[0].Spec.Metrics) == 0 {
- // manually add in the defaulting logic
- obj.Items[0].Spec.Metrics = []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.ResourceMetricSourceType,
- Resource: &autoscalingv2.ResourceMetricSource{
- Name: v1.ResourceCPU,
- },
- },
- }
- }
- // and... convert to autoscaling v1 to return the right type
- objv1, err := unsafeConvertToVersionVia(obj, autoscalingv1.SchemeGroupVersion)
- if err != nil {
- return true, nil, err
- }
- return true, objv1, nil
- })
- fakeClient.AddReactor("list", "pods", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- tc.Lock()
- defer tc.Unlock()
- obj := &v1.PodList{}
- specifiedCPURequests := tc.reportedCPURequests != nil
- numPodsToCreate := int(tc.statusReplicas)
- if specifiedCPURequests {
- numPodsToCreate = len(tc.reportedCPURequests)
- }
- for i := 0; i < numPodsToCreate; i++ {
- podReadiness := v1.ConditionTrue
- if tc.reportedPodReadiness != nil {
- podReadiness = tc.reportedPodReadiness[i]
- }
- var podStartTime metav1.Time
- if tc.reportedPodStartTime != nil {
- podStartTime = tc.reportedPodStartTime[i]
- }
- podPhase := v1.PodRunning
- if tc.reportedPodPhase != nil {
- podPhase = tc.reportedPodPhase[i]
- }
- podDeletionTimestamp := false
- if tc.reportedPodDeletionTimestamp != nil {
- podDeletionTimestamp = tc.reportedPodDeletionTimestamp[i]
- }
- podName := fmt.Sprintf("%s-%d", podNamePrefix, i)
- reportedCPURequest := resource.MustParse("1.0")
- if specifiedCPURequests {
- reportedCPURequest = tc.reportedCPURequests[i]
- }
- pod := v1.Pod{
- Status: v1.PodStatus{
- Phase: podPhase,
- Conditions: []v1.PodCondition{
- {
- Type: v1.PodReady,
- Status: podReadiness,
- LastTransitionTime: podStartTime,
- },
- },
- StartTime: &podStartTime,
- },
- ObjectMeta: metav1.ObjectMeta{
- Name: podName,
- Namespace: namespace,
- Labels: map[string]string{
- "name": podNamePrefix,
- },
- },
- Spec: v1.PodSpec{
- Containers: []v1.Container{
- {
- Resources: v1.ResourceRequirements{
- Requests: v1.ResourceList{
- v1.ResourceCPU: reportedCPURequest,
- },
- },
- },
- },
- },
- }
- if podDeletionTimestamp {
- pod.DeletionTimestamp = &metav1.Time{Time: time.Now()}
- }
- obj.Items = append(obj.Items, pod)
- }
- return true, obj, nil
- })
- fakeClient.AddReactor("update", "horizontalpodautoscalers", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- handled, obj, err := func() (handled bool, ret *autoscalingv1.HorizontalPodAutoscaler, err error) {
- tc.Lock()
- defer tc.Unlock()
- obj := action.(core.UpdateAction).GetObject().(*autoscalingv1.HorizontalPodAutoscaler)
- assert.Equal(t, namespace, obj.Namespace, "the HPA namespace should be as expected")
- assert.Equal(t, hpaName, obj.Name, "the HPA name should be as expected")
- assert.Equal(t, tc.expectedDesiredReplicas, obj.Status.DesiredReplicas, "the desired replica count reported in the object status should be as expected")
- if tc.verifyCPUCurrent {
- if assert.NotNil(t, obj.Status.CurrentCPUUtilizationPercentage, "the reported CPU utilization percentage should be non-nil") {
- assert.Equal(t, tc.CPUCurrent, *obj.Status.CurrentCPUUtilizationPercentage, "the report CPU utilization percentage should be as expected")
- }
- }
- var actualConditions []autoscalingv1.HorizontalPodAutoscalerCondition
- if err := json.Unmarshal([]byte(obj.ObjectMeta.Annotations[autoscaling.HorizontalPodAutoscalerConditionsAnnotation]), &actualConditions); err != nil {
- return true, nil, err
- }
- // TODO: it's ok not to sort these because statusOk
- // contains all the conditions, so we'll never be appending.
- // Default to statusOk when missing any specific conditions
- if tc.expectedConditions == nil {
- tc.expectedConditions = statusOkWithOverrides()
- }
- // clear the message so that we can easily compare
- for i := range actualConditions {
- actualConditions[i].Message = ""
- actualConditions[i].LastTransitionTime = metav1.Time{}
- }
- assert.Equal(t, tc.expectedConditions, actualConditions, "the status conditions should have been as expected")
- tc.statusUpdated = true
- // Every time we reconcile HPA object we are updating status.
- return true, obj, nil
- }()
- if obj != nil {
- tc.processed <- obj.Name
- }
- return handled, obj, err
- })
- fakeScaleClient := &scalefake.FakeScaleClient{}
- fakeScaleClient.AddReactor("get", "replicationcontrollers", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- tc.Lock()
- defer tc.Unlock()
- obj := &autoscalingv1.Scale{
- ObjectMeta: metav1.ObjectMeta{
- Name: tc.resource.name,
- Namespace: namespace,
- },
- Spec: autoscalingv1.ScaleSpec{
- Replicas: tc.specReplicas,
- },
- Status: autoscalingv1.ScaleStatus{
- Replicas: tc.statusReplicas,
- Selector: selector,
- },
- }
- return true, obj, nil
- })
- fakeScaleClient.AddReactor("get", "deployments", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- tc.Lock()
- defer tc.Unlock()
- obj := &autoscalingv1.Scale{
- ObjectMeta: metav1.ObjectMeta{
- Name: tc.resource.name,
- Namespace: namespace,
- },
- Spec: autoscalingv1.ScaleSpec{
- Replicas: tc.specReplicas,
- },
- Status: autoscalingv1.ScaleStatus{
- Replicas: tc.statusReplicas,
- Selector: selector,
- },
- }
- return true, obj, nil
- })
- fakeScaleClient.AddReactor("get", "replicasets", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- tc.Lock()
- defer tc.Unlock()
- obj := &autoscalingv1.Scale{
- ObjectMeta: metav1.ObjectMeta{
- Name: tc.resource.name,
- Namespace: namespace,
- },
- Spec: autoscalingv1.ScaleSpec{
- Replicas: tc.specReplicas,
- },
- Status: autoscalingv1.ScaleStatus{
- Replicas: tc.statusReplicas,
- Selector: selector,
- },
- }
- return true, obj, nil
- })
- fakeScaleClient.AddReactor("update", "replicationcontrollers", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- tc.Lock()
- defer tc.Unlock()
- obj := action.(core.UpdateAction).GetObject().(*autoscalingv1.Scale)
- replicas := action.(core.UpdateAction).GetObject().(*autoscalingv1.Scale).Spec.Replicas
- assert.Equal(t, tc.expectedDesiredReplicas, replicas, "the replica count of the RC should be as expected")
- tc.scaleUpdated = true
- return true, obj, nil
- })
- fakeScaleClient.AddReactor("update", "deployments", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- tc.Lock()
- defer tc.Unlock()
- obj := action.(core.UpdateAction).GetObject().(*autoscalingv1.Scale)
- replicas := action.(core.UpdateAction).GetObject().(*autoscalingv1.Scale).Spec.Replicas
- assert.Equal(t, tc.expectedDesiredReplicas, replicas, "the replica count of the deployment should be as expected")
- tc.scaleUpdated = true
- return true, obj, nil
- })
- fakeScaleClient.AddReactor("update", "replicasets", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- tc.Lock()
- defer tc.Unlock()
- obj := action.(core.UpdateAction).GetObject().(*autoscalingv1.Scale)
- replicas := action.(core.UpdateAction).GetObject().(*autoscalingv1.Scale).Spec.Replicas
- assert.Equal(t, tc.expectedDesiredReplicas, replicas, "the replica count of the replicaset should be as expected")
- tc.scaleUpdated = true
- return true, obj, nil
- })
- fakeWatch := watch.NewFake()
- fakeClient.AddWatchReactor("*", core.DefaultWatchReactor(fakeWatch, nil))
- fakeMetricsClient := &metricsfake.Clientset{}
- fakeMetricsClient.AddReactor("list", "pods", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- tc.Lock()
- defer tc.Unlock()
- metrics := &metricsapi.PodMetricsList{}
- for i, cpu := range tc.reportedLevels {
- // NB: the list reactor actually does label selector filtering for us,
- // so we have to make sure our results match the label selector
- podMetric := metricsapi.PodMetrics{
- ObjectMeta: metav1.ObjectMeta{
- Name: fmt.Sprintf("%s-%d", podNamePrefix, i),
- Namespace: namespace,
- Labels: labelSet,
- },
- Timestamp: metav1.Time{Time: time.Now()},
- Window: metav1.Duration{Duration: time.Minute},
- Containers: []metricsapi.ContainerMetrics{
- {
- Name: "container",
- Usage: v1.ResourceList{
- v1.ResourceCPU: *resource.NewMilliQuantity(
- int64(cpu),
- resource.DecimalSI),
- v1.ResourceMemory: *resource.NewQuantity(
- int64(1024*1024),
- resource.BinarySI),
- },
- },
- },
- }
- metrics.Items = append(metrics.Items, podMetric)
- }
- return true, metrics, nil
- })
- fakeCMClient := &cmfake.FakeCustomMetricsClient{}
- fakeCMClient.AddReactor("get", "*", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- tc.Lock()
- defer tc.Unlock()
- getForAction, wasGetFor := action.(cmfake.GetForAction)
- if !wasGetFor {
- return true, nil, fmt.Errorf("expected a get-for action, got %v instead", action)
- }
- if getForAction.GetName() == "*" {
- metrics := &cmapi.MetricValueList{}
- // multiple objects
- assert.Equal(t, "pods", getForAction.GetResource().Resource, "the type of object that we requested multiple metrics for should have been pods")
- assert.Equal(t, "qps", getForAction.GetMetricName(), "the metric name requested should have been qps, as specified in the metric spec")
- for i, level := range tc.reportedLevels {
- podMetric := cmapi.MetricValue{
- DescribedObject: v1.ObjectReference{
- Kind: "Pod",
- Name: fmt.Sprintf("%s-%d", podNamePrefix, i),
- Namespace: namespace,
- },
- Timestamp: metav1.Time{Time: time.Now()},
- Metric: cmapi.MetricIdentifier{
- Name: "qps",
- },
- Value: *resource.NewMilliQuantity(int64(level), resource.DecimalSI),
- }
- metrics.Items = append(metrics.Items, podMetric)
- }
- return true, metrics, nil
- }
- name := getForAction.GetName()
- mapper := testrestmapper.TestOnlyStaticRESTMapper(legacyscheme.Scheme)
- metrics := &cmapi.MetricValueList{}
- var matchedTarget *autoscalingv2.MetricSpec
- for i, target := range tc.metricsTarget {
- if target.Type == autoscalingv2.ObjectMetricSourceType && name == target.Object.DescribedObject.Name {
- gk := schema.FromAPIVersionAndKind(target.Object.DescribedObject.APIVersion, target.Object.DescribedObject.Kind).GroupKind()
- mapping, err := mapper.RESTMapping(gk)
- if err != nil {
- t.Logf("unable to get mapping for %s: %v", gk.String(), err)
- continue
- }
- groupResource := mapping.Resource.GroupResource()
- if getForAction.GetResource().Resource == groupResource.String() {
- matchedTarget = &tc.metricsTarget[i]
- }
- }
- }
- assert.NotNil(t, matchedTarget, "this request should have matched one of the metric specs")
- assert.Equal(t, "qps", getForAction.GetMetricName(), "the metric name requested should have been qps, as specified in the metric spec")
- metrics.Items = []cmapi.MetricValue{
- {
- DescribedObject: v1.ObjectReference{
- Kind: matchedTarget.Object.DescribedObject.Kind,
- APIVersion: matchedTarget.Object.DescribedObject.APIVersion,
- Name: name,
- },
- Timestamp: metav1.Time{Time: time.Now()},
- Metric: cmapi.MetricIdentifier{
- Name: "qps",
- },
- Value: *resource.NewMilliQuantity(int64(tc.reportedLevels[0]), resource.DecimalSI),
- },
- }
- return true, metrics, nil
- })
- fakeEMClient := &emfake.FakeExternalMetricsClient{}
- fakeEMClient.AddReactor("list", "*", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- tc.Lock()
- defer tc.Unlock()
- listAction, wasList := action.(core.ListAction)
- if !wasList {
- return true, nil, fmt.Errorf("expected a list action, got %v instead", action)
- }
- metrics := &emapi.ExternalMetricValueList{}
- assert.Equal(t, "qps", listAction.GetResource().Resource, "the metric name requested should have been qps, as specified in the metric spec")
- for _, level := range tc.reportedLevels {
- metric := emapi.ExternalMetricValue{
- Timestamp: metav1.Time{Time: time.Now()},
- MetricName: "qps",
- Value: *resource.NewMilliQuantity(int64(level), resource.DecimalSI),
- }
- metrics.Items = append(metrics.Items, metric)
- }
- return true, metrics, nil
- })
- return fakeClient, fakeMetricsClient, fakeCMClient, fakeEMClient, fakeScaleClient
- }
- func (tc *testCase) verifyResults(t *testing.T) {
- tc.Lock()
- defer tc.Unlock()
- assert.Equal(t, tc.specReplicas != tc.expectedDesiredReplicas, tc.scaleUpdated, "the scale should only be updated if we expected a change in replicas")
- assert.True(t, tc.statusUpdated, "the status should have been updated")
- if tc.verifyEvents {
- assert.Equal(t, tc.specReplicas != tc.expectedDesiredReplicas, tc.eventCreated, "an event should have been created only if we expected a change in replicas")
- }
- }
- func (tc *testCase) setupController(t *testing.T) (*HorizontalController, informers.SharedInformerFactory) {
- testClient, testMetricsClient, testCMClient, testEMClient, testScaleClient := tc.prepareTestClient(t)
- if tc.testClient != nil {
- testClient = tc.testClient
- }
- if tc.testMetricsClient != nil {
- testMetricsClient = tc.testMetricsClient
- }
- if tc.testCMClient != nil {
- testCMClient = tc.testCMClient
- }
- if tc.testEMClient != nil {
- testEMClient = tc.testEMClient
- }
- if tc.testScaleClient != nil {
- testScaleClient = tc.testScaleClient
- }
- metricsClient := metrics.NewRESTMetricsClient(
- testMetricsClient.MetricsV1beta1(),
- testCMClient,
- testEMClient,
- )
- eventClient := &fake.Clientset{}
- eventClient.AddReactor("create", "events", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- tc.Lock()
- defer tc.Unlock()
- obj := action.(core.CreateAction).GetObject().(*v1.Event)
- if tc.verifyEvents {
- switch obj.Reason {
- case "SuccessfulRescale":
- assert.Equal(t, fmt.Sprintf("New size: %d; reason: cpu resource utilization (percentage of request) above target", tc.expectedDesiredReplicas), obj.Message)
- case "DesiredReplicasComputed":
- assert.Equal(t, fmt.Sprintf(
- "Computed the desired num of replicas: %d (avgCPUutil: %d, current replicas: %d)",
- tc.expectedDesiredReplicas,
- (int64(tc.reportedLevels[0])*100)/tc.reportedCPURequests[0].MilliValue(), tc.specReplicas), obj.Message)
- default:
- assert.False(t, true, fmt.Sprintf("Unexpected event: %s / %s", obj.Reason, obj.Message))
- }
- }
- tc.eventCreated = true
- return true, obj, nil
- })
- informerFactory := informers.NewSharedInformerFactory(testClient, controller.NoResyncPeriodFunc())
- defaultDownscalestabilizationWindow := 5 * time.Minute
- hpaController := NewHorizontalController(
- eventClient.CoreV1(),
- testScaleClient,
- testClient.AutoscalingV1(),
- testrestmapper.TestOnlyStaticRESTMapper(legacyscheme.Scheme),
- metricsClient,
- informerFactory.Autoscaling().V1().HorizontalPodAutoscalers(),
- informerFactory.Core().V1().Pods(),
- 100*time.Millisecond, // we need non-zero resync period to avoid race conditions
- defaultDownscalestabilizationWindow,
- defaultTestingTolerance,
- defaultTestingCpuInitializationPeriod,
- defaultTestingDelayOfInitialReadinessStatus,
- )
- hpaController.hpaListerSynced = alwaysReady
- if tc.recommendations != nil {
- hpaController.recommendations["test-namespace/test-hpa"] = tc.recommendations
- }
- return hpaController, informerFactory
- }
- func hotCpuCreationTime() metav1.Time {
- return metav1.Time{Time: time.Now()}
- }
- func coolCpuCreationTime() metav1.Time {
- return metav1.Time{Time: time.Now().Add(-3 * time.Minute)}
- }
- func (tc *testCase) runTestWithController(t *testing.T, hpaController *HorizontalController, informerFactory informers.SharedInformerFactory) {
- stop := make(chan struct{})
- defer close(stop)
- informerFactory.Start(stop)
- go hpaController.Run(stop)
- tc.Lock()
- shouldWait := tc.verifyEvents
- tc.Unlock()
- if shouldWait {
- // We need to wait for events to be broadcasted (sleep for longer than record.sleepDuration).
- timeoutTime := time.Now().Add(2 * time.Second)
- for now := time.Now(); timeoutTime.After(now); now = time.Now() {
- sleepUntil := timeoutTime.Sub(now)
- select {
- case <-tc.processed:
- // drain the chan of any sent events to keep it from filling before the timeout
- case <-time.After(sleepUntil):
- // timeout reached, ready to verifyResults
- }
- }
- } else {
- // Wait for HPA to be processed.
- <-tc.processed
- }
- tc.verifyResults(t)
- }
- func (tc *testCase) runTest(t *testing.T) {
- hpaController, informerFactory := tc.setupController(t)
- tc.runTestWithController(t, hpaController, informerFactory)
- }
- func TestScaleUp(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 5,
- CPUTarget: 30,
- verifyCPUCurrent: true,
- reportedLevels: []uint64{300, 500, 700},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- useMetricsAPI: true,
- }
- tc.runTest(t)
- }
- func TestScaleUpUnreadyLessScale(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 4,
- CPUTarget: 30,
- CPUCurrent: 60,
- verifyCPUCurrent: true,
- reportedLevels: []uint64{300, 500, 700},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- reportedPodReadiness: []v1.ConditionStatus{v1.ConditionFalse, v1.ConditionTrue, v1.ConditionTrue},
- useMetricsAPI: true,
- }
- tc.runTest(t)
- }
- func TestScaleUpHotCpuLessScale(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 4,
- CPUTarget: 30,
- CPUCurrent: 60,
- verifyCPUCurrent: true,
- reportedLevels: []uint64{300, 500, 700},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- reportedPodStartTime: []metav1.Time{hotCpuCreationTime(), coolCpuCreationTime(), coolCpuCreationTime()},
- useMetricsAPI: true,
- }
- tc.runTest(t)
- }
- func TestScaleUpUnreadyNoScale(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 3,
- CPUTarget: 30,
- CPUCurrent: 40,
- verifyCPUCurrent: true,
- reportedLevels: []uint64{400, 500, 700},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- reportedPodReadiness: []v1.ConditionStatus{v1.ConditionTrue, v1.ConditionFalse, v1.ConditionFalse},
- useMetricsAPI: true,
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "ReadyForNewScale",
- }),
- }
- tc.runTest(t)
- }
- func TestScaleUpHotCpuNoScale(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 3,
- CPUTarget: 30,
- CPUCurrent: 40,
- verifyCPUCurrent: true,
- reportedLevels: []uint64{400, 500, 700},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- reportedPodReadiness: []v1.ConditionStatus{v1.ConditionTrue, v1.ConditionFalse, v1.ConditionFalse},
- reportedPodStartTime: []metav1.Time{coolCpuCreationTime(), hotCpuCreationTime(), hotCpuCreationTime()},
- useMetricsAPI: true,
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "ReadyForNewScale",
- }),
- }
- tc.runTest(t)
- }
- func TestScaleUpIgnoresFailedPods(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 2,
- statusReplicas: 2,
- expectedDesiredReplicas: 4,
- CPUTarget: 30,
- CPUCurrent: 60,
- verifyCPUCurrent: true,
- reportedLevels: []uint64{500, 700},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- reportedPodReadiness: []v1.ConditionStatus{v1.ConditionTrue, v1.ConditionTrue, v1.ConditionFalse, v1.ConditionFalse},
- reportedPodPhase: []v1.PodPhase{v1.PodRunning, v1.PodRunning, v1.PodFailed, v1.PodFailed},
- useMetricsAPI: true,
- }
- tc.runTest(t)
- }
- func TestScaleUpIgnoresDeletionPods(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 2,
- statusReplicas: 2,
- expectedDesiredReplicas: 4,
- CPUTarget: 30,
- CPUCurrent: 60,
- verifyCPUCurrent: true,
- reportedLevels: []uint64{500, 700},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- reportedPodReadiness: []v1.ConditionStatus{v1.ConditionTrue, v1.ConditionTrue, v1.ConditionFalse, v1.ConditionFalse},
- reportedPodPhase: []v1.PodPhase{v1.PodRunning, v1.PodRunning, v1.PodRunning, v1.PodRunning},
- reportedPodDeletionTimestamp: []bool{false, false, true, true},
- useMetricsAPI: true,
- }
- tc.runTest(t)
- }
- func TestScaleUpDeployment(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 5,
- CPUTarget: 30,
- verifyCPUCurrent: true,
- reportedLevels: []uint64{300, 500, 700},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- useMetricsAPI: true,
- resource: &fakeResource{
- name: "test-dep",
- apiVersion: "apps/v1",
- kind: "Deployment",
- },
- }
- tc.runTest(t)
- }
- func TestScaleUpReplicaSet(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 5,
- CPUTarget: 30,
- verifyCPUCurrent: true,
- reportedLevels: []uint64{300, 500, 700},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- useMetricsAPI: true,
- resource: &fakeResource{
- name: "test-replicaset",
- apiVersion: "apps/v1",
- kind: "ReplicaSet",
- },
- }
- tc.runTest(t)
- }
- func TestScaleUpCM(t *testing.T) {
- averageValue := resource.MustParse("15.0")
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 4,
- CPUTarget: 0,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.PodsMetricSourceType,
- Pods: &autoscalingv2.PodsMetricSource{
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- },
- Target: autoscalingv2.MetricTarget{
- AverageValue: &averageValue,
- },
- },
- },
- },
- reportedLevels: []uint64{20000, 10000, 30000},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- }
- tc.runTest(t)
- }
- func TestScaleUpCMUnreadyAndHotCpuNoLessScale(t *testing.T) {
- averageValue := resource.MustParse("15.0")
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 6,
- CPUTarget: 0,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.PodsMetricSourceType,
- Pods: &autoscalingv2.PodsMetricSource{
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- },
- Target: autoscalingv2.MetricTarget{
- AverageValue: &averageValue,
- },
- },
- },
- },
- reportedLevels: []uint64{50000, 10000, 30000},
- reportedPodReadiness: []v1.ConditionStatus{v1.ConditionTrue, v1.ConditionTrue, v1.ConditionFalse},
- reportedPodStartTime: []metav1.Time{coolCpuCreationTime(), coolCpuCreationTime(), hotCpuCreationTime()},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- }
- tc.runTest(t)
- }
- func TestScaleUpCMUnreadyandCpuHot(t *testing.T) {
- averageValue := resource.MustParse("15.0")
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 6,
- CPUTarget: 0,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.PodsMetricSourceType,
- Pods: &autoscalingv2.PodsMetricSource{
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- },
- Target: autoscalingv2.MetricTarget{
- AverageValue: &averageValue,
- },
- },
- },
- },
- reportedLevels: []uint64{50000, 15000, 30000},
- reportedPodReadiness: []v1.ConditionStatus{v1.ConditionFalse, v1.ConditionTrue, v1.ConditionFalse},
- reportedPodStartTime: []metav1.Time{hotCpuCreationTime(), coolCpuCreationTime(), hotCpuCreationTime()},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "SucceededRescale",
- }, autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.ScalingLimited,
- Status: v1.ConditionTrue,
- Reason: "TooManyReplicas",
- }),
- }
- tc.runTest(t)
- }
- func TestScaleUpHotCpuNoScaleWouldScaleDown(t *testing.T) {
- averageValue := resource.MustParse("15.0")
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 6,
- CPUTarget: 0,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.PodsMetricSourceType,
- Pods: &autoscalingv2.PodsMetricSource{
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- },
- Target: autoscalingv2.MetricTarget{
- AverageValue: &averageValue,
- },
- },
- },
- },
- reportedLevels: []uint64{50000, 15000, 30000},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- reportedPodStartTime: []metav1.Time{hotCpuCreationTime(), coolCpuCreationTime(), hotCpuCreationTime()},
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "SucceededRescale",
- }, autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.ScalingLimited,
- Status: v1.ConditionTrue,
- Reason: "TooManyReplicas",
- }),
- }
- tc.runTest(t)
- }
- func TestScaleUpCMObject(t *testing.T) {
- targetValue := resource.MustParse("15.0")
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 4,
- CPUTarget: 0,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.ObjectMetricSourceType,
- Object: &autoscalingv2.ObjectMetricSource{
- DescribedObject: autoscalingv2.CrossVersionObjectReference{
- APIVersion: "apps/v1",
- Kind: "Deployment",
- Name: "some-deployment",
- },
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- },
- Target: autoscalingv2.MetricTarget{
- Value: &targetValue,
- },
- },
- },
- },
- reportedLevels: []uint64{20000},
- }
- tc.runTest(t)
- }
- func TestScaleUpFromZeroCMObject(t *testing.T) {
- targetValue := resource.MustParse("15.0")
- tc := testCase{
- minReplicas: 0,
- maxReplicas: 6,
- specReplicas: 0,
- statusReplicas: 0,
- expectedDesiredReplicas: 2,
- CPUTarget: 0,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.ObjectMetricSourceType,
- Object: &autoscalingv2.ObjectMetricSource{
- DescribedObject: autoscalingv2.CrossVersionObjectReference{
- APIVersion: "apps/v1",
- Kind: "Deployment",
- Name: "some-deployment",
- },
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- },
- Target: autoscalingv2.MetricTarget{
- Value: &targetValue,
- },
- },
- },
- },
- reportedLevels: []uint64{20000},
- }
- tc.runTest(t)
- }
- func TestScaleUpFromZeroIgnoresToleranceCMObject(t *testing.T) {
- targetValue := resource.MustParse("1.0")
- tc := testCase{
- minReplicas: 0,
- maxReplicas: 6,
- specReplicas: 0,
- statusReplicas: 0,
- expectedDesiredReplicas: 1,
- CPUTarget: 0,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.ObjectMetricSourceType,
- Object: &autoscalingv2.ObjectMetricSource{
- DescribedObject: autoscalingv2.CrossVersionObjectReference{
- APIVersion: "apps/v1",
- Kind: "Deployment",
- Name: "some-deployment",
- },
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- },
- Target: autoscalingv2.MetricTarget{
- Value: &targetValue,
- },
- },
- },
- },
- reportedLevels: []uint64{1000},
- }
- tc.runTest(t)
- }
- func TestScaleUpPerPodCMObject(t *testing.T) {
- targetAverageValue := resource.MustParse("10.0")
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 4,
- CPUTarget: 0,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.ObjectMetricSourceType,
- Object: &autoscalingv2.ObjectMetricSource{
- DescribedObject: autoscalingv2.CrossVersionObjectReference{
- APIVersion: "apps/v1",
- Kind: "Deployment",
- Name: "some-deployment",
- },
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- },
- Target: autoscalingv2.MetricTarget{
- AverageValue: &targetAverageValue,
- },
- },
- },
- },
- reportedLevels: []uint64{40000},
- }
- tc.runTest(t)
- }
- func TestScaleUpCMExternal(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 4,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.ExternalMetricSourceType,
- External: &autoscalingv2.ExternalMetricSource{
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- Selector: &metav1.LabelSelector{},
- },
- Target: autoscalingv2.MetricTarget{
- Value: resource.NewMilliQuantity(6666, resource.DecimalSI),
- },
- },
- },
- },
- reportedLevels: []uint64{8600},
- }
- tc.runTest(t)
- }
- func TestScaleUpPerPodCMExternal(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 4,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.ExternalMetricSourceType,
- External: &autoscalingv2.ExternalMetricSource{
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- Selector: &metav1.LabelSelector{},
- },
- Target: autoscalingv2.MetricTarget{
- AverageValue: resource.NewMilliQuantity(2222, resource.DecimalSI),
- },
- },
- },
- },
- reportedLevels: []uint64{8600},
- }
- tc.runTest(t)
- }
- func TestScaleDown(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 5,
- statusReplicas: 5,
- expectedDesiredReplicas: 3,
- CPUTarget: 50,
- verifyCPUCurrent: true,
- reportedLevels: []uint64{100, 300, 500, 250, 250},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- useMetricsAPI: true,
- recommendations: []timestampedRecommendation{},
- }
- tc.runTest(t)
- }
- func TestScaleDownWithScalingRules(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- scaleUpRules: generateScalingRules(0, 0, 100, 15, 30),
- specReplicas: 5,
- statusReplicas: 5,
- expectedDesiredReplicas: 3,
- CPUTarget: 50,
- verifyCPUCurrent: true,
- reportedLevels: []uint64{100, 300, 500, 250, 250},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- useMetricsAPI: true,
- recommendations: []timestampedRecommendation{},
- }
- tc.runTest(t)
- }
- func TestScaleUpOneMetricInvalid(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 4,
- CPUTarget: 30,
- verifyCPUCurrent: true,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: "CheddarCheese",
- },
- },
- reportedLevels: []uint64{300, 400, 500},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- }
- tc.runTest(t)
- }
- func TestScaleUpFromZeroOneMetricInvalid(t *testing.T) {
- tc := testCase{
- minReplicas: 0,
- maxReplicas: 6,
- specReplicas: 0,
- statusReplicas: 0,
- expectedDesiredReplicas: 4,
- CPUTarget: 30,
- verifyCPUCurrent: true,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: "CheddarCheese",
- },
- },
- reportedLevels: []uint64{300, 400, 500},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- recommendations: []timestampedRecommendation{},
- }
- tc.runTest(t)
- }
- func TestScaleUpBothMetricsEmpty(t *testing.T) { // Switch to missing
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 3,
- CPUTarget: 0,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: "CheddarCheese",
- },
- },
- reportedLevels: []uint64{},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- expectedConditions: []autoscalingv1.HorizontalPodAutoscalerCondition{
- {Type: autoscalingv1.AbleToScale, Status: v1.ConditionTrue, Reason: "SucceededGetScale"},
- {Type: autoscalingv1.ScalingActive, Status: v1.ConditionFalse, Reason: "InvalidMetricSourceType"},
- },
- }
- tc.runTest(t)
- }
- func TestScaleDownStabilizeInitialSize(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 5,
- statusReplicas: 5,
- expectedDesiredReplicas: 5,
- CPUTarget: 50,
- verifyCPUCurrent: true,
- reportedLevels: []uint64{100, 300, 500, 250, 250},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- useMetricsAPI: true,
- recommendations: nil,
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "ReadyForNewScale",
- }, autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "ScaleDownStabilized",
- }),
- }
- tc.runTest(t)
- }
- func TestScaleDownCM(t *testing.T) {
- averageValue := resource.MustParse("20.0")
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 5,
- statusReplicas: 5,
- expectedDesiredReplicas: 3,
- CPUTarget: 0,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.PodsMetricSourceType,
- Pods: &autoscalingv2.PodsMetricSource{
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- },
- Target: autoscalingv2.MetricTarget{
- AverageValue: &averageValue,
- },
- },
- },
- },
- reportedLevels: []uint64{12000, 12000, 12000, 12000, 12000},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- recommendations: []timestampedRecommendation{},
- }
- tc.runTest(t)
- }
- func TestScaleDownCMObject(t *testing.T) {
- targetValue := resource.MustParse("20.0")
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 5,
- statusReplicas: 5,
- expectedDesiredReplicas: 3,
- CPUTarget: 0,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.ObjectMetricSourceType,
- Object: &autoscalingv2.ObjectMetricSource{
- DescribedObject: autoscalingv2.CrossVersionObjectReference{
- APIVersion: "apps/v1",
- Kind: "Deployment",
- Name: "some-deployment",
- },
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- },
- Target: autoscalingv2.MetricTarget{
- Value: &targetValue,
- },
- },
- },
- },
- reportedLevels: []uint64{12000},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- recommendations: []timestampedRecommendation{},
- }
- tc.runTest(t)
- }
- func TestScaleDownToZeroCMObject(t *testing.T) {
- targetValue := resource.MustParse("20.0")
- tc := testCase{
- minReplicas: 0,
- maxReplicas: 6,
- specReplicas: 5,
- statusReplicas: 5,
- expectedDesiredReplicas: 0,
- CPUTarget: 0,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.ObjectMetricSourceType,
- Object: &autoscalingv2.ObjectMetricSource{
- DescribedObject: autoscalingv2.CrossVersionObjectReference{
- APIVersion: "apps/v1",
- Kind: "Deployment",
- Name: "some-deployment",
- },
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- },
- Target: autoscalingv2.MetricTarget{
- Value: &targetValue,
- },
- },
- },
- },
- reportedLevels: []uint64{0},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- recommendations: []timestampedRecommendation{},
- }
- tc.runTest(t)
- }
- func TestScaleDownPerPodCMObject(t *testing.T) {
- targetAverageValue := resource.MustParse("20.0")
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 5,
- statusReplicas: 5,
- expectedDesiredReplicas: 3,
- CPUTarget: 0,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.ObjectMetricSourceType,
- Object: &autoscalingv2.ObjectMetricSource{
- DescribedObject: autoscalingv2.CrossVersionObjectReference{
- APIVersion: "apps/v1",
- Kind: "Deployment",
- Name: "some-deployment",
- },
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- },
- Target: autoscalingv2.MetricTarget{
- AverageValue: &targetAverageValue,
- },
- },
- },
- },
- reportedLevels: []uint64{60000},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- recommendations: []timestampedRecommendation{},
- }
- tc.runTest(t)
- }
- func TestScaleDownCMExternal(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 5,
- statusReplicas: 5,
- expectedDesiredReplicas: 3,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.ExternalMetricSourceType,
- External: &autoscalingv2.ExternalMetricSource{
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- Selector: &metav1.LabelSelector{},
- },
- Target: autoscalingv2.MetricTarget{
- Value: resource.NewMilliQuantity(14400, resource.DecimalSI),
- },
- },
- },
- },
- reportedLevels: []uint64{8600},
- recommendations: []timestampedRecommendation{},
- }
- tc.runTest(t)
- }
- func TestScaleDownToZeroCMExternal(t *testing.T) {
- tc := testCase{
- minReplicas: 0,
- maxReplicas: 6,
- specReplicas: 5,
- statusReplicas: 5,
- expectedDesiredReplicas: 0,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.ExternalMetricSourceType,
- External: &autoscalingv2.ExternalMetricSource{
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- Selector: &metav1.LabelSelector{},
- },
- Target: autoscalingv2.MetricTarget{
- Value: resource.NewMilliQuantity(14400, resource.DecimalSI),
- },
- },
- },
- },
- reportedLevels: []uint64{0},
- recommendations: []timestampedRecommendation{},
- }
- tc.runTest(t)
- }
- func TestScaleDownPerPodCMExternal(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 5,
- statusReplicas: 5,
- expectedDesiredReplicas: 3,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.ExternalMetricSourceType,
- External: &autoscalingv2.ExternalMetricSource{
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- Selector: &metav1.LabelSelector{},
- },
- Target: autoscalingv2.MetricTarget{
- AverageValue: resource.NewMilliQuantity(3000, resource.DecimalSI),
- },
- },
- },
- },
- reportedLevels: []uint64{8600},
- recommendations: []timestampedRecommendation{},
- }
- tc.runTest(t)
- }
- func TestScaleDownIncludeUnreadyPods(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 5,
- statusReplicas: 5,
- expectedDesiredReplicas: 2,
- CPUTarget: 50,
- CPUCurrent: 30,
- verifyCPUCurrent: true,
- reportedLevels: []uint64{100, 300, 500, 250, 250},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- useMetricsAPI: true,
- reportedPodReadiness: []v1.ConditionStatus{v1.ConditionTrue, v1.ConditionTrue, v1.ConditionTrue, v1.ConditionFalse, v1.ConditionFalse},
- recommendations: []timestampedRecommendation{},
- }
- tc.runTest(t)
- }
- func TestScaleDownOneMetricInvalid(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 5,
- statusReplicas: 5,
- expectedDesiredReplicas: 3,
- CPUTarget: 50,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: "CheddarCheese",
- },
- },
- reportedLevels: []uint64{100, 300, 500, 250, 250},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- useMetricsAPI: true,
- recommendations: []timestampedRecommendation{},
- }
- tc.runTest(t)
- }
- func TestScaleDownOneMetricEmpty(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 5,
- statusReplicas: 5,
- expectedDesiredReplicas: 3,
- CPUTarget: 50,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.ExternalMetricSourceType,
- External: &autoscalingv2.ExternalMetricSource{
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- Selector: &metav1.LabelSelector{},
- },
- Target: autoscalingv2.MetricTarget{
- Type: autoscalingv2.AverageValueMetricType,
- AverageValue: resource.NewMilliQuantity(1000, resource.DecimalSI),
- },
- },
- },
- },
- reportedLevels: []uint64{100, 300, 500, 250, 250},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- useMetricsAPI: true,
- recommendations: []timestampedRecommendation{},
- }
- _, _, _, testEMClient, _ := tc.prepareTestClient(t)
- testEMClient.PrependReactor("list", "*", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- return true, &emapi.ExternalMetricValueList{}, fmt.Errorf("something went wrong")
- })
- tc.testEMClient = testEMClient
- tc.runTest(t)
- }
- func TestScaleDownIgnoreHotCpuPods(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 5,
- statusReplicas: 5,
- expectedDesiredReplicas: 2,
- CPUTarget: 50,
- CPUCurrent: 30,
- verifyCPUCurrent: true,
- reportedLevels: []uint64{100, 300, 500, 250, 250},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- useMetricsAPI: true,
- reportedPodStartTime: []metav1.Time{coolCpuCreationTime(), coolCpuCreationTime(), coolCpuCreationTime(), hotCpuCreationTime(), hotCpuCreationTime()},
- recommendations: []timestampedRecommendation{},
- }
- tc.runTest(t)
- }
- func TestScaleDownIgnoresFailedPods(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 5,
- statusReplicas: 5,
- expectedDesiredReplicas: 3,
- CPUTarget: 50,
- CPUCurrent: 28,
- verifyCPUCurrent: true,
- reportedLevels: []uint64{100, 300, 500, 250, 250},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- useMetricsAPI: true,
- reportedPodReadiness: []v1.ConditionStatus{v1.ConditionTrue, v1.ConditionTrue, v1.ConditionTrue, v1.ConditionTrue, v1.ConditionTrue, v1.ConditionFalse, v1.ConditionFalse},
- reportedPodPhase: []v1.PodPhase{v1.PodRunning, v1.PodRunning, v1.PodRunning, v1.PodRunning, v1.PodRunning, v1.PodFailed, v1.PodFailed},
- recommendations: []timestampedRecommendation{},
- }
- tc.runTest(t)
- }
- func TestScaleDownIgnoresDeletionPods(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 5,
- statusReplicas: 5,
- expectedDesiredReplicas: 3,
- CPUTarget: 50,
- CPUCurrent: 28,
- verifyCPUCurrent: true,
- reportedLevels: []uint64{100, 300, 500, 250, 250},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- useMetricsAPI: true,
- reportedPodReadiness: []v1.ConditionStatus{v1.ConditionTrue, v1.ConditionTrue, v1.ConditionTrue, v1.ConditionTrue, v1.ConditionTrue, v1.ConditionFalse, v1.ConditionFalse},
- reportedPodPhase: []v1.PodPhase{v1.PodRunning, v1.PodRunning, v1.PodRunning, v1.PodRunning, v1.PodRunning, v1.PodRunning, v1.PodRunning},
- reportedPodDeletionTimestamp: []bool{false, false, false, false, false, true, true},
- recommendations: []timestampedRecommendation{},
- }
- tc.runTest(t)
- }
- func TestTolerance(t *testing.T) {
- tc := testCase{
- minReplicas: 1,
- maxReplicas: 5,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 3,
- CPUTarget: 100,
- reportedLevels: []uint64{1010, 1030, 1020},
- reportedCPURequests: []resource.Quantity{resource.MustParse("0.9"), resource.MustParse("1.0"), resource.MustParse("1.1")},
- useMetricsAPI: true,
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "ReadyForNewScale",
- }),
- }
- tc.runTest(t)
- }
- func TestToleranceCM(t *testing.T) {
- averageValue := resource.MustParse("20.0")
- tc := testCase{
- minReplicas: 1,
- maxReplicas: 5,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 3,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.PodsMetricSourceType,
- Pods: &autoscalingv2.PodsMetricSource{
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- },
- Target: autoscalingv2.MetricTarget{
- AverageValue: &averageValue,
- },
- },
- },
- },
- reportedLevels: []uint64{20000, 20001, 21000},
- reportedCPURequests: []resource.Quantity{resource.MustParse("0.9"), resource.MustParse("1.0"), resource.MustParse("1.1")},
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "ReadyForNewScale",
- }),
- }
- tc.runTest(t)
- }
- func TestToleranceCMObject(t *testing.T) {
- targetValue := resource.MustParse("20.0")
- tc := testCase{
- minReplicas: 1,
- maxReplicas: 5,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 3,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.ObjectMetricSourceType,
- Object: &autoscalingv2.ObjectMetricSource{
- DescribedObject: autoscalingv2.CrossVersionObjectReference{
- APIVersion: "apps/v1",
- Kind: "Deployment",
- Name: "some-deployment",
- },
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- },
- Target: autoscalingv2.MetricTarget{
- Value: &targetValue,
- },
- },
- },
- },
- reportedLevels: []uint64{20050},
- reportedCPURequests: []resource.Quantity{resource.MustParse("0.9"), resource.MustParse("1.0"), resource.MustParse("1.1")},
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "ReadyForNewScale",
- }),
- }
- tc.runTest(t)
- }
- func TestToleranceCMExternal(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 4,
- statusReplicas: 4,
- expectedDesiredReplicas: 4,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.ExternalMetricSourceType,
- External: &autoscalingv2.ExternalMetricSource{
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- Selector: &metav1.LabelSelector{},
- },
- Target: autoscalingv2.MetricTarget{
- Value: resource.NewMilliQuantity(8666, resource.DecimalSI),
- },
- },
- },
- },
- reportedLevels: []uint64{8600},
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "ReadyForNewScale",
- }),
- }
- tc.runTest(t)
- }
- func TestTolerancePerPodCMObject(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 4,
- statusReplicas: 4,
- expectedDesiredReplicas: 4,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.ObjectMetricSourceType,
- Object: &autoscalingv2.ObjectMetricSource{
- DescribedObject: autoscalingv2.CrossVersionObjectReference{
- APIVersion: "apps/v1",
- Kind: "Deployment",
- Name: "some-deployment",
- },
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- Selector: &metav1.LabelSelector{},
- },
- Target: autoscalingv2.MetricTarget{
- AverageValue: resource.NewMilliQuantity(2200, resource.DecimalSI),
- },
- },
- },
- },
- reportedLevels: []uint64{8600},
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "ReadyForNewScale",
- }),
- }
- tc.runTest(t)
- }
- func TestTolerancePerPodCMExternal(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 4,
- statusReplicas: 4,
- expectedDesiredReplicas: 4,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.ExternalMetricSourceType,
- External: &autoscalingv2.ExternalMetricSource{
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- Selector: &metav1.LabelSelector{},
- },
- Target: autoscalingv2.MetricTarget{
- AverageValue: resource.NewMilliQuantity(2200, resource.DecimalSI),
- },
- },
- },
- },
- reportedLevels: []uint64{8600},
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "ReadyForNewScale",
- }),
- }
- tc.runTest(t)
- }
- func TestMinReplicas(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 5,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 2,
- CPUTarget: 90,
- reportedLevels: []uint64{10, 95, 10},
- reportedCPURequests: []resource.Quantity{resource.MustParse("0.9"), resource.MustParse("1.0"), resource.MustParse("1.1")},
- useMetricsAPI: true,
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.ScalingLimited,
- Status: v1.ConditionTrue,
- Reason: "TooFewReplicas",
- }),
- recommendations: []timestampedRecommendation{},
- }
- tc.runTest(t)
- }
- func TestZeroMinReplicasDesiredZero(t *testing.T) {
- tc := testCase{
- minReplicas: 0,
- maxReplicas: 5,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 0,
- CPUTarget: 90,
- reportedLevels: []uint64{0, 0, 0},
- reportedCPURequests: []resource.Quantity{resource.MustParse("0.9"), resource.MustParse("1.0"), resource.MustParse("1.1")},
- useMetricsAPI: true,
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.ScalingLimited,
- Status: v1.ConditionFalse,
- Reason: "DesiredWithinRange",
- }),
- recommendations: []timestampedRecommendation{},
- }
- tc.runTest(t)
- }
- func TestMinReplicasDesiredZero(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 5,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 2,
- CPUTarget: 90,
- reportedLevels: []uint64{0, 0, 0},
- reportedCPURequests: []resource.Quantity{resource.MustParse("0.9"), resource.MustParse("1.0"), resource.MustParse("1.1")},
- useMetricsAPI: true,
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.ScalingLimited,
- Status: v1.ConditionTrue,
- Reason: "TooFewReplicas",
- }),
- recommendations: []timestampedRecommendation{},
- }
- tc.runTest(t)
- }
- func TestZeroReplicas(t *testing.T) {
- tc := testCase{
- minReplicas: 3,
- maxReplicas: 5,
- specReplicas: 0,
- statusReplicas: 0,
- expectedDesiredReplicas: 0,
- CPUTarget: 90,
- reportedLevels: []uint64{},
- reportedCPURequests: []resource.Quantity{},
- useMetricsAPI: true,
- expectedConditions: []autoscalingv1.HorizontalPodAutoscalerCondition{
- {Type: autoscalingv1.AbleToScale, Status: v1.ConditionTrue, Reason: "SucceededGetScale"},
- {Type: autoscalingv1.ScalingActive, Status: v1.ConditionFalse, Reason: "ScalingDisabled"},
- },
- }
- tc.runTest(t)
- }
- func TestTooFewReplicas(t *testing.T) {
- tc := testCase{
- minReplicas: 3,
- maxReplicas: 5,
- specReplicas: 2,
- statusReplicas: 2,
- expectedDesiredReplicas: 3,
- CPUTarget: 90,
- reportedLevels: []uint64{},
- reportedCPURequests: []resource.Quantity{},
- useMetricsAPI: true,
- expectedConditions: []autoscalingv1.HorizontalPodAutoscalerCondition{
- {Type: autoscalingv1.AbleToScale, Status: v1.ConditionTrue, Reason: "SucceededRescale"},
- },
- }
- tc.runTest(t)
- }
- func TestTooManyReplicas(t *testing.T) {
- tc := testCase{
- minReplicas: 3,
- maxReplicas: 5,
- specReplicas: 10,
- statusReplicas: 10,
- expectedDesiredReplicas: 5,
- CPUTarget: 90,
- reportedLevels: []uint64{},
- reportedCPURequests: []resource.Quantity{},
- useMetricsAPI: true,
- expectedConditions: []autoscalingv1.HorizontalPodAutoscalerCondition{
- {Type: autoscalingv1.AbleToScale, Status: v1.ConditionTrue, Reason: "SucceededRescale"},
- },
- }
- tc.runTest(t)
- }
- func TestMaxReplicas(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 5,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 5,
- CPUTarget: 90,
- reportedLevels: []uint64{8000, 9500, 1000},
- reportedCPURequests: []resource.Quantity{resource.MustParse("0.9"), resource.MustParse("1.0"), resource.MustParse("1.1")},
- useMetricsAPI: true,
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.ScalingLimited,
- Status: v1.ConditionTrue,
- Reason: "TooManyReplicas",
- }),
- }
- tc.runTest(t)
- }
- func TestSuperfluousMetrics(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 4,
- statusReplicas: 4,
- expectedDesiredReplicas: 6,
- CPUTarget: 100,
- reportedLevels: []uint64{4000, 9500, 3000, 7000, 3200, 2000},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- useMetricsAPI: true,
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.ScalingLimited,
- Status: v1.ConditionTrue,
- Reason: "TooManyReplicas",
- }),
- }
- tc.runTest(t)
- }
- func TestMissingMetrics(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 4,
- statusReplicas: 4,
- expectedDesiredReplicas: 3,
- CPUTarget: 100,
- reportedLevels: []uint64{400, 95},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- useMetricsAPI: true,
- recommendations: []timestampedRecommendation{},
- }
- tc.runTest(t)
- }
- func TestEmptyMetrics(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 4,
- statusReplicas: 4,
- expectedDesiredReplicas: 4,
- CPUTarget: 100,
- reportedLevels: []uint64{},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- useMetricsAPI: true,
- expectedConditions: []autoscalingv1.HorizontalPodAutoscalerCondition{
- {Type: autoscalingv1.AbleToScale, Status: v1.ConditionTrue, Reason: "SucceededGetScale"},
- {Type: autoscalingv1.ScalingActive, Status: v1.ConditionFalse, Reason: "FailedGetResourceMetric"},
- },
- }
- tc.runTest(t)
- }
- func TestEmptyCPURequest(t *testing.T) {
- tc := testCase{
- minReplicas: 1,
- maxReplicas: 5,
- specReplicas: 1,
- statusReplicas: 1,
- expectedDesiredReplicas: 1,
- CPUTarget: 100,
- reportedLevels: []uint64{200},
- reportedCPURequests: []resource.Quantity{},
- useMetricsAPI: true,
- expectedConditions: []autoscalingv1.HorizontalPodAutoscalerCondition{
- {Type: autoscalingv1.AbleToScale, Status: v1.ConditionTrue, Reason: "SucceededGetScale"},
- {Type: autoscalingv1.ScalingActive, Status: v1.ConditionFalse, Reason: "FailedGetResourceMetric"},
- },
- }
- tc.runTest(t)
- }
- func TestEventCreated(t *testing.T) {
- tc := testCase{
- minReplicas: 1,
- maxReplicas: 5,
- specReplicas: 1,
- statusReplicas: 1,
- expectedDesiredReplicas: 2,
- CPUTarget: 50,
- reportedLevels: []uint64{200},
- reportedCPURequests: []resource.Quantity{resource.MustParse("0.2")},
- verifyEvents: true,
- useMetricsAPI: true,
- }
- tc.runTest(t)
- }
- func TestEventNotCreated(t *testing.T) {
- tc := testCase{
- minReplicas: 1,
- maxReplicas: 5,
- specReplicas: 2,
- statusReplicas: 2,
- expectedDesiredReplicas: 2,
- CPUTarget: 50,
- reportedLevels: []uint64{200, 200},
- reportedCPURequests: []resource.Quantity{resource.MustParse("0.4"), resource.MustParse("0.4")},
- verifyEvents: true,
- useMetricsAPI: true,
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "ReadyForNewScale",
- }),
- }
- tc.runTest(t)
- }
- func TestMissingReports(t *testing.T) {
- tc := testCase{
- minReplicas: 1,
- maxReplicas: 5,
- specReplicas: 4,
- statusReplicas: 4,
- expectedDesiredReplicas: 2,
- CPUTarget: 50,
- reportedLevels: []uint64{200},
- reportedCPURequests: []resource.Quantity{resource.MustParse("0.2")},
- useMetricsAPI: true,
- recommendations: []timestampedRecommendation{},
- }
- tc.runTest(t)
- }
- func TestUpscaleCap(t *testing.T) {
- tc := testCase{
- minReplicas: 1,
- maxReplicas: 100,
- specReplicas: 3,
- statusReplicas: 3,
- scaleUpRules: generateScalingRules(0, 0, 700, 60, 0),
- initialReplicas: 3,
- expectedDesiredReplicas: 24,
- CPUTarget: 10,
- reportedLevels: []uint64{100, 200, 300},
- reportedCPURequests: []resource.Quantity{resource.MustParse("0.1"), resource.MustParse("0.1"), resource.MustParse("0.1")},
- useMetricsAPI: true,
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.ScalingLimited,
- Status: v1.ConditionTrue,
- Reason: "ScaleUpLimit",
- }),
- }
- tc.runTest(t)
- }
- func TestUpscaleCapGreaterThanMaxReplicas(t *testing.T) {
- tc := testCase{
- minReplicas: 1,
- maxReplicas: 20,
- specReplicas: 3,
- statusReplicas: 3,
- scaleUpRules: generateScalingRules(0, 0, 700, 60, 0),
- initialReplicas: 3,
- // expectedDesiredReplicas would be 24 without maxReplicas
- expectedDesiredReplicas: 20,
- CPUTarget: 10,
- reportedLevels: []uint64{100, 200, 300},
- reportedCPURequests: []resource.Quantity{resource.MustParse("0.1"), resource.MustParse("0.1"), resource.MustParse("0.1")},
- useMetricsAPI: true,
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.ScalingLimited,
- Status: v1.ConditionTrue,
- Reason: "TooManyReplicas",
- }),
- }
- tc.runTest(t)
- }
- func TestMoreReplicasThanSpecNoScale(t *testing.T) {
- tc := testCase{
- minReplicas: 1,
- maxReplicas: 8,
- specReplicas: 4,
- statusReplicas: 5, // Deployment update with 25% surge.
- expectedDesiredReplicas: 4,
- CPUTarget: 50,
- reportedLevels: []uint64{500, 500, 500, 500, 500},
- reportedCPURequests: []resource.Quantity{
- resource.MustParse("1"),
- resource.MustParse("1"),
- resource.MustParse("1"),
- resource.MustParse("1"),
- resource.MustParse("1"),
- },
- useMetricsAPI: true,
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "ReadyForNewScale",
- }),
- }
- tc.runTest(t)
- }
- func TestConditionInvalidSelectorMissing(t *testing.T) {
- tc := testCase{
- minReplicas: 1,
- maxReplicas: 100,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 3,
- CPUTarget: 10,
- reportedLevels: []uint64{100, 200, 300},
- reportedCPURequests: []resource.Quantity{resource.MustParse("0.1"), resource.MustParse("0.1"), resource.MustParse("0.1")},
- useMetricsAPI: true,
- expectedConditions: []autoscalingv1.HorizontalPodAutoscalerCondition{
- {
- Type: autoscalingv1.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "SucceededGetScale",
- },
- {
- Type: autoscalingv1.ScalingActive,
- Status: v1.ConditionFalse,
- Reason: "InvalidSelector",
- },
- },
- }
- _, _, _, _, testScaleClient := tc.prepareTestClient(t)
- tc.testScaleClient = testScaleClient
- testScaleClient.PrependReactor("get", "replicationcontrollers", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- obj := &autoscalingv1.Scale{
- ObjectMeta: metav1.ObjectMeta{
- Name: tc.resource.name,
- },
- Spec: autoscalingv1.ScaleSpec{
- Replicas: tc.specReplicas,
- },
- Status: autoscalingv1.ScaleStatus{
- Replicas: tc.specReplicas,
- },
- }
- return true, obj, nil
- })
- tc.runTest(t)
- }
- func TestConditionInvalidSelectorUnparsable(t *testing.T) {
- tc := testCase{
- minReplicas: 1,
- maxReplicas: 100,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 3,
- CPUTarget: 10,
- reportedLevels: []uint64{100, 200, 300},
- reportedCPURequests: []resource.Quantity{resource.MustParse("0.1"), resource.MustParse("0.1"), resource.MustParse("0.1")},
- useMetricsAPI: true,
- expectedConditions: []autoscalingv1.HorizontalPodAutoscalerCondition{
- {
- Type: autoscalingv1.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "SucceededGetScale",
- },
- {
- Type: autoscalingv1.ScalingActive,
- Status: v1.ConditionFalse,
- Reason: "InvalidSelector",
- },
- },
- }
- _, _, _, _, testScaleClient := tc.prepareTestClient(t)
- tc.testScaleClient = testScaleClient
- testScaleClient.PrependReactor("get", "replicationcontrollers", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- obj := &autoscalingv1.Scale{
- ObjectMeta: metav1.ObjectMeta{
- Name: tc.resource.name,
- },
- Spec: autoscalingv1.ScaleSpec{
- Replicas: tc.specReplicas,
- },
- Status: autoscalingv1.ScaleStatus{
- Replicas: tc.specReplicas,
- Selector: "cheddar cheese",
- },
- }
- return true, obj, nil
- })
- tc.runTest(t)
- }
- func TestConditionFailedGetMetrics(t *testing.T) {
- targetValue := resource.MustParse("15.0")
- averageValue := resource.MustParse("15.0")
- metricsTargets := map[string][]autoscalingv2.MetricSpec{
- "FailedGetResourceMetric": nil,
- "FailedGetPodsMetric": {
- {
- Type: autoscalingv2.PodsMetricSourceType,
- Pods: &autoscalingv2.PodsMetricSource{
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- },
- Target: autoscalingv2.MetricTarget{
- AverageValue: &averageValue,
- },
- },
- },
- },
- "FailedGetObjectMetric": {
- {
- Type: autoscalingv2.ObjectMetricSourceType,
- Object: &autoscalingv2.ObjectMetricSource{
- DescribedObject: autoscalingv2.CrossVersionObjectReference{
- APIVersion: "apps/v1",
- Kind: "Deployment",
- Name: "some-deployment",
- },
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- },
- Target: autoscalingv2.MetricTarget{
- Value: &targetValue,
- },
- },
- },
- },
- "FailedGetExternalMetric": {
- {
- Type: autoscalingv2.ExternalMetricSourceType,
- External: &autoscalingv2.ExternalMetricSource{
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- Selector: &metav1.LabelSelector{},
- },
- Target: autoscalingv2.MetricTarget{
- Value: resource.NewMilliQuantity(300, resource.DecimalSI),
- },
- },
- },
- },
- }
- for reason, specs := range metricsTargets {
- tc := testCase{
- minReplicas: 1,
- maxReplicas: 100,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 3,
- CPUTarget: 10,
- reportedLevels: []uint64{100, 200, 300},
- reportedCPURequests: []resource.Quantity{resource.MustParse("0.1"), resource.MustParse("0.1"), resource.MustParse("0.1")},
- useMetricsAPI: true,
- }
- _, testMetricsClient, testCMClient, testEMClient, _ := tc.prepareTestClient(t)
- tc.testMetricsClient = testMetricsClient
- tc.testCMClient = testCMClient
- tc.testEMClient = testEMClient
- testMetricsClient.PrependReactor("list", "pods", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- return true, &metricsapi.PodMetricsList{}, fmt.Errorf("something went wrong")
- })
- testCMClient.PrependReactor("get", "*", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- return true, &cmapi.MetricValueList{}, fmt.Errorf("something went wrong")
- })
- testEMClient.PrependReactor("list", "*", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- return true, &emapi.ExternalMetricValueList{}, fmt.Errorf("something went wrong")
- })
- tc.expectedConditions = []autoscalingv1.HorizontalPodAutoscalerCondition{
- {Type: autoscalingv1.AbleToScale, Status: v1.ConditionTrue, Reason: "SucceededGetScale"},
- {Type: autoscalingv1.ScalingActive, Status: v1.ConditionFalse, Reason: reason},
- }
- if specs != nil {
- tc.CPUTarget = 0
- } else {
- tc.CPUTarget = 10
- }
- tc.metricsTarget = specs
- tc.runTest(t)
- }
- }
- func TestConditionInvalidSourceType(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 3,
- CPUTarget: 0,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: "CheddarCheese",
- },
- },
- reportedLevels: []uint64{20000},
- expectedConditions: []autoscalingv1.HorizontalPodAutoscalerCondition{
- {
- Type: autoscalingv1.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "SucceededGetScale",
- },
- {
- Type: autoscalingv1.ScalingActive,
- Status: v1.ConditionFalse,
- Reason: "InvalidMetricSourceType",
- },
- },
- }
- tc.runTest(t)
- }
- func TestConditionFailedGetScale(t *testing.T) {
- tc := testCase{
- minReplicas: 1,
- maxReplicas: 100,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 3,
- CPUTarget: 10,
- reportedLevels: []uint64{100, 200, 300},
- reportedCPURequests: []resource.Quantity{resource.MustParse("0.1"), resource.MustParse("0.1"), resource.MustParse("0.1")},
- useMetricsAPI: true,
- expectedConditions: []autoscalingv1.HorizontalPodAutoscalerCondition{
- {
- Type: autoscalingv1.AbleToScale,
- Status: v1.ConditionFalse,
- Reason: "FailedGetScale",
- },
- },
- }
- _, _, _, _, testScaleClient := tc.prepareTestClient(t)
- tc.testScaleClient = testScaleClient
- testScaleClient.PrependReactor("get", "replicationcontrollers", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- return true, &autoscalingv1.Scale{}, fmt.Errorf("something went wrong")
- })
- tc.runTest(t)
- }
- func TestConditionFailedUpdateScale(t *testing.T) {
- tc := testCase{
- minReplicas: 1,
- maxReplicas: 5,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 3,
- CPUTarget: 100,
- reportedLevels: []uint64{150, 150, 150},
- reportedCPURequests: []resource.Quantity{resource.MustParse("0.1"), resource.MustParse("0.1"), resource.MustParse("0.1")},
- useMetricsAPI: true,
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionFalse,
- Reason: "FailedUpdateScale",
- }),
- }
- _, _, _, _, testScaleClient := tc.prepareTestClient(t)
- tc.testScaleClient = testScaleClient
- testScaleClient.PrependReactor("update", "replicationcontrollers", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- return true, &autoscalingv1.Scale{}, fmt.Errorf("something went wrong")
- })
- tc.runTest(t)
- }
- func NoTestBackoffUpscale(t *testing.T) {
- time := metav1.Time{Time: time.Now()}
- tc := testCase{
- minReplicas: 1,
- maxReplicas: 5,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 3,
- CPUTarget: 100,
- reportedLevels: []uint64{150, 150, 150},
- reportedCPURequests: []resource.Quantity{resource.MustParse("0.1"), resource.MustParse("0.1"), resource.MustParse("0.1")},
- useMetricsAPI: true,
- lastScaleTime: &time,
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "ReadyForNewScale",
- }, autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "SucceededRescale",
- }),
- }
- tc.runTest(t)
- }
- func TestNoBackoffUpscaleCM(t *testing.T) {
- averageValue := resource.MustParse("15.0")
- time := metav1.Time{Time: time.Now()}
- tc := testCase{
- minReplicas: 1,
- maxReplicas: 5,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 4,
- CPUTarget: 0,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.PodsMetricSourceType,
- Pods: &autoscalingv2.PodsMetricSource{
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- },
- Target: autoscalingv2.MetricTarget{
- AverageValue: &averageValue,
- },
- },
- },
- },
- reportedLevels: []uint64{20000, 10000, 30000},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- //useMetricsAPI: true,
- lastScaleTime: &time,
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "ReadyForNewScale",
- }, autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "SucceededRescale",
- }, autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.ScalingLimited,
- Status: v1.ConditionFalse,
- Reason: "DesiredWithinRange",
- }),
- }
- tc.runTest(t)
- }
- func TestNoBackoffUpscaleCMNoBackoffCpu(t *testing.T) {
- averageValue := resource.MustParse("15.0")
- time := metav1.Time{Time: time.Now()}
- tc := testCase{
- minReplicas: 1,
- maxReplicas: 5,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 5,
- CPUTarget: 10,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.PodsMetricSourceType,
- Pods: &autoscalingv2.PodsMetricSource{
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- },
- Target: autoscalingv2.MetricTarget{
- AverageValue: &averageValue,
- },
- },
- },
- },
- reportedLevels: []uint64{20000, 10000, 30000},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- useMetricsAPI: true,
- lastScaleTime: &time,
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "ReadyForNewScale",
- }, autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "SucceededRescale",
- }, autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.ScalingLimited,
- Status: v1.ConditionTrue,
- Reason: "TooManyReplicas",
- }),
- }
- tc.runTest(t)
- }
- func TestStabilizeDownscale(t *testing.T) {
- tc := testCase{
- minReplicas: 1,
- maxReplicas: 5,
- specReplicas: 4,
- statusReplicas: 4,
- expectedDesiredReplicas: 4,
- CPUTarget: 100,
- reportedLevels: []uint64{50, 50, 50},
- reportedCPURequests: []resource.Quantity{resource.MustParse("0.1"), resource.MustParse("0.1"), resource.MustParse("0.1")},
- useMetricsAPI: true,
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "ReadyForNewScale",
- }, autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "ScaleDownStabilized",
- }),
- recommendations: []timestampedRecommendation{
- {10, time.Now().Add(-10 * time.Minute)},
- {4, time.Now().Add(-1 * time.Minute)},
- },
- }
- tc.runTest(t)
- }
- // TestComputedToleranceAlgImplementation is a regression test which
- // back-calculates a minimal percentage for downscaling based on a small percentage
- // increase in pod utilization which is calibrated against the tolerance value.
- func TestComputedToleranceAlgImplementation(t *testing.T) {
- startPods := int32(10)
- // 150 mCPU per pod.
- totalUsedCPUOfAllPods := uint64(startPods * 150)
- // Each pod starts out asking for 2X what is really needed.
- // This means we will have a 50% ratio of used/requested
- totalRequestedCPUOfAllPods := int32(2 * totalUsedCPUOfAllPods)
- requestedToUsed := float64(totalRequestedCPUOfAllPods / int32(totalUsedCPUOfAllPods))
- // Spread the amount we ask over 10 pods. We can add some jitter later in reportedLevels.
- perPodRequested := totalRequestedCPUOfAllPods / startPods
- // Force a minimal scaling event by satisfying (tolerance < 1 - resourcesUsedRatio).
- target := math.Abs(1/(requestedToUsed*(1-defaultTestingTolerance))) + .01
- finalCPUPercentTarget := int32(target * 100)
- resourcesUsedRatio := float64(totalUsedCPUOfAllPods) / float64(float64(totalRequestedCPUOfAllPods)*target)
- // i.e. .60 * 20 -> scaled down expectation.
- finalPods := int32(math.Ceil(resourcesUsedRatio * float64(startPods)))
- // To breach tolerance we will create a utilization ratio difference of tolerance to usageRatioToleranceValue)
- tc1 := testCase{
- minReplicas: 0,
- maxReplicas: 1000,
- specReplicas: startPods,
- statusReplicas: startPods,
- expectedDesiredReplicas: finalPods,
- CPUTarget: finalCPUPercentTarget,
- reportedLevels: []uint64{
- totalUsedCPUOfAllPods / 10,
- totalUsedCPUOfAllPods / 10,
- totalUsedCPUOfAllPods / 10,
- totalUsedCPUOfAllPods / 10,
- totalUsedCPUOfAllPods / 10,
- totalUsedCPUOfAllPods / 10,
- totalUsedCPUOfAllPods / 10,
- totalUsedCPUOfAllPods / 10,
- totalUsedCPUOfAllPods / 10,
- totalUsedCPUOfAllPods / 10,
- },
- reportedCPURequests: []resource.Quantity{
- resource.MustParse(fmt.Sprint(perPodRequested+100) + "m"),
- resource.MustParse(fmt.Sprint(perPodRequested-100) + "m"),
- resource.MustParse(fmt.Sprint(perPodRequested+10) + "m"),
- resource.MustParse(fmt.Sprint(perPodRequested-10) + "m"),
- resource.MustParse(fmt.Sprint(perPodRequested+2) + "m"),
- resource.MustParse(fmt.Sprint(perPodRequested-2) + "m"),
- resource.MustParse(fmt.Sprint(perPodRequested+1) + "m"),
- resource.MustParse(fmt.Sprint(perPodRequested-1) + "m"),
- resource.MustParse(fmt.Sprint(perPodRequested) + "m"),
- resource.MustParse(fmt.Sprint(perPodRequested) + "m"),
- },
- useMetricsAPI: true,
- recommendations: []timestampedRecommendation{},
- }
- tc1.runTest(t)
- target = math.Abs(1/(requestedToUsed*(1-defaultTestingTolerance))) + .004
- finalCPUPercentTarget = int32(target * 100)
- tc2 := testCase{
- minReplicas: 0,
- maxReplicas: 1000,
- specReplicas: startPods,
- statusReplicas: startPods,
- expectedDesiredReplicas: startPods,
- CPUTarget: finalCPUPercentTarget,
- reportedLevels: []uint64{
- totalUsedCPUOfAllPods / 10,
- totalUsedCPUOfAllPods / 10,
- totalUsedCPUOfAllPods / 10,
- totalUsedCPUOfAllPods / 10,
- totalUsedCPUOfAllPods / 10,
- totalUsedCPUOfAllPods / 10,
- totalUsedCPUOfAllPods / 10,
- totalUsedCPUOfAllPods / 10,
- totalUsedCPUOfAllPods / 10,
- totalUsedCPUOfAllPods / 10,
- },
- reportedCPURequests: []resource.Quantity{
- resource.MustParse(fmt.Sprint(perPodRequested+100) + "m"),
- resource.MustParse(fmt.Sprint(perPodRequested-100) + "m"),
- resource.MustParse(fmt.Sprint(perPodRequested+10) + "m"),
- resource.MustParse(fmt.Sprint(perPodRequested-10) + "m"),
- resource.MustParse(fmt.Sprint(perPodRequested+2) + "m"),
- resource.MustParse(fmt.Sprint(perPodRequested-2) + "m"),
- resource.MustParse(fmt.Sprint(perPodRequested+1) + "m"),
- resource.MustParse(fmt.Sprint(perPodRequested-1) + "m"),
- resource.MustParse(fmt.Sprint(perPodRequested) + "m"),
- resource.MustParse(fmt.Sprint(perPodRequested) + "m"),
- },
- useMetricsAPI: true,
- recommendations: []timestampedRecommendation{},
- expectedConditions: statusOkWithOverrides(autoscalingv2.HorizontalPodAutoscalerCondition{
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionTrue,
- Reason: "ReadyForNewScale",
- }),
- }
- tc2.runTest(t)
- }
- func TestScaleUpRCImmediately(t *testing.T) {
- time := metav1.Time{Time: time.Now()}
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 1,
- statusReplicas: 1,
- expectedDesiredReplicas: 2,
- verifyCPUCurrent: false,
- reportedLevels: []uint64{0, 0, 0, 0},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- useMetricsAPI: true,
- lastScaleTime: &time,
- expectedConditions: []autoscalingv1.HorizontalPodAutoscalerCondition{
- {Type: autoscalingv1.AbleToScale, Status: v1.ConditionTrue, Reason: "SucceededRescale"},
- },
- }
- tc.runTest(t)
- }
- func TestScaleDownRCImmediately(t *testing.T) {
- time := metav1.Time{Time: time.Now()}
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 5,
- specReplicas: 6,
- statusReplicas: 6,
- expectedDesiredReplicas: 5,
- CPUTarget: 50,
- reportedLevels: []uint64{8000, 9500, 1000},
- reportedCPURequests: []resource.Quantity{resource.MustParse("0.9"), resource.MustParse("1.0"), resource.MustParse("1.1")},
- useMetricsAPI: true,
- lastScaleTime: &time,
- expectedConditions: []autoscalingv1.HorizontalPodAutoscalerCondition{
- {Type: autoscalingv1.AbleToScale, Status: v1.ConditionTrue, Reason: "SucceededRescale"},
- },
- }
- tc.runTest(t)
- }
- func TestAvoidUncessaryUpdates(t *testing.T) {
- now := metav1.Time{Time: time.Now().Add(-time.Hour)}
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 2,
- statusReplicas: 2,
- expectedDesiredReplicas: 2,
- CPUTarget: 30,
- CPUCurrent: 40,
- verifyCPUCurrent: true,
- reportedLevels: []uint64{400, 500, 700},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- reportedPodStartTime: []metav1.Time{coolCpuCreationTime(), hotCpuCreationTime(), hotCpuCreationTime()},
- useMetricsAPI: true,
- lastScaleTime: &now,
- recommendations: []timestampedRecommendation{},
- }
- testClient, _, _, _, _ := tc.prepareTestClient(t)
- tc.testClient = testClient
- testClient.PrependReactor("list", "horizontalpodautoscalers", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- tc.Lock()
- defer tc.Unlock()
- // fake out the verification logic and mark that we're done processing
- go func() {
- // wait a tick and then mark that we're finished (otherwise, we have no
- // way to indicate that we're finished, because the function decides not to do anything)
- time.Sleep(1 * time.Second)
- tc.Lock()
- tc.statusUpdated = true
- tc.Unlock()
- tc.processed <- "test-hpa"
- }()
- quantity := resource.MustParse("400m")
- obj := &autoscalingv2.HorizontalPodAutoscalerList{
- Items: []autoscalingv2.HorizontalPodAutoscaler{
- {
- ObjectMeta: metav1.ObjectMeta{
- Name: "test-hpa",
- Namespace: "test-namespace",
- SelfLink: "experimental/v1/namespaces/test-namespace/horizontalpodautoscalers/test-hpa",
- },
- Spec: autoscalingv2.HorizontalPodAutoscalerSpec{
- ScaleTargetRef: autoscalingv2.CrossVersionObjectReference{
- Kind: "ReplicationController",
- Name: "test-rc",
- APIVersion: "v1",
- },
- MinReplicas: &tc.minReplicas,
- MaxReplicas: tc.maxReplicas,
- },
- Status: autoscalingv2.HorizontalPodAutoscalerStatus{
- CurrentReplicas: tc.specReplicas,
- DesiredReplicas: tc.specReplicas,
- LastScaleTime: tc.lastScaleTime,
- CurrentMetrics: []autoscalingv2.MetricStatus{
- {
- Type: autoscalingv2.ResourceMetricSourceType,
- Resource: &autoscalingv2.ResourceMetricStatus{
- Name: v1.ResourceCPU,
- Current: autoscalingv2.MetricValueStatus{
- AverageValue: &quantity,
- AverageUtilization: &tc.CPUCurrent,
- },
- },
- },
- },
- Conditions: []autoscalingv2.HorizontalPodAutoscalerCondition{
- {
- Type: autoscalingv2.AbleToScale,
- Status: v1.ConditionTrue,
- LastTransitionTime: *tc.lastScaleTime,
- Reason: "ReadyForNewScale",
- Message: "recommended size matches current size",
- },
- {
- Type: autoscalingv2.ScalingActive,
- Status: v1.ConditionTrue,
- LastTransitionTime: *tc.lastScaleTime,
- Reason: "ValidMetricFound",
- Message: "the HPA was able to successfully calculate a replica count from cpu resource utilization (percentage of request)",
- },
- {
- Type: autoscalingv2.ScalingLimited,
- Status: v1.ConditionTrue,
- LastTransitionTime: *tc.lastScaleTime,
- Reason: "TooFewReplicas",
- Message: "the desired replica count is less than the minimum replica count",
- },
- },
- },
- },
- },
- }
- // and... convert to autoscaling v1 to return the right type
- objv1, err := unsafeConvertToVersionVia(obj, autoscalingv1.SchemeGroupVersion)
- if err != nil {
- return true, nil, err
- }
- return true, objv1, nil
- })
- testClient.PrependReactor("update", "horizontalpodautoscalers", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- assert.Fail(t, "should not have attempted to update the HPA when nothing changed")
- // mark that we've processed this HPA
- tc.processed <- ""
- return true, nil, fmt.Errorf("unexpected call")
- })
- controller, informerFactory := tc.setupController(t)
- tc.runTestWithController(t, controller, informerFactory)
- }
- func TestConvertDesiredReplicasWithRules(t *testing.T) {
- conversionTestCases := []struct {
- currentReplicas int32
- expectedDesiredReplicas int32
- hpaMinReplicas int32
- hpaMaxReplicas int32
- expectedConvertedDesiredReplicas int32
- expectedCondition string
- annotation string
- }{
- {
- currentReplicas: 5,
- expectedDesiredReplicas: 7,
- hpaMinReplicas: 3,
- hpaMaxReplicas: 8,
- expectedConvertedDesiredReplicas: 7,
- expectedCondition: "DesiredWithinRange",
- annotation: "prenormalized desired replicas within range",
- },
- {
- currentReplicas: 3,
- expectedDesiredReplicas: 1,
- hpaMinReplicas: 2,
- hpaMaxReplicas: 8,
- expectedConvertedDesiredReplicas: 2,
- expectedCondition: "TooFewReplicas",
- annotation: "prenormalized desired replicas < minReplicas",
- },
- {
- currentReplicas: 1,
- expectedDesiredReplicas: 0,
- hpaMinReplicas: 0,
- hpaMaxReplicas: 10,
- expectedConvertedDesiredReplicas: 0,
- expectedCondition: "DesiredWithinRange",
- annotation: "prenormalized desired zeroed replicas within range",
- },
- {
- currentReplicas: 20,
- expectedDesiredReplicas: 1000,
- hpaMinReplicas: 1,
- hpaMaxReplicas: 10,
- expectedConvertedDesiredReplicas: 10,
- expectedCondition: "TooManyReplicas",
- annotation: "maxReplicas is the limit because maxReplicas < scaleUpLimit",
- },
- {
- currentReplicas: 3,
- expectedDesiredReplicas: 1000,
- hpaMinReplicas: 1,
- hpaMaxReplicas: 2000,
- expectedConvertedDesiredReplicas: calculateScaleUpLimit(3),
- expectedCondition: "ScaleUpLimit",
- annotation: "scaleUpLimit is the limit because scaleUpLimit < maxReplicas",
- },
- }
- for _, ctc := range conversionTestCases {
- t.Run(ctc.annotation, func(t *testing.T) {
- actualConvertedDesiredReplicas, actualCondition, _ := convertDesiredReplicasWithRules(
- ctc.currentReplicas, ctc.expectedDesiredReplicas, ctc.hpaMinReplicas, ctc.hpaMaxReplicas,
- )
- assert.Equal(t, ctc.expectedConvertedDesiredReplicas, actualConvertedDesiredReplicas, ctc.annotation)
- assert.Equal(t, ctc.expectedCondition, actualCondition, ctc.annotation)
- })
- }
- }
- func generateScalingRules(pods, podsPeriod, percent, percentPeriod, stabilizationWindow int32) *autoscalingv2.HPAScalingRules {
- policy := autoscalingv2.MaxPolicySelect
- directionBehavior := autoscalingv2.HPAScalingRules{
- StabilizationWindowSeconds: utilpointer.Int32Ptr(stabilizationWindow),
- SelectPolicy: &policy,
- }
- if pods != 0 {
- directionBehavior.Policies = append(directionBehavior.Policies,
- autoscalingv2.HPAScalingPolicy{Type: autoscalingv2.PodsScalingPolicy, Value: pods, PeriodSeconds: podsPeriod})
- }
- if percent != 0 {
- directionBehavior.Policies = append(directionBehavior.Policies,
- autoscalingv2.HPAScalingPolicy{Type: autoscalingv2.PercentScalingPolicy, Value: percent, PeriodSeconds: percentPeriod})
- }
- return &directionBehavior
- }
- // generateEventsUniformDistribution generates events that uniformly spread in the time window
- // time.Now()-periodSeconds ; time.Now()
- // It split the time window into several segments (by the number of events) and put the event in the center of the segment
- // it is needed if you want to create events for several policies (to check how "outdated" flag is set).
- // E.g. generateEventsUniformDistribution([]int{1,2,3,4}, 120) will spread events uniformly for the last 120 seconds:
- //
- // 1 2 3 4
- // -----------------------------------------------
- // ^ ^ ^ ^ ^
- // -120s -90s -60s -30s now()
- // And we can safely have two different stabilizationWindows:
- // - 60s (guaranteed to have last half of events)
- // - 120s (guaranteed to have all events)
- func generateEventsUniformDistribution(rawEvents []int, periodSeconds int) []timestampedScaleEvent {
- events := make([]timestampedScaleEvent, len(rawEvents))
- segmentDuration := float64(periodSeconds) / float64(len(rawEvents))
- for idx, event := range rawEvents {
- segmentBoundary := time.Duration(float64(periodSeconds) - segmentDuration*float64(idx+1) + segmentDuration/float64(2))
- events[idx] = timestampedScaleEvent{
- replicaChange: int32(event),
- timestamp: time.Now().Add(-time.Second * segmentBoundary),
- }
- }
- return events
- }
- func TestNormalizeDesiredReplicas(t *testing.T) {
- tests := []struct {
- name string
- key string
- recommendations []timestampedRecommendation
- prenormalizedDesiredReplicas int32
- expectedStabilizedReplicas int32
- expectedLogLength int
- }{
- {
- "empty log",
- "",
- []timestampedRecommendation{},
- 5,
- 5,
- 1,
- },
- {
- "stabilize",
- "",
- []timestampedRecommendation{
- {4, time.Now().Add(-2 * time.Minute)},
- {5, time.Now().Add(-1 * time.Minute)},
- },
- 3,
- 5,
- 3,
- },
- {
- "no stabilize",
- "",
- []timestampedRecommendation{
- {1, time.Now().Add(-2 * time.Minute)},
- {2, time.Now().Add(-1 * time.Minute)},
- },
- 3,
- 3,
- 3,
- },
- {
- "no stabilize - old recommendations",
- "",
- []timestampedRecommendation{
- {10, time.Now().Add(-10 * time.Minute)},
- {9, time.Now().Add(-9 * time.Minute)},
- },
- 3,
- 3,
- 2,
- },
- {
- "stabilize - old recommendations",
- "",
- []timestampedRecommendation{
- {10, time.Now().Add(-10 * time.Minute)},
- {4, time.Now().Add(-1 * time.Minute)},
- {5, time.Now().Add(-2 * time.Minute)},
- {9, time.Now().Add(-9 * time.Minute)},
- },
- 3,
- 5,
- 4,
- },
- }
- for _, tc := range tests {
- hc := HorizontalController{
- downscaleStabilisationWindow: 5 * time.Minute,
- recommendations: map[string][]timestampedRecommendation{
- tc.key: tc.recommendations,
- },
- }
- r := hc.stabilizeRecommendation(tc.key, tc.prenormalizedDesiredReplicas)
- if r != tc.expectedStabilizedReplicas {
- t.Errorf("[%s] got %d stabilized replicas, expected %d", tc.name, r, tc.expectedStabilizedReplicas)
- }
- if len(hc.recommendations[tc.key]) != tc.expectedLogLength {
- t.Errorf("[%s] after stabilization recommendations log has %d entries, expected %d", tc.name, len(hc.recommendations[tc.key]), tc.expectedLogLength)
- }
- }
- }
- func TestScalingWithRules(t *testing.T) {
- type TestCase struct {
- name string
- key string
- // controller arguments
- scaleUpEvents []timestampedScaleEvent
- scaleDownEvents []timestampedScaleEvent
- // HPA Spec arguments
- specMinReplicas int32
- specMaxReplicas int32
- scaleUpRules *autoscalingv2.HPAScalingRules
- scaleDownRules *autoscalingv2.HPAScalingRules
- // external world state
- currentReplicas int32
- prenormalizedDesiredReplicas int32
- // test expected result
- expectedReplicas int32
- expectedCondition string
- testThis bool
- }
- tests := []TestCase{
- {
- currentReplicas: 5,
- prenormalizedDesiredReplicas: 7,
- specMinReplicas: 3,
- specMaxReplicas: 8,
- expectedReplicas: 7,
- expectedCondition: "DesiredWithinRange",
- name: "prenormalized desired replicas within range",
- },
- {
- currentReplicas: 3,
- prenormalizedDesiredReplicas: 1,
- specMinReplicas: 2,
- specMaxReplicas: 8,
- expectedReplicas: 2,
- expectedCondition: "TooFewReplicas",
- name: "prenormalized desired replicas < minReplicas",
- },
- {
- currentReplicas: 1,
- prenormalizedDesiredReplicas: 0,
- specMinReplicas: 0,
- specMaxReplicas: 10,
- expectedReplicas: 0,
- expectedCondition: "DesiredWithinRange",
- name: "prenormalized desired replicas within range when minReplicas is 0",
- },
- {
- currentReplicas: 20,
- prenormalizedDesiredReplicas: 1000,
- specMinReplicas: 1,
- specMaxReplicas: 10,
- expectedReplicas: 10,
- expectedCondition: "TooManyReplicas",
- name: "maxReplicas is the limit because maxReplicas < scaleUpLimit",
- },
- {
- currentReplicas: 100,
- prenormalizedDesiredReplicas: 1000,
- specMinReplicas: 100,
- specMaxReplicas: 150,
- expectedReplicas: 150,
- expectedCondition: "TooManyReplicas",
- name: "desired replica count is more than the maximum replica count",
- },
- {
- currentReplicas: 3,
- prenormalizedDesiredReplicas: 1000,
- specMinReplicas: 1,
- specMaxReplicas: 2000,
- expectedReplicas: 4,
- expectedCondition: "ScaleUpLimit",
- scaleUpRules: generateScalingRules(0, 0, 1, 60, 0),
- name: "scaleUpLimit is the limit because scaleUpLimit < maxReplicas with user policies",
- },
- {
- currentReplicas: 1000,
- prenormalizedDesiredReplicas: 3,
- specMinReplicas: 3,
- specMaxReplicas: 2000,
- scaleDownRules: generateScalingRules(20, 60, 0, 0, 0),
- expectedReplicas: 980,
- expectedCondition: "ScaleDownLimit",
- name: "scaleDownLimit is the limit because scaleDownLimit > minReplicas with user defined policies",
- testThis: true,
- },
- // ScaleUp without PeriodSeconds usage
- {
- name: "scaleUp with default behavior",
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- currentReplicas: 10,
- prenormalizedDesiredReplicas: 50,
- expectedReplicas: 20,
- expectedCondition: "ScaleUpLimit",
- },
- {
- name: "scaleUp with pods policy larger than percent policy",
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleUpRules: generateScalingRules(100, 60, 100, 60, 0),
- currentReplicas: 10,
- prenormalizedDesiredReplicas: 500,
- expectedReplicas: 110,
- expectedCondition: "ScaleUpLimit",
- },
- {
- name: "scaleUp with percent policy larger than pods policy",
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleUpRules: generateScalingRules(2, 60, 100, 60, 0),
- currentReplicas: 10,
- prenormalizedDesiredReplicas: 500,
- expectedReplicas: 20,
- expectedCondition: "ScaleUpLimit",
- },
- {
- name: "scaleUp with spec MaxReplicas limitation with large pod policy",
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleUpRules: generateScalingRules(100, 60, 0, 0, 0),
- currentReplicas: 10,
- prenormalizedDesiredReplicas: 50,
- expectedReplicas: 50,
- expectedCondition: "DesiredWithinRange",
- },
- {
- name: "scaleUp with spec MaxReplicas limitation with large percent policy",
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleUpRules: generateScalingRules(10000, 60, 0, 0, 0),
- currentReplicas: 10,
- prenormalizedDesiredReplicas: 50,
- expectedReplicas: 50,
- expectedCondition: "DesiredWithinRange",
- },
- {
- name: "scaleUp with pod policy limitation",
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleUpRules: generateScalingRules(30, 60, 0, 0, 0),
- currentReplicas: 10,
- prenormalizedDesiredReplicas: 50,
- expectedReplicas: 40,
- expectedCondition: "ScaleUpLimit",
- },
- {
- name: "scaleUp with percent policy limitation",
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleUpRules: generateScalingRules(0, 0, 200, 60, 0),
- currentReplicas: 10,
- prenormalizedDesiredReplicas: 50,
- expectedReplicas: 30,
- expectedCondition: "ScaleUpLimit",
- },
- {
- name: "scaleDown with percent policy larger than pod policy",
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleDownRules: generateScalingRules(20, 60, 1, 60, 300),
- currentReplicas: 100,
- prenormalizedDesiredReplicas: 2,
- expectedReplicas: 80,
- expectedCondition: "ScaleDownLimit",
- },
- {
- name: "scaleDown with pod policy larger than percent policy",
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleDownRules: generateScalingRules(2, 60, 1, 60, 300),
- currentReplicas: 100,
- prenormalizedDesiredReplicas: 2,
- expectedReplicas: 98,
- expectedCondition: "ScaleDownLimit",
- },
- {
- name: "scaleDown with spec MinReplicas=nil limitation with large pod policy",
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleDownRules: generateScalingRules(100, 60, 0, 0, 300),
- currentReplicas: 10,
- prenormalizedDesiredReplicas: 0,
- expectedReplicas: 1,
- expectedCondition: "TooFewReplicas",
- },
- {
- name: "scaleDown with spec MinReplicas limitation with large pod policy",
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleDownRules: generateScalingRules(100, 60, 0, 0, 300),
- currentReplicas: 10,
- prenormalizedDesiredReplicas: 0,
- expectedReplicas: 1,
- expectedCondition: "TooFewReplicas",
- },
- {
- name: "scaleDown with spec MinReplicas limitation with large percent policy",
- specMinReplicas: 5,
- specMaxReplicas: 1000,
- scaleDownRules: generateScalingRules(0, 0, 100, 60, 300),
- currentReplicas: 10,
- prenormalizedDesiredReplicas: 2,
- expectedReplicas: 5,
- expectedCondition: "TooFewReplicas",
- },
- {
- name: "scaleDown with pod policy limitation",
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleDownRules: generateScalingRules(5, 60, 0, 0, 300),
- currentReplicas: 10,
- prenormalizedDesiredReplicas: 2,
- expectedReplicas: 5,
- expectedCondition: "ScaleDownLimit",
- },
- {
- name: "scaleDown with percent policy limitation",
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleDownRules: generateScalingRules(0, 0, 50, 60, 300),
- currentReplicas: 10,
- prenormalizedDesiredReplicas: 5,
- expectedReplicas: 5,
- expectedCondition: "DesiredWithinRange",
- },
- {
- name: "scaleUp with spec MaxReplicas limitation with large pod policy and events",
- scaleUpEvents: generateEventsUniformDistribution([]int{1, 5, 9}, 120),
- specMinReplicas: 1,
- specMaxReplicas: 200,
- scaleUpRules: generateScalingRules(300, 60, 0, 0, 0),
- currentReplicas: 100,
- prenormalizedDesiredReplicas: 500,
- expectedReplicas: 200, // 200 < 100 - 15 + 300
- expectedCondition: "TooManyReplicas",
- },
- {
- name: "scaleUp with spec MaxReplicas limitation with large percent policy and events",
- scaleUpEvents: generateEventsUniformDistribution([]int{1, 5, 9}, 120),
- specMinReplicas: 1,
- specMaxReplicas: 200,
- scaleUpRules: generateScalingRules(0, 0, 10000, 60, 0),
- currentReplicas: 100,
- prenormalizedDesiredReplicas: 500,
- expectedReplicas: 200,
- expectedCondition: "TooManyReplicas",
- },
- {
- // corner case for calculating the scaleUpLimit, when we changed pod policy after a lot of scaleUp events
- // in this case we shouldn't allow scale up, though, the naive formula will suggest that scaleUplimit is less then CurrentReplicas (100-15+5 < 100)
- name: "scaleUp with currentReplicas limitation with rate.PeriodSeconds with a lot of recent scale up events",
- scaleUpEvents: generateEventsUniformDistribution([]int{1, 5, 9}, 120),
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleUpRules: generateScalingRules(5, 120, 0, 0, 0),
- currentReplicas: 100,
- prenormalizedDesiredReplicas: 500,
- expectedReplicas: 100, // 120 seconds ago we had (100 - 15) replicas, now the rate.Pods = 5,
- expectedCondition: "ScaleUpLimit",
- },
- {
- name: "scaleUp with pod policy and previous scale up events",
- scaleUpEvents: generateEventsUniformDistribution([]int{1, 5, 9}, 120),
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleUpRules: generateScalingRules(150, 120, 0, 0, 0),
- currentReplicas: 100,
- prenormalizedDesiredReplicas: 500,
- expectedReplicas: 235, // 100 - 15 + 150
- expectedCondition: "ScaleUpLimit",
- },
- {
- name: "scaleUp with percent policy and previous scale up events",
- scaleUpEvents: generateEventsUniformDistribution([]int{1, 5, 9}, 120),
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleUpRules: generateScalingRules(0, 0, 200, 120, 0),
- currentReplicas: 100,
- prenormalizedDesiredReplicas: 500,
- expectedReplicas: 255, // (100 - 15) + 200%
- expectedCondition: "ScaleUpLimit",
- },
- // ScaleDown with PeriodSeconds usage
- {
- name: "scaleDown with default policy and previous events",
- scaleDownEvents: generateEventsUniformDistribution([]int{1, 5, 9}, 120),
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- currentReplicas: 10,
- prenormalizedDesiredReplicas: 5,
- expectedReplicas: 5, // without scaleDown rate limitations the PeriodSeconds does not influence anything
- expectedCondition: "DesiredWithinRange",
- },
- {
- name: "scaleDown with spec MinReplicas=nil limitation with large pod policy and previous events",
- scaleDownEvents: generateEventsUniformDistribution([]int{1, 5, 9}, 120),
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleDownRules: generateScalingRules(115, 120, 0, 0, 300),
- currentReplicas: 100,
- prenormalizedDesiredReplicas: 0,
- expectedReplicas: 1,
- expectedCondition: "TooFewReplicas",
- },
- {
- name: "scaleDown with spec MinReplicas limitation with large pod policy and previous events",
- scaleDownEvents: generateEventsUniformDistribution([]int{1, 5, 9}, 120),
- specMinReplicas: 5,
- specMaxReplicas: 1000,
- scaleDownRules: generateScalingRules(130, 120, 0, 0, 300),
- currentReplicas: 100,
- prenormalizedDesiredReplicas: 0,
- expectedReplicas: 5,
- expectedCondition: "TooFewReplicas",
- },
- {
- name: "scaleDown with spec MinReplicas limitation with large percent policy and previous events",
- scaleDownEvents: generateEventsUniformDistribution([]int{1, 5, 9}, 120),
- specMinReplicas: 5,
- specMaxReplicas: 1000,
- scaleDownRules: generateScalingRules(0, 0, 100, 120, 300), // 100% removal - is always to 0 => limited by MinReplicas
- currentReplicas: 100,
- prenormalizedDesiredReplicas: 2,
- expectedReplicas: 5,
- expectedCondition: "TooFewReplicas",
- },
- {
- name: "scaleDown with pod policy limitation and previous events",
- scaleDownEvents: generateEventsUniformDistribution([]int{1, 5, 9}, 120),
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleDownRules: generateScalingRules(5, 120, 0, 0, 300),
- currentReplicas: 100,
- prenormalizedDesiredReplicas: 2,
- expectedReplicas: 100, // 100 + 15 - 5
- expectedCondition: "ScaleDownLimit",
- },
- {
- name: "scaleDown with percent policy limitation and previous events",
- scaleDownEvents: generateEventsUniformDistribution([]int{2, 4, 6}, 120),
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleDownRules: generateScalingRules(0, 0, 50, 120, 300),
- currentReplicas: 100,
- prenormalizedDesiredReplicas: 0,
- expectedReplicas: 56, // (100 + 12) - 50%
- expectedCondition: "ScaleDownLimit",
- },
- {
- // corner case for calculating the scaleDownLimit, when we changed pod or percent policy after a lot of scaleDown events
- // in this case we shouldn't allow scale down, though, the naive formula will suggest that scaleDownlimit is more then CurrentReplicas (100+30-10% > 100)
- name: "scaleDown with previous events preventing further scale down",
- scaleDownEvents: generateEventsUniformDistribution([]int{10, 10, 10}, 120),
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleDownRules: generateScalingRules(0, 0, 10, 120, 300),
- currentReplicas: 100,
- prenormalizedDesiredReplicas: 0,
- expectedReplicas: 100, // (100 + 30) - 10% = 117 is more then 100 (currentReplicas), keep 100
- expectedCondition: "ScaleDownLimit",
- },
- {
- // corner case, the same as above, but calculation shows that we should go below zero
- name: "scaleDown with with previous events still allowing more scale down",
- scaleDownEvents: generateEventsUniformDistribution([]int{10, 10, 10}, 120),
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleDownRules: generateScalingRules(0, 0, 1000, 120, 300),
- currentReplicas: 10,
- prenormalizedDesiredReplicas: 5,
- expectedReplicas: 5, // (10 + 30) - 1000% = -360 is less than 0 and less then 5 (desired by metrics), set 5
- expectedCondition: "DesiredWithinRange",
- },
- {
- name: "check 'outdated' flag for events for one behavior for up",
- scaleUpEvents: generateEventsUniformDistribution([]int{8, 12, 9, 11}, 120),
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleUpRules: generateScalingRules(1000, 60, 0, 0, 0),
- currentReplicas: 100,
- prenormalizedDesiredReplicas: 200,
- expectedReplicas: 200,
- expectedCondition: "DesiredWithinRange",
- },
- {
- name: "check that events were not marked 'outdated' for two different policies in the behavior for up",
- scaleUpEvents: generateEventsUniformDistribution([]int{8, 12, 9, 11}, 120),
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleUpRules: generateScalingRules(1000, 120, 100, 60, 0),
- currentReplicas: 100,
- prenormalizedDesiredReplicas: 200,
- expectedReplicas: 200,
- expectedCondition: "DesiredWithinRange",
- },
- {
- name: "check that events were marked 'outdated' for two different policies in the behavior for up",
- scaleUpEvents: generateEventsUniformDistribution([]int{8, 12, 9, 11}, 120),
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleUpRules: generateScalingRules(1000, 30, 100, 60, 0),
- currentReplicas: 100,
- prenormalizedDesiredReplicas: 200,
- expectedReplicas: 200,
- expectedCondition: "DesiredWithinRange",
- },
- {
- name: "check 'outdated' flag for events for one behavior for down",
- scaleDownEvents: generateEventsUniformDistribution([]int{8, 12, 9, 11}, 120),
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleDownRules: generateScalingRules(1000, 60, 0, 0, 300),
- currentReplicas: 100,
- prenormalizedDesiredReplicas: 5,
- expectedReplicas: 5,
- expectedCondition: "DesiredWithinRange",
- },
- {
- name: "check that events were not marked 'outdated' for two different policies in the behavior for down",
- scaleDownEvents: generateEventsUniformDistribution([]int{8, 12, 9, 11}, 120),
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleDownRules: generateScalingRules(1000, 120, 100, 60, 300),
- currentReplicas: 100,
- prenormalizedDesiredReplicas: 5,
- expectedReplicas: 5,
- expectedCondition: "DesiredWithinRange",
- },
- {
- name: "check that events were marked 'outdated' for two different policies in the behavior for down",
- scaleDownEvents: generateEventsUniformDistribution([]int{8, 12, 9, 11}, 120),
- specMinReplicas: 1,
- specMaxReplicas: 1000,
- scaleDownRules: generateScalingRules(1000, 30, 100, 60, 300),
- currentReplicas: 100,
- prenormalizedDesiredReplicas: 5,
- expectedReplicas: 5,
- expectedCondition: "DesiredWithinRange",
- },
- }
- for _, tc := range tests {
- t.Run(tc.name, func(t *testing.T) {
- if tc.testThis {
- return
- }
- hc := HorizontalController{
- scaleUpEvents: map[string][]timestampedScaleEvent{
- tc.key: tc.scaleUpEvents,
- },
- scaleDownEvents: map[string][]timestampedScaleEvent{
- tc.key: tc.scaleDownEvents,
- },
- }
- arg := NormalizationArg{
- Key: tc.key,
- ScaleUpBehavior: autoscalingapiv2beta2.GenerateHPAScaleUpRules(tc.scaleUpRules),
- ScaleDownBehavior: autoscalingapiv2beta2.GenerateHPAScaleDownRules(tc.scaleDownRules),
- MinReplicas: tc.specMinReplicas,
- MaxReplicas: tc.specMaxReplicas,
- DesiredReplicas: tc.prenormalizedDesiredReplicas,
- CurrentReplicas: tc.currentReplicas,
- }
- replicas, condition, _ := hc.convertDesiredReplicasWithBehaviorRate(arg)
- assert.Equal(t, tc.expectedReplicas, replicas, "expected replicas do not match with converted replicas")
- assert.Equal(t, tc.expectedCondition, condition, "HPA condition does not match with expected condition")
- })
- }
- }
- // TestStoreScaleEvents tests events storage and usage
- func TestStoreScaleEvents(t *testing.T) {
- type TestCase struct {
- name string
- key string
- replicaChange int32
- prevScaleEvents []timestampedScaleEvent
- newScaleEvents []timestampedScaleEvent
- scalingRules *autoscalingv2.HPAScalingRules
- expectedReplicasChange int32
- }
- tests := []TestCase{
- {
- name: "empty entries with default behavior",
- replicaChange: 5,
- prevScaleEvents: []timestampedScaleEvent{}, // no history -> 0 replica change
- newScaleEvents: []timestampedScaleEvent{}, // no behavior -> no events are stored
- expectedReplicasChange: 0,
- },
- {
- name: "empty entries with two-policy-behavior",
- replicaChange: 5,
- prevScaleEvents: []timestampedScaleEvent{}, // no history -> 0 replica change
- newScaleEvents: []timestampedScaleEvent{{5, time.Now(), false}},
- scalingRules: generateScalingRules(10, 60, 100, 60, 0),
- expectedReplicasChange: 0,
- },
- {
- name: "one outdated entry to be kept untouched without behavior",
- replicaChange: 5,
- prevScaleEvents: []timestampedScaleEvent{
- {7, time.Now().Add(-time.Second * time.Duration(61)), false}, // outdated event, should be replaced
- },
- newScaleEvents: []timestampedScaleEvent{
- {7, time.Now(), false}, // no behavior -> we don't touch stored events
- },
- expectedReplicasChange: 0,
- },
- {
- name: "one outdated entry to be replaced with behavior",
- replicaChange: 5,
- prevScaleEvents: []timestampedScaleEvent{
- {7, time.Now().Add(-time.Second * time.Duration(61)), false}, // outdated event, should be replaced
- },
- newScaleEvents: []timestampedScaleEvent{
- {5, time.Now(), false},
- },
- scalingRules: generateScalingRules(10, 60, 100, 60, 0),
- expectedReplicasChange: 0,
- },
- {
- name: "one actual entry to be not touched with behavior",
- replicaChange: 5,
- prevScaleEvents: []timestampedScaleEvent{
- {7, time.Now().Add(-time.Second * time.Duration(58)), false},
- },
- newScaleEvents: []timestampedScaleEvent{
- {7, time.Now(), false},
- {5, time.Now(), false},
- },
- scalingRules: generateScalingRules(10, 60, 100, 60, 0),
- expectedReplicasChange: 7,
- },
- {
- name: "two entries, one of them to be replaced",
- replicaChange: 5,
- prevScaleEvents: []timestampedScaleEvent{
- {7, time.Now().Add(-time.Second * time.Duration(61)), false}, // outdated event, should be replaced
- {6, time.Now().Add(-time.Second * time.Duration(59)), false},
- },
- newScaleEvents: []timestampedScaleEvent{
- {5, time.Now(), false},
- {6, time.Now(), false},
- },
- scalingRules: generateScalingRules(10, 60, 0, 0, 0),
- expectedReplicasChange: 6,
- },
- {
- name: "replace one entry, use policies with different periods",
- replicaChange: 5,
- prevScaleEvents: []timestampedScaleEvent{
- {8, time.Now().Add(-time.Second * time.Duration(29)), false},
- {6, time.Now().Add(-time.Second * time.Duration(59)), false},
- {7, time.Now().Add(-time.Second * time.Duration(61)), false}, // outdated event, should be marked as outdated
- {9, time.Now().Add(-time.Second * time.Duration(61)), false}, // outdated event, should be replaced
- },
- newScaleEvents: []timestampedScaleEvent{
- {8, time.Now(), false},
- {6, time.Now(), false},
- {7, time.Now(), true},
- {5, time.Now(), false},
- },
- scalingRules: generateScalingRules(10, 60, 100, 30, 0),
- expectedReplicasChange: 14,
- },
- {
- name: "two entries, both actual",
- replicaChange: 5,
- prevScaleEvents: []timestampedScaleEvent{
- {7, time.Now().Add(-time.Second * time.Duration(58)), false},
- {6, time.Now().Add(-time.Second * time.Duration(59)), false},
- },
- newScaleEvents: []timestampedScaleEvent{
- {7, time.Now(), false},
- {6, time.Now(), false},
- {5, time.Now(), false},
- },
- scalingRules: generateScalingRules(10, 120, 100, 30, 0),
- expectedReplicasChange: 13,
- },
- }
- for _, tc := range tests {
- t.Run(tc.name, func(t *testing.T) {
- // testing scale up
- var behaviorUp *autoscalingv2.HorizontalPodAutoscalerBehavior
- if tc.scalingRules != nil {
- behaviorUp = &autoscalingv2.HorizontalPodAutoscalerBehavior{
- ScaleUp: tc.scalingRules,
- }
- }
- hcUp := HorizontalController{
- scaleUpEvents: map[string][]timestampedScaleEvent{
- tc.key: append([]timestampedScaleEvent{}, tc.prevScaleEvents...),
- },
- }
- gotReplicasChangeUp := getReplicasChangePerPeriod(60, hcUp.scaleUpEvents[tc.key])
- assert.Equal(t, tc.expectedReplicasChange, gotReplicasChangeUp)
- hcUp.storeScaleEvent(behaviorUp, tc.key, 10, 10+tc.replicaChange)
- if !assert.Len(t, hcUp.scaleUpEvents[tc.key], len(tc.newScaleEvents), "up: scale events differ in length") {
- return
- }
- for i, gotEvent := range hcUp.scaleUpEvents[tc.key] {
- expEvent := tc.newScaleEvents[i]
- assert.Equal(t, expEvent.replicaChange, gotEvent.replicaChange, "up: idx:%v replicaChange", i)
- assert.Equal(t, expEvent.outdated, gotEvent.outdated, "up: idx:%v outdated", i)
- }
- // testing scale down
- var behaviorDown *autoscalingv2.HorizontalPodAutoscalerBehavior
- if tc.scalingRules != nil {
- behaviorDown = &autoscalingv2.HorizontalPodAutoscalerBehavior{
- ScaleDown: tc.scalingRules,
- }
- }
- hcDown := HorizontalController{
- scaleDownEvents: map[string][]timestampedScaleEvent{
- tc.key: append([]timestampedScaleEvent{}, tc.prevScaleEvents...),
- },
- }
- gotReplicasChangeDown := getReplicasChangePerPeriod(60, hcDown.scaleDownEvents[tc.key])
- assert.Equal(t, tc.expectedReplicasChange, gotReplicasChangeDown)
- hcDown.storeScaleEvent(behaviorDown, tc.key, 10, 10-tc.replicaChange)
- if !assert.Len(t, hcDown.scaleDownEvents[tc.key], len(tc.newScaleEvents), "down: scale events differ in length") {
- return
- }
- for i, gotEvent := range hcDown.scaleDownEvents[tc.key] {
- expEvent := tc.newScaleEvents[i]
- assert.Equal(t, expEvent.replicaChange, gotEvent.replicaChange, "down: idx:%v replicaChange", i)
- assert.Equal(t, expEvent.outdated, gotEvent.outdated, "down: idx:%v outdated", i)
- }
- })
- }
- }
- func TestNormalizeDesiredReplicasWithBehavior(t *testing.T) {
- type TestCase struct {
- name string
- key string
- recommendations []timestampedRecommendation
- currentReplicas int32
- prenormalizedDesiredReplicas int32
- expectedStabilizedReplicas int32
- expectedRecommendations []timestampedRecommendation
- scaleUpStabilizationWindowSeconds int32
- scaleDownStabilizationWindowSeconds int32
- }
- tests := []TestCase{
- {
- name: "empty recommendations for scaling down",
- key: "",
- recommendations: []timestampedRecommendation{},
- currentReplicas: 100,
- prenormalizedDesiredReplicas: 5,
- expectedStabilizedReplicas: 5,
- expectedRecommendations: []timestampedRecommendation{
- {5, time.Now()},
- },
- },
- {
- name: "simple scale down stabilization",
- key: "",
- recommendations: []timestampedRecommendation{
- {4, time.Now().Add(-2 * time.Minute)},
- {5, time.Now().Add(-1 * time.Minute)}},
- currentReplicas: 100,
- prenormalizedDesiredReplicas: 3,
- expectedStabilizedReplicas: 5,
- expectedRecommendations: []timestampedRecommendation{
- {4, time.Now()},
- {5, time.Now()},
- {3, time.Now()},
- },
- scaleDownStabilizationWindowSeconds: 60 * 3,
- },
- {
- name: "simple scale up stabilization",
- key: "",
- recommendations: []timestampedRecommendation{
- {4, time.Now().Add(-2 * time.Minute)},
- {5, time.Now().Add(-1 * time.Minute)}},
- currentReplicas: 1,
- prenormalizedDesiredReplicas: 7,
- expectedStabilizedReplicas: 4,
- expectedRecommendations: []timestampedRecommendation{
- {4, time.Now()},
- {5, time.Now()},
- {7, time.Now()},
- },
- scaleUpStabilizationWindowSeconds: 60 * 5,
- },
- {
- name: "no scale down stabilization",
- key: "",
- recommendations: []timestampedRecommendation{
- {1, time.Now().Add(-2 * time.Minute)},
- {2, time.Now().Add(-1 * time.Minute)}},
- currentReplicas: 100, // to apply scaleDown delay we should have current > desired
- prenormalizedDesiredReplicas: 3,
- expectedStabilizedReplicas: 3,
- expectedRecommendations: []timestampedRecommendation{
- {1, time.Now()},
- {2, time.Now()},
- {3, time.Now()},
- },
- scaleUpStabilizationWindowSeconds: 60 * 5,
- },
- {
- name: "no scale up stabilization",
- key: "",
- recommendations: []timestampedRecommendation{
- {4, time.Now().Add(-2 * time.Minute)},
- {5, time.Now().Add(-1 * time.Minute)}},
- currentReplicas: 1, // to apply scaleDown delay we should have current > desired
- prenormalizedDesiredReplicas: 3,
- expectedStabilizedReplicas: 3,
- expectedRecommendations: []timestampedRecommendation{
- {4, time.Now()},
- {5, time.Now()},
- {3, time.Now()},
- },
- scaleDownStabilizationWindowSeconds: 60 * 5,
- },
- {
- name: "no scale down stabilization, reuse recommendation element",
- key: "",
- recommendations: []timestampedRecommendation{
- {10, time.Now().Add(-10 * time.Minute)},
- {9, time.Now().Add(-9 * time.Minute)}},
- currentReplicas: 100, // to apply scaleDown delay we should have current > desired
- prenormalizedDesiredReplicas: 3,
- expectedStabilizedReplicas: 3,
- expectedRecommendations: []timestampedRecommendation{
- {10, time.Now()},
- {3, time.Now()},
- },
- },
- {
- name: "no scale up stabilization, reuse recommendation element",
- key: "",
- recommendations: []timestampedRecommendation{
- {10, time.Now().Add(-10 * time.Minute)},
- {9, time.Now().Add(-9 * time.Minute)}},
- currentReplicas: 1,
- prenormalizedDesiredReplicas: 100,
- expectedStabilizedReplicas: 100,
- expectedRecommendations: []timestampedRecommendation{
- {10, time.Now()},
- {100, time.Now()},
- },
- },
- {
- name: "scale down stabilization, reuse one of obsolete recommendation element",
- key: "",
- recommendations: []timestampedRecommendation{
- {10, time.Now().Add(-10 * time.Minute)},
- {4, time.Now().Add(-1 * time.Minute)},
- {5, time.Now().Add(-2 * time.Minute)},
- {9, time.Now().Add(-9 * time.Minute)}},
- currentReplicas: 100,
- prenormalizedDesiredReplicas: 3,
- expectedStabilizedReplicas: 5,
- expectedRecommendations: []timestampedRecommendation{
- {10, time.Now()},
- {4, time.Now()},
- {5, time.Now()},
- {3, time.Now()},
- },
- scaleDownStabilizationWindowSeconds: 3 * 60,
- },
- {
- // we can reuse only the first recommendation element
- // as the scale up delay = 150 (set in test), scale down delay = 300 (by default)
- // hence, only the first recommendation is obsolete for both scale up and scale down
- name: "scale up stabilization, reuse one of obsolete recommendation element",
- key: "",
- recommendations: []timestampedRecommendation{
- {10, time.Now().Add(-100 * time.Minute)},
- {6, time.Now().Add(-1 * time.Minute)},
- {5, time.Now().Add(-2 * time.Minute)},
- {9, time.Now().Add(-3 * time.Minute)}},
- currentReplicas: 1,
- prenormalizedDesiredReplicas: 100,
- expectedStabilizedReplicas: 5,
- expectedRecommendations: []timestampedRecommendation{
- {100, time.Now()},
- {6, time.Now()},
- {5, time.Now()},
- {9, time.Now()},
- },
- scaleUpStabilizationWindowSeconds: 300,
- },
- }
- for _, tc := range tests {
- t.Run(tc.name, func(t *testing.T) {
- hc := HorizontalController{
- recommendations: map[string][]timestampedRecommendation{
- tc.key: tc.recommendations,
- },
- }
- arg := NormalizationArg{
- Key: tc.key,
- DesiredReplicas: tc.prenormalizedDesiredReplicas,
- CurrentReplicas: tc.currentReplicas,
- ScaleUpBehavior: &autoscalingv2.HPAScalingRules{
- StabilizationWindowSeconds: &tc.scaleUpStabilizationWindowSeconds,
- },
- ScaleDownBehavior: &autoscalingv2.HPAScalingRules{
- StabilizationWindowSeconds: &tc.scaleDownStabilizationWindowSeconds,
- },
- }
- r, _, _ := hc.stabilizeRecommendationWithBehaviors(arg)
- assert.Equal(t, tc.expectedStabilizedReplicas, r, "expected replicas do not match")
- if !assert.Len(t, hc.recommendations[tc.key], len(tc.expectedRecommendations), "stored recommendations differ in length") {
- return
- }
- for i, r := range hc.recommendations[tc.key] {
- expectedRecommendation := tc.expectedRecommendations[i]
- assert.Equal(t, expectedRecommendation.recommendation, r.recommendation, "stored recommendation differs at position %d", i)
- }
- })
- }
- }
- func TestScaleUpOneMetricEmpty(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 3,
- statusReplicas: 3,
- expectedDesiredReplicas: 4,
- CPUTarget: 30,
- verifyCPUCurrent: true,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.ExternalMetricSourceType,
- External: &autoscalingv2.ExternalMetricSource{
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- Selector: &metav1.LabelSelector{},
- },
- Target: autoscalingv2.MetricTarget{
- Value: resource.NewMilliQuantity(100, resource.DecimalSI),
- },
- },
- },
- },
- reportedLevels: []uint64{300, 400, 500},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- }
- _, _, _, testEMClient, _ := tc.prepareTestClient(t)
- testEMClient.PrependReactor("list", "*", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- return true, &emapi.ExternalMetricValueList{}, fmt.Errorf("something went wrong")
- })
- tc.testEMClient = testEMClient
- tc.runTest(t)
- }
- func TestNoScaleDownOneMetricInvalid(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 5,
- statusReplicas: 5,
- expectedDesiredReplicas: 5,
- CPUTarget: 50,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: "CheddarCheese",
- },
- },
- reportedLevels: []uint64{100, 300, 500, 250, 250},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- useMetricsAPI: true,
- expectedConditions: []autoscalingv1.HorizontalPodAutoscalerCondition{
- {Type: autoscalingv1.AbleToScale, Status: v1.ConditionTrue, Reason: "ScaleDownStabilized"},
- {Type: autoscalingv1.ScalingActive, Status: v1.ConditionTrue, Reason: "ValidMetricFound"},
- {Type: autoscalingv1.ScalingLimited, Status: v1.ConditionFalse, Reason: "DesiredWithinRange"},
- },
- }
- tc.runTest(t)
- }
- func TestNoScaleDownOneMetricEmpty(t *testing.T) {
- tc := testCase{
- minReplicas: 2,
- maxReplicas: 6,
- specReplicas: 5,
- statusReplicas: 5,
- expectedDesiredReplicas: 5,
- CPUTarget: 50,
- metricsTarget: []autoscalingv2.MetricSpec{
- {
- Type: autoscalingv2.ExternalMetricSourceType,
- External: &autoscalingv2.ExternalMetricSource{
- Metric: autoscalingv2.MetricIdentifier{
- Name: "qps",
- Selector: &metav1.LabelSelector{},
- },
- Target: autoscalingv2.MetricTarget{
- Value: resource.NewMilliQuantity(1000, resource.DecimalSI),
- },
- },
- },
- },
- reportedLevels: []uint64{100, 300, 500, 250, 250},
- reportedCPURequests: []resource.Quantity{resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0"), resource.MustParse("1.0")},
- useMetricsAPI: true,
- expectedConditions: []autoscalingv1.HorizontalPodAutoscalerCondition{
- {Type: autoscalingv1.AbleToScale, Status: v1.ConditionTrue, Reason: "ScaleDownStabilized"},
- {Type: autoscalingv1.ScalingActive, Status: v1.ConditionTrue, Reason: "ValidMetricFound"},
- {Type: autoscalingv1.ScalingLimited, Status: v1.ConditionFalse, Reason: "DesiredWithinRange"},
- },
- }
- _, _, _, testEMClient, _ := tc.prepareTestClient(t)
- testEMClient.PrependReactor("list", "*", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- return true, &emapi.ExternalMetricValueList{}, fmt.Errorf("something went wrong")
- })
- tc.testEMClient = testEMClient
- tc.runTest(t)
- }
|