add unit test

This commit is contained in:
Qingqing Zheng 2019-12-13 14:21:24 -08:00
parent 4152f38d54
commit 8953544b15

View File

@ -59,6 +59,13 @@ func TestParseConfig(t *testing.T) {
"CloudProviderRateLimitBucketWrite": 1, "CloudProviderRateLimitBucketWrite": 1,
"cloudProviderRateLimitQPS": 1, "cloudProviderRateLimitQPS": 1,
"CloudProviderRateLimitQPSWrite": 1, "CloudProviderRateLimitQPSWrite": 1,
"availabilitySetNodesCacheTTL": 100,
"vmssCacheTTL": 100,
"vmssVirtualMachinesCacheTTL": 100,
"vmCacheTTL": 100,
"lbCacheTTL": 100,
"nsgCacheTTL": 100,
"rtCacheTTL": 100,
"location": "location", "location": "location",
"maximumLoadBalancerRuleCount": 1, "maximumLoadBalancerRuleCount": 1,
"primaryAvailabilitySetName": "primaryAvailabilitySetName", "primaryAvailabilitySetName": "primaryAvailabilitySetName",
@ -97,6 +104,13 @@ func TestParseConfig(t *testing.T) {
CloudProviderRateLimitBucketWrite: 1, CloudProviderRateLimitBucketWrite: 1,
CloudProviderRateLimitQPS: 1, CloudProviderRateLimitQPS: 1,
CloudProviderRateLimitQPSWrite: 1, CloudProviderRateLimitQPSWrite: 1,
AvailabilitySetNodesCacheTTL: 100,
VmssCacheTTL: 100,
VmssVirtualMachinesCacheTTL: 100,
VMCacheTTL: 100,
LbCacheTTL: 100,
NsgCacheTTL: 100,
RtCacheTTL: 100,
Location: "location", Location: "location",
MaximumLoadBalancerRuleCount: 1, MaximumLoadBalancerRuleCount: 1,
PrimaryAvailabilitySetName: "primaryAvailabilitySetName", PrimaryAvailabilitySetName: "primaryAvailabilitySetName",
@ -1572,7 +1586,14 @@ func TestNewCloudFromJSON(t *testing.T) {
"cloudProviderBackoff": true, "cloudProviderBackoff": true,
"cloudProviderRatelimit": true, "cloudProviderRatelimit": true,
"cloudProviderRateLimitQPS": 0.5, "cloudProviderRateLimitQPS": 0.5,
"cloudProviderRateLimitBucket": 5 "cloudProviderRateLimitBucket": 5,
"availabilitySetNodesCacheTTL": 100,
"vmssCacheTTL": 100,
"vmssVirtualMachinesCacheTTL": 100,
"vmCacheTTL": 100,
"lbCacheTTL": 100,
"nsgCacheTTL": 100,
"rtCacheTTL": 100,
}` }`
validateConfig(t, config) validateConfig(t, config)
} }
@ -1622,6 +1643,13 @@ cloudProviderBackoffJitter: 1.0
cloudProviderRatelimit: true cloudProviderRatelimit: true
cloudProviderRateLimitQPS: 0.5 cloudProviderRateLimitQPS: 0.5
cloudProviderRateLimitBucket: 5 cloudProviderRateLimitBucket: 5
availabilitySetNodesCacheTTL: 100
vmssCacheTTL: 100
vmssVirtualMachinesCacheTTL: 100
vmCacheTTL: 100
lbCacheTTL: 100
nsgCacheTTL: 100
rtCacheTTL: 100
` `
validateConfig(t, config) validateConfig(t, config)
} }
@ -1695,6 +1723,27 @@ func validateConfig(t *testing.T, config string) {
if azureCloud.CloudProviderRateLimitBucket != 5 { if azureCloud.CloudProviderRateLimitBucket != 5 {
t.Errorf("got incorrect value for CloudProviderRateLimitBucket") t.Errorf("got incorrect value for CloudProviderRateLimitBucket")
} }
if azureCloud.AvailabilitySetNodesCacheTTL != 100 {
t.Errorf("got incorrect value for availabilitySetNodesCacheTTL")
}
if azureCloud.VmssCacheTTL != 100 {
t.Errorf("got incorrect value for vmssCacheTTL")
}
if azureCloud.VmssVirtualMachinesCacheTTL != 100 {
t.Errorf("got incorrect value for vmssVirtualMachinesCacheTTL")
}
if azureCloud.VMCacheTTL != 100 {
t.Errorf("got incorrect value for vmCacheTTL")
}
if azureCloud.LbCacheTTL != 100 {
t.Errorf("got incorrect value for lbCacheTTL")
}
if azureCloud.NsgCacheTTL != 100 {
t.Errorf("got incorrect value for nsgCacheTTL")
}
if azureCloud.RtCacheTTL != 100 {
t.Errorf("got incorrect value for rtCacheTTL")
}
} }
func getCloudFromConfig(t *testing.T, config string) *Cloud { func getCloudFromConfig(t *testing.T, config string) *Cloud {