Commit 47f7672e authored by Cheng Xing's avatar Cheng Xing

Generated code for gce_disks refactor

parent d33c1e3b
...@@ -312,29 +312,18 @@ func TestDisksGroup(t *testing.T) { ...@@ -312,29 +312,18 @@ func TestDisksGroup(t *testing.T) {
mock := NewMockGCE(pr) mock := NewMockGCE(pr)
var key *meta.Key var key *meta.Key
keyAlpha := meta.ZonalKey("key-alpha", "location")
key = keyAlpha
keyGA := meta.ZonalKey("key-ga", "location") keyGA := meta.ZonalKey("key-ga", "location")
key = keyGA key = keyGA
// Ignore unused variables. // Ignore unused variables.
_, _, _ = ctx, mock, key _, _, _ = ctx, mock, key
// Get not found. // Get not found.
if _, err := mock.AlphaDisks().Get(ctx, key); err == nil {
t.Errorf("AlphaDisks().Get(%v, %v) = _, nil; want error", ctx, key)
}
if _, err := mock.Disks().Get(ctx, key); err == nil { if _, err := mock.Disks().Get(ctx, key); err == nil {
t.Errorf("Disks().Get(%v, %v) = _, nil; want error", ctx, key) t.Errorf("Disks().Get(%v, %v) = _, nil; want error", ctx, key)
} }
// Insert. // Insert.
{ {
obj := &alpha.Disk{}
if err := mock.AlphaDisks().Insert(ctx, keyAlpha, obj); err != nil {
t.Errorf("AlphaDisks().Insert(%v, %v, %v) = %v; want nil", ctx, keyAlpha, obj, err)
}
}
{
obj := &ga.Disk{} obj := &ga.Disk{}
if err := mock.Disks().Insert(ctx, keyGA, obj); err != nil { if err := mock.Disks().Insert(ctx, keyGA, obj); err != nil {
t.Errorf("Disks().Insert(%v, %v, %v) = %v; want nil", ctx, keyGA, obj, err) t.Errorf("Disks().Insert(%v, %v, %v) = %v; want nil", ctx, keyGA, obj, err)
...@@ -342,36 +331,17 @@ func TestDisksGroup(t *testing.T) { ...@@ -342,36 +331,17 @@ func TestDisksGroup(t *testing.T) {
} }
// Get across versions. // Get across versions.
if obj, err := mock.AlphaDisks().Get(ctx, key); err != nil {
t.Errorf("AlphaDisks().Get(%v, %v) = %v, %v; want nil", ctx, key, obj, err)
}
if obj, err := mock.Disks().Get(ctx, key); err != nil { if obj, err := mock.Disks().Get(ctx, key); err != nil {
t.Errorf("Disks().Get(%v, %v) = %v, %v; want nil", ctx, key, obj, err) t.Errorf("Disks().Get(%v, %v) = %v, %v; want nil", ctx, key, obj, err)
} }
// List. // List.
mock.MockAlphaDisks.Objects[*keyAlpha] = mock.MockAlphaDisks.Obj(&alpha.Disk{Name: keyAlpha.Name})
mock.MockDisks.Objects[*keyGA] = mock.MockDisks.Obj(&ga.Disk{Name: keyGA.Name}) mock.MockDisks.Objects[*keyGA] = mock.MockDisks.Obj(&ga.Disk{Name: keyGA.Name})
want := map[string]bool{ want := map[string]bool{
"key-alpha": true, "key-ga": true,
"key-ga": true,
} }
_ = want // ignore unused variables. _ = want // ignore unused variables.
{ {
objs, err := mock.AlphaDisks().List(ctx, location, filter.None)
if err != nil {
t.Errorf("AlphaDisks().List(%v, %v, %v) = %v, %v; want _, nil", ctx, location, filter.None, objs, err)
} else {
got := map[string]bool{}
for _, obj := range objs {
got[obj.Name] = true
}
if !reflect.DeepEqual(got, want) {
t.Errorf("AlphaDisks().List(); got %+v, want %+v", got, want)
}
}
}
{
objs, err := mock.Disks().List(ctx, location, filter.None) objs, err := mock.Disks().List(ctx, location, filter.None)
if err != nil { if err != nil {
t.Errorf("Disks().List(%v, %v, %v) = %v, %v; want _, nil", ctx, location, filter.None, objs, err) t.Errorf("Disks().List(%v, %v, %v) = %v, %v; want _, nil", ctx, location, filter.None, objs, err)
...@@ -387,17 +357,11 @@ func TestDisksGroup(t *testing.T) { ...@@ -387,17 +357,11 @@ func TestDisksGroup(t *testing.T) {
} }
// Delete across versions. // Delete across versions.
if err := mock.AlphaDisks().Delete(ctx, keyAlpha); err != nil {
t.Errorf("AlphaDisks().Delete(%v, %v) = %v; want nil", ctx, keyAlpha, err)
}
if err := mock.Disks().Delete(ctx, keyGA); err != nil { if err := mock.Disks().Delete(ctx, keyGA); err != nil {
t.Errorf("Disks().Delete(%v, %v) = %v; want nil", ctx, keyGA, err) t.Errorf("Disks().Delete(%v, %v) = %v; want nil", ctx, keyGA, err)
} }
// Delete not found. // Delete not found.
if err := mock.AlphaDisks().Delete(ctx, keyAlpha); err == nil {
t.Errorf("AlphaDisks().Delete(%v, %v) = nil; want error", ctx, keyAlpha)
}
if err := mock.Disks().Delete(ctx, keyGA); err == nil { if err := mock.Disks().Delete(ctx, keyGA); err == nil {
t.Errorf("Disks().Delete(%v, %v) = nil; want error", ctx, keyGA) t.Errorf("Disks().Delete(%v, %v) = nil; want error", ctx, keyGA)
} }
...@@ -1315,39 +1279,39 @@ func TestRegionDisksGroup(t *testing.T) { ...@@ -1315,39 +1279,39 @@ func TestRegionDisksGroup(t *testing.T) {
mock := NewMockGCE(pr) mock := NewMockGCE(pr)
var key *meta.Key var key *meta.Key
keyAlpha := meta.RegionalKey("key-alpha", "location") keyBeta := meta.RegionalKey("key-beta", "location")
key = keyAlpha key = keyBeta
// Ignore unused variables. // Ignore unused variables.
_, _, _ = ctx, mock, key _, _, _ = ctx, mock, key
// Get not found. // Get not found.
if _, err := mock.AlphaRegionDisks().Get(ctx, key); err == nil { if _, err := mock.BetaRegionDisks().Get(ctx, key); err == nil {
t.Errorf("AlphaRegionDisks().Get(%v, %v) = _, nil; want error", ctx, key) t.Errorf("BetaRegionDisks().Get(%v, %v) = _, nil; want error", ctx, key)
} }
// Insert. // Insert.
{ {
obj := &alpha.Disk{} obj := &beta.Disk{}
if err := mock.AlphaRegionDisks().Insert(ctx, keyAlpha, obj); err != nil { if err := mock.BetaRegionDisks().Insert(ctx, keyBeta, obj); err != nil {
t.Errorf("AlphaRegionDisks().Insert(%v, %v, %v) = %v; want nil", ctx, keyAlpha, obj, err) t.Errorf("BetaRegionDisks().Insert(%v, %v, %v) = %v; want nil", ctx, keyBeta, obj, err)
} }
} }
// Get across versions. // Get across versions.
if obj, err := mock.AlphaRegionDisks().Get(ctx, key); err != nil { if obj, err := mock.BetaRegionDisks().Get(ctx, key); err != nil {
t.Errorf("AlphaRegionDisks().Get(%v, %v) = %v, %v; want nil", ctx, key, obj, err) t.Errorf("BetaRegionDisks().Get(%v, %v) = %v, %v; want nil", ctx, key, obj, err)
} }
// List. // List.
mock.MockAlphaRegionDisks.Objects[*keyAlpha] = mock.MockAlphaRegionDisks.Obj(&alpha.Disk{Name: keyAlpha.Name}) mock.MockBetaRegionDisks.Objects[*keyBeta] = mock.MockBetaRegionDisks.Obj(&beta.Disk{Name: keyBeta.Name})
want := map[string]bool{ want := map[string]bool{
"key-alpha": true, "key-beta": true,
} }
_ = want // ignore unused variables. _ = want // ignore unused variables.
{ {
objs, err := mock.AlphaRegionDisks().List(ctx, location, filter.None) objs, err := mock.BetaRegionDisks().List(ctx, location, filter.None)
if err != nil { if err != nil {
t.Errorf("AlphaRegionDisks().List(%v, %v, %v) = %v, %v; want _, nil", ctx, location, filter.None, objs, err) t.Errorf("BetaRegionDisks().List(%v, %v, %v) = %v, %v; want _, nil", ctx, location, filter.None, objs, err)
} else { } else {
got := map[string]bool{} got := map[string]bool{}
for _, obj := range objs { for _, obj := range objs {
...@@ -1360,13 +1324,13 @@ func TestRegionDisksGroup(t *testing.T) { ...@@ -1360,13 +1324,13 @@ func TestRegionDisksGroup(t *testing.T) {
} }
// Delete across versions. // Delete across versions.
if err := mock.AlphaRegionDisks().Delete(ctx, keyAlpha); err != nil { if err := mock.BetaRegionDisks().Delete(ctx, keyBeta); err != nil {
t.Errorf("AlphaRegionDisks().Delete(%v, %v) = %v; want nil", ctx, keyAlpha, err) t.Errorf("BetaRegionDisks().Delete(%v, %v) = %v; want nil", ctx, keyBeta, err)
} }
// Delete not found. // Delete not found.
if err := mock.AlphaRegionDisks().Delete(ctx, keyAlpha); err == nil { if err := mock.BetaRegionDisks().Delete(ctx, keyBeta); err == nil {
t.Errorf("AlphaRegionDisks().Delete(%v, %v) = nil; want error", ctx, keyAlpha) t.Errorf("BetaRegionDisks().Delete(%v, %v) = nil; want error", ctx, keyBeta)
} }
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment