Cloudpods/backend/pkg/cloudcommon/db/quotas/quotakeys_test.go

// Copyright 2019 Yunion
//
// 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 quotas

import (
	"testing"
)

func TestRelation(t *testing.T) {
	keys := []SZonalCloudResourceKeys{
		//Top :=
		{},
		// Domain1 :=
		{
			SRegionalCloudResourceKeys: SRegionalCloudResourceKeys{
				SCloudResourceKeys: SCloudResourceKeys{
					SBaseProjectQuotaKeys: SBaseProjectQuotaKeys{
						SBaseDomainQuotaKeys: SBaseDomainQuotaKeys{
							DomainId: "domain1",
						},
					},
				},
			},
		},
		// Domain2 :=
		{
			SRegionalCloudResourceKeys: SRegionalCloudResourceKeys{
				SCloudResourceKeys: SCloudResourceKeys{
					SBaseProjectQuotaKeys: SBaseProjectQuotaKeys{
						SBaseDomainQuotaKeys: SBaseDomainQuotaKeys{
							DomainId: "domain2",
						},
					},
				},
			},
		},
		// Project11 :=
		{
			SRegionalCloudResourceKeys: SRegionalCloudResourceKeys{
				SCloudResourceKeys: SCloudResourceKeys{
					SBaseProjectQuotaKeys: SBaseProjectQuotaKeys{
						SBaseDomainQuotaKeys: SBaseDomainQuotaKeys{
							DomainId: "domain1",
						},
						ProjectId: "project1",
					},
				},
			},
		},
		// Project12 :=
		{
			SRegionalCloudResourceKeys: SRegionalCloudResourceKeys{
				SCloudResourceKeys: SCloudResourceKeys{
					SBaseProjectQuotaKeys: SBaseProjectQuotaKeys{
						SBaseDomainQuotaKeys: SBaseDomainQuotaKeys{
							DomainId: "domain1",
						},
						ProjectId: "project2",
					},
				},
			},
		},
		// Project21 :=
		{
			SRegionalCloudResourceKeys: SRegionalCloudResourceKeys{
				SCloudResourceKeys: SCloudResourceKeys{
					SBaseProjectQuotaKeys: SBaseProjectQuotaKeys{
						SBaseDomainQuotaKeys: SBaseDomainQuotaKeys{
							DomainId: "domain2",
						},
						ProjectId: "project1",
					},
				},
			},
		},
		// Project11Region1 :=
		{
			SRegionalCloudResourceKeys: SRegionalCloudResourceKeys{
				SCloudResourceKeys: SCloudResourceKeys{
					SBaseProjectQuotaKeys: SBaseProjectQuotaKeys{
						SBaseDomainQuotaKeys: SBaseDomainQuotaKeys{
							DomainId: "domain1",
						},
						ProjectId: "project1",
					},
				},
				SRegionalBaseKeys: SRegionalBaseKeys{
					RegionId: "region1",
				},
			},
		},
		// Project11Region2 :=
		{
			SRegionalCloudResourceKeys: SRegionalCloudResourceKeys{
				SCloudResourceKeys: SCloudResourceKeys{
					SBaseProjectQuotaKeys: SBaseProjectQuotaKeys{
						SBaseDomainQuotaKeys: SBaseDomainQuotaKeys{
							DomainId: "domain1",
						},
						ProjectId: "project1",
					},
				},
				SRegionalBaseKeys: SRegionalBaseKeys{
					RegionId: "region2",
				},
			},
		},
		// Project11Aliyun :=
		{
			SRegionalCloudResourceKeys: SRegionalCloudResourceKeys{
				SCloudResourceKeys: SCloudResourceKeys{
					SBaseProjectQuotaKeys: SBaseProjectQuotaKeys{
						SBaseDomainQuotaKeys: SBaseDomainQuotaKeys{
							DomainId: "domain1",
						},
						ProjectId: "project1",
					},
					SCloudResourceBaseKeys: SCloudResourceBaseKeys{
						Provider: "Aliyun",
					},
				},
			},
		},
		// Project11AliyunRegion1 :=
		{
			SRegionalCloudResourceKeys: SRegionalCloudResourceKeys{
				SCloudResourceKeys: SCloudResourceKeys{
					SBaseProjectQuotaKeys: SBaseProjectQuotaKeys{
						SBaseDomainQuotaKeys: SBaseDomainQuotaKeys{
							DomainId: "domain1",
						},
						ProjectId: "project1",
					},
					SCloudResourceBaseKeys: SCloudResourceBaseKeys{
						Provider: "Aliyun",
					},
				},
				SRegionalBaseKeys: SRegionalBaseKeys{
					RegionId: "region1",
				},
			},
		},
	}
	want := [][]TQuotaKeysRelation{
		{
			QuotaKeysEqual,
			QuotaKeysContain,
			QuotaKeysContain,
			QuotaKeysContain,
			QuotaKeysContain,
			QuotaKeysContain,
			QuotaKeysContain,
			QuotaKeysContain,
			QuotaKeysContain,
			QuotaKeysContain,
		},
		{
			QuotaKeysBelong,
			QuotaKeysEqual,
			QuotaKeysExclude,
			QuotaKeysContain,
			QuotaKeysContain,
			QuotaKeysExclude,
			QuotaKeysContain,
			QuotaKeysContain,
			QuotaKeysContain,
			QuotaKeysContain,
		},
		{
			QuotaKeysBelong,
			QuotaKeysExclude,
			QuotaKeysEqual,
			QuotaKeysExclude,
			QuotaKeysExclude,
			QuotaKeysContain,
			QuotaKeysExclude,
			QuotaKeysExclude,
			QuotaKeysExclude,
			QuotaKeysExclude,
		},
		{
			QuotaKeysBelong,
			QuotaKeysBelong,
			QuotaKeysExclude,
			QuotaKeysEqual,
			QuotaKeysExclude,
			QuotaKeysExclude,
			QuotaKeysContain,
			QuotaKeysContain,
			QuotaKeysContain,
			QuotaKeysContain,
		},
		{
			QuotaKeysBelong,
			QuotaKeysBelong,
			QuotaKeysExclude,
			QuotaKeysExclude,
			QuotaKeysEqual,
			QuotaKeysExclude,
			QuotaKeysExclude,
			QuotaKeysExclude,
			QuotaKeysExclude,
			QuotaKeysExclude,
		},
		{
			QuotaKeysBelong,
			QuotaKeysExclude,
			QuotaKeysBelong,
			QuotaKeysExclude,
			QuotaKeysExclude,
			QuotaKeysEqual,
			QuotaKeysExclude,
			QuotaKeysExclude,
			QuotaKeysExclude,
			QuotaKeysExclude,
		},
		{
			QuotaKeysBelong,
			QuotaKeysBelong,
			QuotaKeysExclude,
			QuotaKeysBelong,
			QuotaKeysExclude,
			QuotaKeysExclude,
			QuotaKeysEqual,
			QuotaKeysExclude,
			QuotaKeysExclude,
			QuotaKeysContain,
		},
		{
			QuotaKeysBelong,
			QuotaKeysBelong,
			QuotaKeysExclude,
			QuotaKeysBelong,
			QuotaKeysExclude,
			QuotaKeysExclude,
			QuotaKeysExclude,
			QuotaKeysEqual,
			QuotaKeysExclude,
			QuotaKeysExclude,
		},
		{
			QuotaKeysBelong,
			QuotaKeysBelong,
			QuotaKeysExclude,
			QuotaKeysBelong,
			QuotaKeysExclude,
			QuotaKeysExclude,
			QuotaKeysExclude,
			QuotaKeysExclude,
			QuotaKeysEqual,
			QuotaKeysContain,
		},
		{
			QuotaKeysBelong,
			QuotaKeysBelong,
			QuotaKeysExclude,
			QuotaKeysBelong,
			QuotaKeysExclude,
			QuotaKeysExclude,
			QuotaKeysBelong,
			QuotaKeysExclude,
			QuotaKeysBelong,
			QuotaKeysEqual,
		},
	}
	for i := range keys {
		for j := range keys {
			rel := relation(keys[i], keys[j])
			if rel != want[i][j] {
				t.Errorf("i=%d j=%d %#v %#v got %s want %s", i, j, keys[i], keys[j], rel, want[i][j])
			}
		}
	}
}