Intis/entity-maker
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Generate cloud instead televendcloud entities

Browse Source
This commit is contained in:
Eden Kirin
2025-12-15 10:37:28 +01:00
parent 0a4030c389
commit 5f5f1ad114
8 changed files with 24 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
internal/generator/generator_test.go
View File

@ -17,7 +17,7 @@ func TestGetRelationshipName(t *testing.T) {
{"with _id suffix", "author_info_id", "author_info"},
{"with user_id", "user_id", "user"},
{"without _id", "status", "status"},
{"just id", "id", "id"}, // "id" doesn't have "_id" suffix, so returns as-is
{"just id", "id", "id"}, // "id" doesn't have "_id" suffix, so returns as-is
{"empty", "", ""},
}
@ -370,8 +370,8 @@ func TestGenerateRepository(t *testing.T) {
"CRUDRepository",
"model_cls = User",
"from televend_core.databases.base_repository import CRUDRepository",
"from televend_core.databases.televend_repositories.user.filter import",
"from televend_core.databases.televend_repositories.user.model import User",
"from televend_core.databases.cloud_repositories.user.filter import",
"from televend_core.databases.cloud_repositories.user.model import User",
}
for _, expected := range expectedStrings {
@ -508,7 +508,7 @@ func TestGenerateTable(t *testing.T) {
},
expected: []string{
"Enum",
"from televend_core.databases.televend_repositories.order.enum import",
"from televend_core.databases.cloud_repositories.order.enum import",
"OrderStatus",
`Enum(`,
`*OrderStatus.to_value_list()`,
@ -580,7 +580,7 @@ func TestGenerateModel(t *testing.T) {
EnumTypes: map[string]database.EnumType{},
},
expected: []string{
"from televend_core.databases.televend_repositories.user.model import User",
"from televend_core.databases.cloud_repositories.user.model import User",
"user_id: int",
"user: User",
"title: str",
@ -791,7 +791,7 @@ func TestGenerateFactory(t *testing.T) {
"from __future__ import annotations",
"from typing import Type",
"import factory",
"class UserFactory(TelevendBaseFactory):",
"class UserFactory(CloudBaseFactory):",
"alive = True",
"id = None",
`name = factory.Faker("pystr", max_chars=100)`,
@ -816,7 +816,7 @@ func TestGenerateFactory(t *testing.T) {
EnumTypes: map[string]database.EnumType{},
},
expected: []string{
"from televend_core.databases.televend_repositories.user.factory import",
"from televend_core.databases.cloud_repositories.user.factory import",
"UserFactory",
`user = CustomSelfAttribute("..user", UserFactory)`,
"user_id = factory.LazyAttribute(lambda a: a.user.id if a.user else None)",