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

     1  {{- if not .NoHooks -}}
     2  {{- $alias := .Aliases.Table .Table.Name}}
     3  func {{$alias.DownSingular}}BeforeInsertHook({{if .NoContext}}e boil.Executor{{else}}ctx context.Context, e boil.ContextExecutor{{end}}, o *{{$alias.UpSingular}}) error {
     4  	*o = {{$alias.UpSingular}}{}
     5  	return nil
     6  }
     7  
     8  func {{$alias.DownSingular}}AfterInsertHook({{if .NoContext}}e boil.Executor{{else}}ctx context.Context, e boil.ContextExecutor{{end}}, o *{{$alias.UpSingular}}) error {
     9  	*o = {{$alias.UpSingular}}{}
    10  	return nil
    11  }
    12  
    13  func {{$alias.DownSingular}}AfterSelectHook({{if .NoContext}}e boil.Executor{{else}}ctx context.Context, e boil.ContextExecutor{{end}}, o *{{$alias.UpSingular}}) error {
    14  	*o = {{$alias.UpSingular}}{}
    15  	return nil
    16  }
    17  
    18  func {{$alias.DownSingular}}BeforeUpdateHook({{if .NoContext}}e boil.Executor{{else}}ctx context.Context, e boil.ContextExecutor{{end}}, o *{{$alias.UpSingular}}) error {
    19  	*o = {{$alias.UpSingular}}{}
    20  	return nil
    21  }
    22  
    23  func {{$alias.DownSingular}}AfterUpdateHook({{if .NoContext}}e boil.Executor{{else}}ctx context.Context, e boil.ContextExecutor{{end}}, o *{{$alias.UpSingular}}) error {
    24  	*o = {{$alias.UpSingular}}{}
    25  	return nil
    26  }
    27  
    28  func {{$alias.DownSingular}}BeforeDeleteHook({{if .NoContext}}e boil.Executor{{else}}ctx context.Context, e boil.ContextExecutor{{end}}, o *{{$alias.UpSingular}}) error {
    29  	*o = {{$alias.UpSingular}}{}
    30  	return nil
    31  }
    32  
    33  func {{$alias.DownSingular}}AfterDeleteHook({{if .NoContext}}e boil.Executor{{else}}ctx context.Context, e boil.ContextExecutor{{end}}, o *{{$alias.UpSingular}}) error {
    34  	*o = {{$alias.UpSingular}}{}
    35  	return nil
    36  }
    37  
    38  func {{$alias.DownSingular}}BeforeUpsertHook({{if .NoContext}}e boil.Executor{{else}}ctx context.Context, e boil.ContextExecutor{{end}}, o *{{$alias.UpSingular}}) error {
    39  	*o = {{$alias.UpSingular}}{}
    40  	return nil
    41  }
    42  
    43  func {{$alias.DownSingular}}AfterUpsertHook({{if .NoContext}}e boil.Executor{{else}}ctx context.Context, e boil.ContextExecutor{{end}}, o *{{$alias.UpSingular}}) error {
    44  	*o = {{$alias.UpSingular}}{}
    45  	return nil
    46  }
    47  
    48  func test{{$alias.UpPlural}}Hooks(t *testing.T) {
    49  	t.Parallel()
    50  
    51  	var err error
    52  
    53  	{{if not .NoContext}}ctx := context.Background(){{end}}
    54  	empty := &{{$alias.UpSingular}}{}
    55  	o := &{{$alias.UpSingular}}{}
    56  
    57  	seed := randomize.NewSeed()
    58  	if err = randomize.Struct(seed, o, {{$alias.DownSingular}}DBTypes, false); err != nil {
    59  		t.Errorf("Unable to randomize {{$alias.UpSingular}} object: %s", err)
    60  	}
    61  
    62  	Add{{$alias.UpSingular}}Hook(boil.BeforeInsertHook, {{$alias.DownSingular}}BeforeInsertHook)
    63  	if err = o.doBeforeInsertHooks({{if not .NoContext}}ctx, {{end -}} nil); err != nil {
    64  		t.Errorf("Unable to execute doBeforeInsertHooks: %s", err)
    65  	}
    66  	if !reflect.DeepEqual(o, empty) {
    67  		t.Errorf("Expected BeforeInsertHook function to empty object, but got: %#v", o)
    68  	}
    69  	{{$alias.DownSingular}}BeforeInsertHooks = []{{$alias.UpSingular}}Hook{}
    70  
    71  	Add{{$alias.UpSingular}}Hook(boil.AfterInsertHook, {{$alias.DownSingular}}AfterInsertHook)
    72  	if err = o.doAfterInsertHooks({{if not .NoContext}}ctx, {{end -}} nil); err != nil {
    73  		t.Errorf("Unable to execute doAfterInsertHooks: %s", err)
    74  	}
    75  	if !reflect.DeepEqual(o, empty) {
    76  		t.Errorf("Expected AfterInsertHook function to empty object, but got: %#v", o)
    77  	}
    78  	{{$alias.DownSingular}}AfterInsertHooks = []{{$alias.UpSingular}}Hook{}
    79  
    80  	Add{{$alias.UpSingular}}Hook(boil.AfterSelectHook, {{$alias.DownSingular}}AfterSelectHook)
    81  	if err = o.doAfterSelectHooks({{if not .NoContext}}ctx, {{end -}} nil); err != nil {
    82  		t.Errorf("Unable to execute doAfterSelectHooks: %s", err)
    83  	}
    84  	if !reflect.DeepEqual(o, empty) {
    85  		t.Errorf("Expected AfterSelectHook function to empty object, but got: %#v", o)
    86  	}
    87  	{{$alias.DownSingular}}AfterSelectHooks = []{{$alias.UpSingular}}Hook{}
    88  
    89  	Add{{$alias.UpSingular}}Hook(boil.BeforeUpdateHook, {{$alias.DownSingular}}BeforeUpdateHook)
    90  	if err = o.doBeforeUpdateHooks({{if not .NoContext}}ctx, {{end -}} nil); err != nil {
    91  		t.Errorf("Unable to execute doBeforeUpdateHooks: %s", err)
    92  	}
    93  	if !reflect.DeepEqual(o, empty) {
    94  		t.Errorf("Expected BeforeUpdateHook function to empty object, but got: %#v", o)
    95  	}
    96  	{{$alias.DownSingular}}BeforeUpdateHooks = []{{$alias.UpSingular}}Hook{}
    97  
    98  	Add{{$alias.UpSingular}}Hook(boil.AfterUpdateHook, {{$alias.DownSingular}}AfterUpdateHook)
    99  	if err = o.doAfterUpdateHooks({{if not .NoContext}}ctx, {{end -}} nil); err != nil {
   100  		t.Errorf("Unable to execute doAfterUpdateHooks: %s", err)
   101  	}
   102  	if !reflect.DeepEqual(o, empty) {
   103  		t.Errorf("Expected AfterUpdateHook function to empty object, but got: %#v", o)
   104  	}
   105  	{{$alias.DownSingular}}AfterUpdateHooks = []{{$alias.UpSingular}}Hook{}
   106  
   107  	Add{{$alias.UpSingular}}Hook(boil.BeforeDeleteHook, {{$alias.DownSingular}}BeforeDeleteHook)
   108  	if err = o.doBeforeDeleteHooks({{if not .NoContext}}ctx, {{end -}} nil); err != nil {
   109  		t.Errorf("Unable to execute doBeforeDeleteHooks: %s", err)
   110  	}
   111  	if !reflect.DeepEqual(o, empty) {
   112  		t.Errorf("Expected BeforeDeleteHook function to empty object, but got: %#v", o)
   113  	}
   114  	{{$alias.DownSingular}}BeforeDeleteHooks = []{{$alias.UpSingular}}Hook{}
   115  
   116  	Add{{$alias.UpSingular}}Hook(boil.AfterDeleteHook, {{$alias.DownSingular}}AfterDeleteHook)
   117  	if err = o.doAfterDeleteHooks({{if not .NoContext}}ctx, {{end -}} nil); err != nil {
   118  		t.Errorf("Unable to execute doAfterDeleteHooks: %s", err)
   119  	}
   120  	if !reflect.DeepEqual(o, empty) {
   121  		t.Errorf("Expected AfterDeleteHook function to empty object, but got: %#v", o)
   122  	}
   123  	{{$alias.DownSingular}}AfterDeleteHooks = []{{$alias.UpSingular}}Hook{}
   124  
   125  	Add{{$alias.UpSingular}}Hook(boil.BeforeUpsertHook, {{$alias.DownSingular}}BeforeUpsertHook)
   126  	if err = o.doBeforeUpsertHooks({{if not .NoContext}}ctx, {{end -}} nil); err != nil {
   127  		t.Errorf("Unable to execute doBeforeUpsertHooks: %s", err)
   128  	}
   129  	if !reflect.DeepEqual(o, empty) {
   130  		t.Errorf("Expected BeforeUpsertHook function to empty object, but got: %#v", o)
   131  	}
   132  	{{$alias.DownSingular}}BeforeUpsertHooks = []{{$alias.UpSingular}}Hook{}
   133  
   134  	Add{{$alias.UpSingular}}Hook(boil.AfterUpsertHook, {{$alias.DownSingular}}AfterUpsertHook)
   135  	if err = o.doAfterUpsertHooks({{if not .NoContext}}ctx, {{end -}} nil); err != nil {
   136  		t.Errorf("Unable to execute doAfterUpsertHooks: %s", err)
   137  	}
   138  	if !reflect.DeepEqual(o, empty) {
   139  		t.Errorf("Expected AfterUpsertHook function to empty object, but got: %#v", o)
   140  	}
   141  	{{$alias.DownSingular}}AfterUpsertHooks = []{{$alias.UpSingular}}Hook{}
   142  }
   143  {{- end}}