github.com/Ali-iotechsys/sqlboiler/v4@v4.0.0-20221208124957-6aec9a5f1f71/templates/test/finishers.go.tpl (about)

     1  {{- $alias := .Aliases.Table .Table.Name}}
     2  func test{{$alias.UpPlural}}Bind(t *testing.T) {
     3  	t.Parallel()
     4  
     5  	seed := randomize.NewSeed()
     6  	var err error
     7  	o := &{{$alias.UpSingular}}{}
     8  	if err = randomize.Struct(seed, o, {{$alias.DownSingular}}DBTypes, true, {{$alias.DownSingular}}ColumnsWithDefault...); err != nil {
     9  		t.Errorf("Unable to randomize {{$alias.UpSingular}} struct: %s", err)
    10  	}
    11  
    12  	{{if not .NoContext}}ctx := context.Background(){{end}}
    13  	tx := MustTx({{if .NoContext}}boil.Begin(){{else}}boil.BeginTx(ctx, nil){{end}})
    14  	defer func() { _ = tx.Rollback() }()
    15  	if err = o.Insert({{if not .NoContext}}ctx, {{end -}} tx, boil.Infer()); err != nil {
    16  		t.Error(err)
    17  	}
    18  
    19  	if err = {{$alias.UpPlural}}().Bind({{if .NoContext}}nil{{else}}ctx{{end}}, tx, o); err != nil {
    20  		t.Error(err)
    21  	}
    22  }
    23  
    24  func test{{$alias.UpPlural}}One(t *testing.T) {
    25  	t.Parallel()
    26  
    27  	seed := randomize.NewSeed()
    28  	var err error
    29  	o := &{{$alias.UpSingular}}{}
    30  	if err = randomize.Struct(seed, o, {{$alias.DownSingular}}DBTypes, true, {{$alias.DownSingular}}ColumnsWithDefault...); err != nil {
    31  		t.Errorf("Unable to randomize {{$alias.UpSingular}} struct: %s", err)
    32  	}
    33  
    34  	{{if not .NoContext}}ctx := context.Background(){{end}}
    35  	tx := MustTx({{if .NoContext}}boil.Begin(){{else}}boil.BeginTx(ctx, nil){{end}})
    36  	defer func() { _ = tx.Rollback() }()
    37  	if err = o.Insert({{if not .NoContext}}ctx, {{end -}} tx, boil.Infer()); err != nil {
    38  		t.Error(err)
    39  	}
    40  
    41  	if x, err := {{$alias.UpPlural}}().One({{if not .NoContext}}ctx, {{end -}} tx); err != nil {
    42  		t.Error(err)
    43  	} else if x == nil {
    44  		t.Error("expected to get a non nil record")
    45  	}
    46  }
    47  
    48  func test{{$alias.UpPlural}}All(t *testing.T) {
    49  	t.Parallel()
    50  
    51  	seed := randomize.NewSeed()
    52  	var err error
    53  	{{$alias.DownSingular}}One := &{{$alias.UpSingular}}{}
    54  	{{$alias.DownSingular}}Two := &{{$alias.UpSingular}}{}
    55  	if err = randomize.Struct(seed, {{$alias.DownSingular}}One, {{$alias.DownSingular}}DBTypes, false, {{$alias.DownSingular}}ColumnsWithDefault...); err != nil {
    56  		t.Errorf("Unable to randomize {{$alias.UpSingular}} struct: %s", err)
    57  	}
    58  	if err = randomize.Struct(seed, {{$alias.DownSingular}}Two, {{$alias.DownSingular}}DBTypes, false, {{$alias.DownSingular}}ColumnsWithDefault...); err != nil {
    59  		t.Errorf("Unable to randomize {{$alias.UpSingular}} struct: %s", err)
    60  	}
    61  
    62  	{{if not .NoContext}}ctx := context.Background(){{end}}
    63  	tx := MustTx({{if .NoContext}}boil.Begin(){{else}}boil.BeginTx(ctx, nil){{end}})
    64  	defer func() { _ = tx.Rollback() }()
    65  	if err = {{$alias.DownSingular}}One.Insert({{if not .NoContext}}ctx, {{end -}} tx, boil.Infer()); err != nil {
    66  		t.Error(err)
    67  	}
    68  	if err = {{$alias.DownSingular}}Two.Insert({{if not .NoContext}}ctx, {{end -}} tx, boil.Infer()); err != nil {
    69  		t.Error(err)
    70  	}
    71  
    72  	slice, err := {{$alias.UpPlural}}().All({{if not .NoContext}}ctx, {{end -}} tx)
    73  	if err != nil {
    74  		t.Error(err)
    75  	}
    76  
    77  	if len(slice) != 2 {
    78  		t.Error("want 2 records, got:", len(slice))
    79  	}
    80  }
    81  
    82  func test{{$alias.UpPlural}}Count(t *testing.T) {
    83  	t.Parallel()
    84  
    85  	var err error
    86  	seed := randomize.NewSeed()
    87  	{{$alias.DownSingular}}One := &{{$alias.UpSingular}}{}
    88  	{{$alias.DownSingular}}Two := &{{$alias.UpSingular}}{}
    89  	if err = randomize.Struct(seed, {{$alias.DownSingular}}One, {{$alias.DownSingular}}DBTypes, false, {{$alias.DownSingular}}ColumnsWithDefault...); err != nil {
    90  		t.Errorf("Unable to randomize {{$alias.UpSingular}} struct: %s", err)
    91  	}
    92  	if err = randomize.Struct(seed, {{$alias.DownSingular}}Two, {{$alias.DownSingular}}DBTypes, false, {{$alias.DownSingular}}ColumnsWithDefault...); err != nil {
    93  		t.Errorf("Unable to randomize {{$alias.UpSingular}} struct: %s", err)
    94  	}
    95  
    96  	{{if not .NoContext}}ctx := context.Background(){{end}}
    97  	tx := MustTx({{if .NoContext}}boil.Begin(){{else}}boil.BeginTx(ctx, nil){{end}})
    98  	defer func() { _ = tx.Rollback() }()
    99  	if err = {{$alias.DownSingular}}One.Insert({{if not .NoContext}}ctx, {{end -}} tx, boil.Infer()); err != nil {
   100  		t.Error(err)
   101  	}
   102  	if err = {{$alias.DownSingular}}Two.Insert({{if not .NoContext}}ctx, {{end -}} tx, boil.Infer()); err != nil {
   103  		t.Error(err)
   104  	}
   105  
   106  	count, err := {{$alias.UpPlural}}().Count({{if not .NoContext}}ctx, {{end -}} tx)
   107  	if err != nil {
   108  		t.Error(err)
   109  	}
   110  
   111  	if count != 2 {
   112  		t.Error("want 2 records, got:", count)
   113  	}
   114  }