add test for invariant

VKCOM · Oct 29, 2024 · 3e0d434 · 3e0d434
1 parent 3f45129
commit 3e0d434
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/internal/tlcodegen/test/codegen_test/goldmaster_invariants_test.go b/internal/tlcodegen/test/codegen_test/goldmaster_invariants_test.go
@@ -0,0 +1,24 @@
+package codegen
+
+import (
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+import "github.com/vkcom/tl/internal/tlcodegen/test/gen/goldmaster/meta"
+
+func TestFunctionHasUnion(t *testing.T) {
+	{
+		fun := meta.FactoryItemByTLName("service5.insertList")
+		if assert.NotNil(t, fun) {
+			assert.True(t, fun.IsFunction())
+			assert.True(t, fun.HasUnionTypesInResult())
+		}
+	}
+	{
+		fun := meta.FactoryItemByTLName("usefulService.getUserEntity")
+		if assert.NotNil(t, fun) {
+			assert.True(t, fun.IsFunction())
+			assert.True(t, !fun.HasUnionTypesInResult())
+		}
+	}
+}