Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions .gitleaksignore
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ api/internal/server/manager_test.go:generic-api-key:767
inventory/internal/store/host_test.go:generic-api-key:996
inventory/internal/store/host_test.go:generic-api-key:1010
inventory/internal/store/host_test.go:generic-api-key:1050
inventory/internal/store/host_test.go:generic-api-key:1051
apiv2/pkg/api/v2/edge-infrastructure-manager-openapi-server.gen.go:generic-api-key:2555
apiv2/pkg/api/v2/edge-infrastructure-manager-openapi-server.gen.go:generic-api-key:2739
apiv2/pkg/api/v2/edge-infrastructure-manager-openapi-server.gen.go:generic-api-key:2707
8 changes: 6 additions & 2 deletions inventory/internal/store/host.go
Original file line number Diff line number Diff line change
Expand Up @@ -173,7 +173,9 @@ func getHostQuery(ctx context.Context, tx *ent.Tx, tenantID, resourceID string,
WithHostGpus()
if nestedLoad {
query.WithInstance(func(query *ent.InstanceResourceQuery) {
query.WithDesiredOs().WithCurrentOs()
query.WithDesiredOs().
WithCurrentOs().
WithOs()
})
} else {
query.WithInstance()
Expand Down Expand Up @@ -470,7 +472,9 @@ func filterHosts(ctx context.Context, client *ent.Client, filter *inv_v1.Resourc
WithHostUsbs().
WithHostGpus().
WithInstance(func(query *ent.InstanceResourceQuery) {
query.WithDesiredOs().WithCurrentOs()
query.WithDesiredOs().
WithCurrentOs().
WithOs()
}).
Where(pred).
Order(orderOpts...)
Expand Down
11 changes: 11 additions & 0 deletions inventory/internal/store/host_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -377,6 +377,7 @@ func Test_Host_BackReferences_Read(t *testing.T) {
instance := inv_testing.CreateInstance(t, host, os)
instance.DesiredOs = os
instance.CurrentOs = os
instance.Os = os
host.Instance = instance
gpu := inv_testing.CreatHostGPU(t, host)
host.HostStorages = append(host.HostStorages, storage)
Expand Down Expand Up @@ -1725,6 +1726,7 @@ func Test_FilterHosts(t *testing.T) {
instance1 := inv_testing.CreateInstance(t, &expHost1, os1)
instance1.DesiredOs = os1
instance1.CurrentOs = os1
instance1.Os = os1
expHost1.Instance = instance1

hostStorage1 := inv_testing.CreateHostStorage(t, &expHost1)
Expand Down Expand Up @@ -2629,6 +2631,7 @@ func Test_NestedFilterHost(t *testing.T) {

instance.DesiredOs = os
instance.CurrentOs = os
instance.Os = os

hostGpu1 := inv_testing.CreatHostGPU(t, host1)
host1.Site = site1
Expand Down Expand Up @@ -2682,6 +2685,14 @@ func Test_NestedFilterHost(t *testing.T) {
resources: []*computev1.HostResource{host1},
valid: true,
},
"FilterByOsID2": {
in: &inv_v1.ResourceFilter{
Filter: fmt.Sprintf(`%s.%s.%s = %q`, hostresource.EdgeInstance, instanceresource.EdgeOs,
operatingsystemresource.FieldResourceID, os.GetResourceId()),
},
resources: []*computev1.HostResource{host1},
valid: true,
},
"FilterByHasHostStorages": {
in: &inv_v1.ResourceFilter{
Filter: fmt.Sprintf(`has(%s)`, hostresource.EdgeHostStorages),
Expand Down
6 changes: 6 additions & 0 deletions inventory/internal/store/instance_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -1324,6 +1324,7 @@ func Test_NestedFilterInstances(t *testing.T) {
instance1.Host = host1
instance1.DesiredOs = os1
instance1.CurrentOs = os1
instance1.Os = os1

instance2 := inv_testing.CreateInstance(t, host2, os1)
workload2 := inv_testing.CreateWorkload(t)
Expand All @@ -1336,21 +1337,25 @@ func Test_NestedFilterInstances(t *testing.T) {
instance4.Host = host5
instance4.DesiredOs = os2
instance4.CurrentOs = os2
instance4.Os = os2
instance2.WorkloadMembers = append(instance2.WorkloadMembers, workloadMember2)
instance2.Host = host2
instance2.DesiredOs = os1
instance2.CurrentOs = os1
instance2.Os = os1

instance3 := inv_testing.CreateInstance(t, host3, os2)
instance3.Host = host3
instance3.DesiredOs = os2
instance3.CurrentOs = os2
instance3.Os = os2

provider := inv_testing.CreateProvider(t, "Test Provider1")
instanceWithProvider := inv_testing.CreateInstanceWithProvider(t, host4, os2, provider)
instanceWithProvider.Host = host4
instanceWithProvider.DesiredOs = os2
instanceWithProvider.CurrentOs = os2
instanceWithProvider.Os = os2
instanceWithProvider.Provider = provider

localaccount := inv_testing.CreateLocalAccount(t,
Expand All @@ -1361,6 +1366,7 @@ func Test_NestedFilterInstances(t *testing.T) {
instanceWithLocalAccount.Host = host6
instanceWithLocalAccount.DesiredOs = os2
instanceWithLocalAccount.CurrentOs = os2
instanceWithLocalAccount.Os = os2
instanceWithLocalAccount.Localaccount = localaccount

testcases := map[string]struct {
Expand Down
2 changes: 2 additions & 0 deletions inventory/pkg/testing/testing_utils.go
Original file line number Diff line number Diff line change
Expand Up @@ -1047,6 +1047,7 @@ func (c *InvResourceDAO) createInstanceWithOpts(
DesiredState: computev1.InstanceState_INSTANCE_STATE_RUNNING,
DesiredOs: osRes,
CurrentOs: osRes, // always create with desired OS == current OS for testing
Os: osRes,
Host: hostRes,
SecurityFeature: osv1.SecurityFeature_SECURITY_FEATURE_UNSPECIFIED,
TenantId: tenantID,
Expand All @@ -1073,6 +1074,7 @@ func (c *InvResourceDAO) createInstanceWithOpts(
// embedded objects for edges.
instResp.DesiredOs = nil
instResp.CurrentOs = nil
instResp.Os = nil
instResp.Host = nil
instResp.WorkloadMembers = nil
instResp.Provider = nil
Expand Down