reset_test.go 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  1. package rest
  2. import (
  3. "fmt"
  4. "git.nspix.com/golang/rest/orm/schema"
  5. "gorm.io/driver/sqlite"
  6. "gorm.io/gorm"
  7. "testing"
  8. )
  9. type TestModel struct {
  10. Id int `json:"id"`
  11. Name string `json:"name"`
  12. Age int `json:"age"`
  13. Description string `json:"description"`
  14. }
  15. func TestLocalDriver(t *testing.T) {
  16. db, err := gorm.Open(sqlite.Open("test.db"), &gorm.Config{})
  17. if err != nil {
  18. t.Fatal(err.Error())
  19. }
  20. Initialize(nil, db)
  21. if err = RegisterModel("test", &TestModel{}); err != nil {
  22. t.Fatal(err.Error())
  23. } else {
  24. t.Log("OK")
  25. }
  26. }
  27. func TestRemoteDriver(t *testing.T) {
  28. db, err := gorm.Open(sqlite.Open("test.db"), &gorm.Config{})
  29. if err != nil {
  30. t.Fatal(err.Error())
  31. }
  32. schema.SetDriver(&schema.RemoteDriver{Url: "https://schema.nspix.com/schema"})
  33. Initialize(nil, db)
  34. if err = RegisterModel("test", &TestModel{}); err != nil {
  35. t.Fatal(err.Error())
  36. } else {
  37. t.Log("OK")
  38. }
  39. }
  40. func TestVisibleField(t *testing.T) {
  41. db, err := gorm.Open(sqlite.Open("test.db"), &gorm.Config{})
  42. if err != nil {
  43. t.Fatal(err.Error())
  44. }
  45. schema.SetDriver(&schema.RemoteDriver{Url: "https://schema.nspix.com/schema"})
  46. Initialize(nil, db)
  47. ss := schema.VisibleField("test", "test_models", "")
  48. for _, s := range ss {
  49. fmt.Println(s)
  50. }
  51. }