|
| 1 | +package templateUtils |
| 2 | + |
| 3 | +import ( |
| 4 | + "testing" |
| 5 | + |
| 6 | + "github.com/stretchr/testify/assert" |
| 7 | +) |
| 8 | + |
| 9 | +func TestHasExtension(t *testing.T) { |
| 10 | + type V1 struct { |
| 11 | + Field1 string |
| 12 | + Field2 string |
| 13 | + } |
| 14 | + type V2 struct { |
| 15 | + Field3 string |
| 16 | + Field4 string |
| 17 | + } |
| 18 | + |
| 19 | + type Base struct { |
| 20 | + Base string |
| 21 | + Base1 []string |
| 22 | + Base2 struct{ a, b, v string } |
| 23 | + } |
| 24 | + |
| 25 | + type ExtendedV1 struct { |
| 26 | + Base |
| 27 | + V1 |
| 28 | + } |
| 29 | + |
| 30 | + type ExtendedV1_ struct { |
| 31 | + *Base |
| 32 | + Extension *V1 |
| 33 | + } |
| 34 | + |
| 35 | + type ExtendedV2 struct { |
| 36 | + Base |
| 37 | + V1 |
| 38 | + V2 |
| 39 | + } |
| 40 | + |
| 41 | + assert.True(t, HasExtension(Base{}, "Base")) |
| 42 | + assert.False(t, HasExtension(Base{}, "V1")) |
| 43 | + |
| 44 | + l1 := &ExtendedV1{} |
| 45 | + l2 := &l1 |
| 46 | + l3 := &l2 |
| 47 | + l4 := &l3 |
| 48 | + |
| 49 | + assert.False(t, HasExtension(ExtendedV1{}, "V2")) |
| 50 | + assert.True(t, HasExtension(ExtendedV1{}, "V1")) |
| 51 | + assert.True(t, HasExtension(ExtendedV1{}, "Base")) |
| 52 | + assert.True(t, HasExtension(ExtendedV1{}, "Base")) |
| 53 | + assert.True(t, HasExtension(l4, "Base")) |
| 54 | + assert.True(t, HasExtension(l4, "V1")) |
| 55 | + assert.True(t, HasExtension(l4, "Field1")) |
| 56 | + assert.False(t, HasExtension(l4, "V2")) |
| 57 | + |
| 58 | + l5 := &ExtendedV2{Base: l1.Base, V1: l1.V1} |
| 59 | + |
| 60 | + assert.True(t, HasExtension(ExtendedV2{}, "V2")) |
| 61 | + assert.True(t, HasExtension(ExtendedV2{}, "V1")) |
| 62 | + assert.True(t, HasExtension(ExtendedV2{}, "Base")) |
| 63 | + assert.True(t, HasExtension(ExtendedV2{}, "Base")) |
| 64 | + assert.True(t, HasExtension(l5, "Base")) |
| 65 | + assert.True(t, HasExtension(l5, "V1")) |
| 66 | + assert.True(t, HasExtension(l5, "Field1")) |
| 67 | + assert.True(t, HasExtension(l5, "Field3")) |
| 68 | + assert.True(t, HasExtension(l5, "Field4")) |
| 69 | + assert.True(t, HasExtension(l5, "V2")) |
| 70 | + assert.True(t, HasExtension(any(l5), "V2")) |
| 71 | + |
| 72 | + assert.False(t, HasExtension(any(new(int)), "Base")) |
| 73 | + assert.False(t, HasExtension(any(nil), "Base")) |
| 74 | + assert.False(t, HasExtension((*ExtendedV2)(nil), "V2")) |
| 75 | + assert.False(t, HasExtension(nil, "V2")) |
| 76 | + |
| 77 | + assert.False(t, HasExtension(new(ExtendedV1_), "Base")) |
| 78 | + assert.False(t, HasExtension(new(ExtendedV1_), "Extension")) |
| 79 | + assert.False(t, HasExtension(&ExtendedV1_{Base: new(Base), Extension: nil}, "Extension")) |
| 80 | + assert.True(t, HasExtension(&ExtendedV1_{Base: new(Base), Extension: nil}, "Base")) |
| 81 | + |
| 82 | + tmp := new(ExtendedV1_) |
| 83 | + assert.False(t, HasExtension(tmp, "Extension")) |
| 84 | + assert.False(t, HasExtension(tmp, "Base")) |
| 85 | + assert.False(t, HasExtension(tmp.Extension, "Field1")) |
| 86 | + tmp.Extension = new(V1) |
| 87 | + assert.True(t, HasExtension(tmp, "Extension")) |
| 88 | + assert.True(t, HasExtension(tmp.Extension, "Field1")) |
| 89 | + tmp.Base = new(Base) |
| 90 | + assert.True(t, HasExtension(tmp, "Base")) |
| 91 | + assert.True(t, HasExtension(tmp.Base, "Base")) |
| 92 | + assert.True(t, HasExtension(tmp.Base, "Base1")) |
| 93 | + assert.True(t, HasExtension(tmp.Base, "Base2")) |
| 94 | + assert.True(t, HasExtension(tmp.Base.Base2, "a")) |
| 95 | + assert.False(t, HasExtension(tmp.Base, "Base3")) |
| 96 | + tmp.Extension = nil |
| 97 | + assert.False(t, HasExtension(tmp, "Extension")) |
| 98 | +} |
0 commit comments