github.com/onflow/flow-go@v0.33.17/fvm/environment/mock/environment.go (about) 1 // Code generated by mockery v2.21.4. DO NOT EDIT. 2 3 package mock 4 5 import ( 6 atree "github.com/onflow/atree" 7 ast "github.com/onflow/cadence/runtime/ast" 8 9 attribute "go.opentelemetry.io/otel/attribute" 10 11 cadence "github.com/onflow/cadence" 12 13 common "github.com/onflow/cadence/runtime/common" 14 15 environment "github.com/onflow/flow-go/fvm/environment" 16 17 flow "github.com/onflow/flow-go/model/flow" 18 19 interpreter "github.com/onflow/cadence/runtime/interpreter" 20 21 meter "github.com/onflow/flow-go/fvm/meter" 22 23 mock "github.com/stretchr/testify/mock" 24 25 oteltrace "go.opentelemetry.io/otel/trace" 26 27 runtime "github.com/onflow/flow-go/fvm/runtime" 28 29 sema "github.com/onflow/cadence/runtime/sema" 30 31 stdlib "github.com/onflow/cadence/runtime/stdlib" 32 33 time "time" 34 35 trace "github.com/onflow/flow-go/module/trace" 36 37 tracing "github.com/onflow/flow-go/fvm/tracing" 38 39 zerolog "github.com/rs/zerolog" 40 ) 41 42 // Environment is an autogenerated mock type for the Environment type 43 type Environment struct { 44 mock.Mock 45 } 46 47 // AccountKeysCount provides a mock function with given fields: address 48 func (_m *Environment) AccountKeysCount(address common.Address) (uint64, error) { 49 ret := _m.Called(address) 50 51 var r0 uint64 52 var r1 error 53 if rf, ok := ret.Get(0).(func(common.Address) (uint64, error)); ok { 54 return rf(address) 55 } 56 if rf, ok := ret.Get(0).(func(common.Address) uint64); ok { 57 r0 = rf(address) 58 } else { 59 r0 = ret.Get(0).(uint64) 60 } 61 62 if rf, ok := ret.Get(1).(func(common.Address) error); ok { 63 r1 = rf(address) 64 } else { 65 r1 = ret.Error(1) 66 } 67 68 return r0, r1 69 } 70 71 // AccountsStorageCapacity provides a mock function with given fields: addresses, payer, maxTxFees 72 func (_m *Environment) AccountsStorageCapacity(addresses []flow.Address, payer flow.Address, maxTxFees uint64) (cadence.Value, error) { 73 ret := _m.Called(addresses, payer, maxTxFees) 74 75 var r0 cadence.Value 76 var r1 error 77 if rf, ok := ret.Get(0).(func([]flow.Address, flow.Address, uint64) (cadence.Value, error)); ok { 78 return rf(addresses, payer, maxTxFees) 79 } 80 if rf, ok := ret.Get(0).(func([]flow.Address, flow.Address, uint64) cadence.Value); ok { 81 r0 = rf(addresses, payer, maxTxFees) 82 } else { 83 if ret.Get(0) != nil { 84 r0 = ret.Get(0).(cadence.Value) 85 } 86 } 87 88 if rf, ok := ret.Get(1).(func([]flow.Address, flow.Address, uint64) error); ok { 89 r1 = rf(addresses, payer, maxTxFees) 90 } else { 91 r1 = ret.Error(1) 92 } 93 94 return r0, r1 95 } 96 97 // AddAccountKey provides a mock function with given fields: address, publicKey, hashAlgo, weight 98 func (_m *Environment) AddAccountKey(address common.Address, publicKey *stdlib.PublicKey, hashAlgo sema.HashAlgorithm, weight int) (*stdlib.AccountKey, error) { 99 ret := _m.Called(address, publicKey, hashAlgo, weight) 100 101 var r0 *stdlib.AccountKey 102 var r1 error 103 if rf, ok := ret.Get(0).(func(common.Address, *stdlib.PublicKey, sema.HashAlgorithm, int) (*stdlib.AccountKey, error)); ok { 104 return rf(address, publicKey, hashAlgo, weight) 105 } 106 if rf, ok := ret.Get(0).(func(common.Address, *stdlib.PublicKey, sema.HashAlgorithm, int) *stdlib.AccountKey); ok { 107 r0 = rf(address, publicKey, hashAlgo, weight) 108 } else { 109 if ret.Get(0) != nil { 110 r0 = ret.Get(0).(*stdlib.AccountKey) 111 } 112 } 113 114 if rf, ok := ret.Get(1).(func(common.Address, *stdlib.PublicKey, sema.HashAlgorithm, int) error); ok { 115 r1 = rf(address, publicKey, hashAlgo, weight) 116 } else { 117 r1 = ret.Error(1) 118 } 119 120 return r0, r1 121 } 122 123 // AddEncodedAccountKey provides a mock function with given fields: address, publicKey 124 func (_m *Environment) AddEncodedAccountKey(address common.Address, publicKey []byte) error { 125 ret := _m.Called(address, publicKey) 126 127 var r0 error 128 if rf, ok := ret.Get(0).(func(common.Address, []byte) error); ok { 129 r0 = rf(address, publicKey) 130 } else { 131 r0 = ret.Error(0) 132 } 133 134 return r0 135 } 136 137 // AllocateStorageIndex provides a mock function with given fields: owner 138 func (_m *Environment) AllocateStorageIndex(owner []byte) (atree.StorageIndex, error) { 139 ret := _m.Called(owner) 140 141 var r0 atree.StorageIndex 142 var r1 error 143 if rf, ok := ret.Get(0).(func([]byte) (atree.StorageIndex, error)); ok { 144 return rf(owner) 145 } 146 if rf, ok := ret.Get(0).(func([]byte) atree.StorageIndex); ok { 147 r0 = rf(owner) 148 } else { 149 if ret.Get(0) != nil { 150 r0 = ret.Get(0).(atree.StorageIndex) 151 } 152 } 153 154 if rf, ok := ret.Get(1).(func([]byte) error); ok { 155 r1 = rf(owner) 156 } else { 157 r1 = ret.Error(1) 158 } 159 160 return r0, r1 161 } 162 163 // BLSAggregatePublicKeys provides a mock function with given fields: publicKeys 164 func (_m *Environment) BLSAggregatePublicKeys(publicKeys []*stdlib.PublicKey) (*stdlib.PublicKey, error) { 165 ret := _m.Called(publicKeys) 166 167 var r0 *stdlib.PublicKey 168 var r1 error 169 if rf, ok := ret.Get(0).(func([]*stdlib.PublicKey) (*stdlib.PublicKey, error)); ok { 170 return rf(publicKeys) 171 } 172 if rf, ok := ret.Get(0).(func([]*stdlib.PublicKey) *stdlib.PublicKey); ok { 173 r0 = rf(publicKeys) 174 } else { 175 if ret.Get(0) != nil { 176 r0 = ret.Get(0).(*stdlib.PublicKey) 177 } 178 } 179 180 if rf, ok := ret.Get(1).(func([]*stdlib.PublicKey) error); ok { 181 r1 = rf(publicKeys) 182 } else { 183 r1 = ret.Error(1) 184 } 185 186 return r0, r1 187 } 188 189 // BLSAggregateSignatures provides a mock function with given fields: signatures 190 func (_m *Environment) BLSAggregateSignatures(signatures [][]byte) ([]byte, error) { 191 ret := _m.Called(signatures) 192 193 var r0 []byte 194 var r1 error 195 if rf, ok := ret.Get(0).(func([][]byte) ([]byte, error)); ok { 196 return rf(signatures) 197 } 198 if rf, ok := ret.Get(0).(func([][]byte) []byte); ok { 199 r0 = rf(signatures) 200 } else { 201 if ret.Get(0) != nil { 202 r0 = ret.Get(0).([]byte) 203 } 204 } 205 206 if rf, ok := ret.Get(1).(func([][]byte) error); ok { 207 r1 = rf(signatures) 208 } else { 209 r1 = ret.Error(1) 210 } 211 212 return r0, r1 213 } 214 215 // BLSVerifyPOP provides a mock function with given fields: publicKey, signature 216 func (_m *Environment) BLSVerifyPOP(publicKey *stdlib.PublicKey, signature []byte) (bool, error) { 217 ret := _m.Called(publicKey, signature) 218 219 var r0 bool 220 var r1 error 221 if rf, ok := ret.Get(0).(func(*stdlib.PublicKey, []byte) (bool, error)); ok { 222 return rf(publicKey, signature) 223 } 224 if rf, ok := ret.Get(0).(func(*stdlib.PublicKey, []byte) bool); ok { 225 r0 = rf(publicKey, signature) 226 } else { 227 r0 = ret.Get(0).(bool) 228 } 229 230 if rf, ok := ret.Get(1).(func(*stdlib.PublicKey, []byte) error); ok { 231 r1 = rf(publicKey, signature) 232 } else { 233 r1 = ret.Error(1) 234 } 235 236 return r0, r1 237 } 238 239 // BorrowCadenceRuntime provides a mock function with given fields: 240 func (_m *Environment) BorrowCadenceRuntime() *runtime.ReusableCadenceRuntime { 241 ret := _m.Called() 242 243 var r0 *runtime.ReusableCadenceRuntime 244 if rf, ok := ret.Get(0).(func() *runtime.ReusableCadenceRuntime); ok { 245 r0 = rf() 246 } else { 247 if ret.Get(0) != nil { 248 r0 = ret.Get(0).(*runtime.ReusableCadenceRuntime) 249 } 250 } 251 252 return r0 253 } 254 255 // CheckPayerBalanceAndGetMaxTxFees provides a mock function with given fields: payer, inclusionEffort, executionEffort 256 func (_m *Environment) CheckPayerBalanceAndGetMaxTxFees(payer flow.Address, inclusionEffort uint64, executionEffort uint64) (cadence.Value, error) { 257 ret := _m.Called(payer, inclusionEffort, executionEffort) 258 259 var r0 cadence.Value 260 var r1 error 261 if rf, ok := ret.Get(0).(func(flow.Address, uint64, uint64) (cadence.Value, error)); ok { 262 return rf(payer, inclusionEffort, executionEffort) 263 } 264 if rf, ok := ret.Get(0).(func(flow.Address, uint64, uint64) cadence.Value); ok { 265 r0 = rf(payer, inclusionEffort, executionEffort) 266 } else { 267 if ret.Get(0) != nil { 268 r0 = ret.Get(0).(cadence.Value) 269 } 270 } 271 272 if rf, ok := ret.Get(1).(func(flow.Address, uint64, uint64) error); ok { 273 r1 = rf(payer, inclusionEffort, executionEffort) 274 } else { 275 r1 = ret.Error(1) 276 } 277 278 return r0, r1 279 } 280 281 // ComputationAvailable provides a mock function with given fields: _a0, _a1 282 func (_m *Environment) ComputationAvailable(_a0 common.ComputationKind, _a1 uint) bool { 283 ret := _m.Called(_a0, _a1) 284 285 var r0 bool 286 if rf, ok := ret.Get(0).(func(common.ComputationKind, uint) bool); ok { 287 r0 = rf(_a0, _a1) 288 } else { 289 r0 = ret.Get(0).(bool) 290 } 291 292 return r0 293 } 294 295 // ComputationIntensities provides a mock function with given fields: 296 func (_m *Environment) ComputationIntensities() meter.MeteredComputationIntensities { 297 ret := _m.Called() 298 299 var r0 meter.MeteredComputationIntensities 300 if rf, ok := ret.Get(0).(func() meter.MeteredComputationIntensities); ok { 301 r0 = rf() 302 } else { 303 if ret.Get(0) != nil { 304 r0 = ret.Get(0).(meter.MeteredComputationIntensities) 305 } 306 } 307 308 return r0 309 } 310 311 // ComputationUsed provides a mock function with given fields: 312 func (_m *Environment) ComputationUsed() (uint64, error) { 313 ret := _m.Called() 314 315 var r0 uint64 316 var r1 error 317 if rf, ok := ret.Get(0).(func() (uint64, error)); ok { 318 return rf() 319 } 320 if rf, ok := ret.Get(0).(func() uint64); ok { 321 r0 = rf() 322 } else { 323 r0 = ret.Get(0).(uint64) 324 } 325 326 if rf, ok := ret.Get(1).(func() error); ok { 327 r1 = rf() 328 } else { 329 r1 = ret.Error(1) 330 } 331 332 return r0, r1 333 } 334 335 // ConvertedServiceEvents provides a mock function with given fields: 336 func (_m *Environment) ConvertedServiceEvents() flow.ServiceEventList { 337 ret := _m.Called() 338 339 var r0 flow.ServiceEventList 340 if rf, ok := ret.Get(0).(func() flow.ServiceEventList); ok { 341 r0 = rf() 342 } else { 343 if ret.Get(0) != nil { 344 r0 = ret.Get(0).(flow.ServiceEventList) 345 } 346 } 347 348 return r0 349 } 350 351 // CreateAccount provides a mock function with given fields: payer 352 func (_m *Environment) CreateAccount(payer common.Address) (common.Address, error) { 353 ret := _m.Called(payer) 354 355 var r0 common.Address 356 var r1 error 357 if rf, ok := ret.Get(0).(func(common.Address) (common.Address, error)); ok { 358 return rf(payer) 359 } 360 if rf, ok := ret.Get(0).(func(common.Address) common.Address); ok { 361 r0 = rf(payer) 362 } else { 363 if ret.Get(0) != nil { 364 r0 = ret.Get(0).(common.Address) 365 } 366 } 367 368 if rf, ok := ret.Get(1).(func(common.Address) error); ok { 369 r1 = rf(payer) 370 } else { 371 r1 = ret.Error(1) 372 } 373 374 return r0, r1 375 } 376 377 // DecodeArgument provides a mock function with given fields: argument, argumentType 378 func (_m *Environment) DecodeArgument(argument []byte, argumentType cadence.Type) (cadence.Value, error) { 379 ret := _m.Called(argument, argumentType) 380 381 var r0 cadence.Value 382 var r1 error 383 if rf, ok := ret.Get(0).(func([]byte, cadence.Type) (cadence.Value, error)); ok { 384 return rf(argument, argumentType) 385 } 386 if rf, ok := ret.Get(0).(func([]byte, cadence.Type) cadence.Value); ok { 387 r0 = rf(argument, argumentType) 388 } else { 389 if ret.Get(0) != nil { 390 r0 = ret.Get(0).(cadence.Value) 391 } 392 } 393 394 if rf, ok := ret.Get(1).(func([]byte, cadence.Type) error); ok { 395 r1 = rf(argument, argumentType) 396 } else { 397 r1 = ret.Error(1) 398 } 399 400 return r0, r1 401 } 402 403 // DeductTransactionFees provides a mock function with given fields: payer, inclusionEffort, executionEffort 404 func (_m *Environment) DeductTransactionFees(payer flow.Address, inclusionEffort uint64, executionEffort uint64) (cadence.Value, error) { 405 ret := _m.Called(payer, inclusionEffort, executionEffort) 406 407 var r0 cadence.Value 408 var r1 error 409 if rf, ok := ret.Get(0).(func(flow.Address, uint64, uint64) (cadence.Value, error)); ok { 410 return rf(payer, inclusionEffort, executionEffort) 411 } 412 if rf, ok := ret.Get(0).(func(flow.Address, uint64, uint64) cadence.Value); ok { 413 r0 = rf(payer, inclusionEffort, executionEffort) 414 } else { 415 if ret.Get(0) != nil { 416 r0 = ret.Get(0).(cadence.Value) 417 } 418 } 419 420 if rf, ok := ret.Get(1).(func(flow.Address, uint64, uint64) error); ok { 421 r1 = rf(payer, inclusionEffort, executionEffort) 422 } else { 423 r1 = ret.Error(1) 424 } 425 426 return r0, r1 427 } 428 429 // EmitEvent provides a mock function with given fields: _a0 430 func (_m *Environment) EmitEvent(_a0 cadence.Event) error { 431 ret := _m.Called(_a0) 432 433 var r0 error 434 if rf, ok := ret.Get(0).(func(cadence.Event) error); ok { 435 r0 = rf(_a0) 436 } else { 437 r0 = ret.Error(0) 438 } 439 440 return r0 441 } 442 443 // Events provides a mock function with given fields: 444 func (_m *Environment) Events() flow.EventsList { 445 ret := _m.Called() 446 447 var r0 flow.EventsList 448 if rf, ok := ret.Get(0).(func() flow.EventsList); ok { 449 r0 = rf() 450 } else { 451 if ret.Get(0) != nil { 452 r0 = ret.Get(0).(flow.EventsList) 453 } 454 } 455 456 return r0 457 } 458 459 // FlushPendingUpdates provides a mock function with given fields: 460 func (_m *Environment) FlushPendingUpdates() (environment.ContractUpdates, error) { 461 ret := _m.Called() 462 463 var r0 environment.ContractUpdates 464 var r1 error 465 if rf, ok := ret.Get(0).(func() (environment.ContractUpdates, error)); ok { 466 return rf() 467 } 468 if rf, ok := ret.Get(0).(func() environment.ContractUpdates); ok { 469 r0 = rf() 470 } else { 471 r0 = ret.Get(0).(environment.ContractUpdates) 472 } 473 474 if rf, ok := ret.Get(1).(func() error); ok { 475 r1 = rf() 476 } else { 477 r1 = ret.Error(1) 478 } 479 480 return r0, r1 481 } 482 483 // GenerateAccountID provides a mock function with given fields: address 484 func (_m *Environment) GenerateAccountID(address common.Address) (uint64, error) { 485 ret := _m.Called(address) 486 487 var r0 uint64 488 var r1 error 489 if rf, ok := ret.Get(0).(func(common.Address) (uint64, error)); ok { 490 return rf(address) 491 } 492 if rf, ok := ret.Get(0).(func(common.Address) uint64); ok { 493 r0 = rf(address) 494 } else { 495 r0 = ret.Get(0).(uint64) 496 } 497 498 if rf, ok := ret.Get(1).(func(common.Address) error); ok { 499 r1 = rf(address) 500 } else { 501 r1 = ret.Error(1) 502 } 503 504 return r0, r1 505 } 506 507 // GenerateUUID provides a mock function with given fields: 508 func (_m *Environment) GenerateUUID() (uint64, error) { 509 ret := _m.Called() 510 511 var r0 uint64 512 var r1 error 513 if rf, ok := ret.Get(0).(func() (uint64, error)); ok { 514 return rf() 515 } 516 if rf, ok := ret.Get(0).(func() uint64); ok { 517 r0 = rf() 518 } else { 519 r0 = ret.Get(0).(uint64) 520 } 521 522 if rf, ok := ret.Get(1).(func() error); ok { 523 r1 = rf() 524 } else { 525 r1 = ret.Error(1) 526 } 527 528 return r0, r1 529 } 530 531 // GetAccount provides a mock function with given fields: address 532 func (_m *Environment) GetAccount(address flow.Address) (*flow.Account, error) { 533 ret := _m.Called(address) 534 535 var r0 *flow.Account 536 var r1 error 537 if rf, ok := ret.Get(0).(func(flow.Address) (*flow.Account, error)); ok { 538 return rf(address) 539 } 540 if rf, ok := ret.Get(0).(func(flow.Address) *flow.Account); ok { 541 r0 = rf(address) 542 } else { 543 if ret.Get(0) != nil { 544 r0 = ret.Get(0).(*flow.Account) 545 } 546 } 547 548 if rf, ok := ret.Get(1).(func(flow.Address) error); ok { 549 r1 = rf(address) 550 } else { 551 r1 = ret.Error(1) 552 } 553 554 return r0, r1 555 } 556 557 // GetAccountAvailableBalance provides a mock function with given fields: address 558 func (_m *Environment) GetAccountAvailableBalance(address common.Address) (uint64, error) { 559 ret := _m.Called(address) 560 561 var r0 uint64 562 var r1 error 563 if rf, ok := ret.Get(0).(func(common.Address) (uint64, error)); ok { 564 return rf(address) 565 } 566 if rf, ok := ret.Get(0).(func(common.Address) uint64); ok { 567 r0 = rf(address) 568 } else { 569 r0 = ret.Get(0).(uint64) 570 } 571 572 if rf, ok := ret.Get(1).(func(common.Address) error); ok { 573 r1 = rf(address) 574 } else { 575 r1 = ret.Error(1) 576 } 577 578 return r0, r1 579 } 580 581 // GetAccountBalance provides a mock function with given fields: address 582 func (_m *Environment) GetAccountBalance(address common.Address) (uint64, error) { 583 ret := _m.Called(address) 584 585 var r0 uint64 586 var r1 error 587 if rf, ok := ret.Get(0).(func(common.Address) (uint64, error)); ok { 588 return rf(address) 589 } 590 if rf, ok := ret.Get(0).(func(common.Address) uint64); ok { 591 r0 = rf(address) 592 } else { 593 r0 = ret.Get(0).(uint64) 594 } 595 596 if rf, ok := ret.Get(1).(func(common.Address) error); ok { 597 r1 = rf(address) 598 } else { 599 r1 = ret.Error(1) 600 } 601 602 return r0, r1 603 } 604 605 // GetAccountContractCode provides a mock function with given fields: location 606 func (_m *Environment) GetAccountContractCode(location common.AddressLocation) ([]byte, error) { 607 ret := _m.Called(location) 608 609 var r0 []byte 610 var r1 error 611 if rf, ok := ret.Get(0).(func(common.AddressLocation) ([]byte, error)); ok { 612 return rf(location) 613 } 614 if rf, ok := ret.Get(0).(func(common.AddressLocation) []byte); ok { 615 r0 = rf(location) 616 } else { 617 if ret.Get(0) != nil { 618 r0 = ret.Get(0).([]byte) 619 } 620 } 621 622 if rf, ok := ret.Get(1).(func(common.AddressLocation) error); ok { 623 r1 = rf(location) 624 } else { 625 r1 = ret.Error(1) 626 } 627 628 return r0, r1 629 } 630 631 // GetAccountContractNames provides a mock function with given fields: address 632 func (_m *Environment) GetAccountContractNames(address common.Address) ([]string, error) { 633 ret := _m.Called(address) 634 635 var r0 []string 636 var r1 error 637 if rf, ok := ret.Get(0).(func(common.Address) ([]string, error)); ok { 638 return rf(address) 639 } 640 if rf, ok := ret.Get(0).(func(common.Address) []string); ok { 641 r0 = rf(address) 642 } else { 643 if ret.Get(0) != nil { 644 r0 = ret.Get(0).([]string) 645 } 646 } 647 648 if rf, ok := ret.Get(1).(func(common.Address) error); ok { 649 r1 = rf(address) 650 } else { 651 r1 = ret.Error(1) 652 } 653 654 return r0, r1 655 } 656 657 // GetAccountKey provides a mock function with given fields: address, index 658 func (_m *Environment) GetAccountKey(address common.Address, index int) (*stdlib.AccountKey, error) { 659 ret := _m.Called(address, index) 660 661 var r0 *stdlib.AccountKey 662 var r1 error 663 if rf, ok := ret.Get(0).(func(common.Address, int) (*stdlib.AccountKey, error)); ok { 664 return rf(address, index) 665 } 666 if rf, ok := ret.Get(0).(func(common.Address, int) *stdlib.AccountKey); ok { 667 r0 = rf(address, index) 668 } else { 669 if ret.Get(0) != nil { 670 r0 = ret.Get(0).(*stdlib.AccountKey) 671 } 672 } 673 674 if rf, ok := ret.Get(1).(func(common.Address, int) error); ok { 675 r1 = rf(address, index) 676 } else { 677 r1 = ret.Error(1) 678 } 679 680 return r0, r1 681 } 682 683 // GetBlockAtHeight provides a mock function with given fields: height 684 func (_m *Environment) GetBlockAtHeight(height uint64) (stdlib.Block, bool, error) { 685 ret := _m.Called(height) 686 687 var r0 stdlib.Block 688 var r1 bool 689 var r2 error 690 if rf, ok := ret.Get(0).(func(uint64) (stdlib.Block, bool, error)); ok { 691 return rf(height) 692 } 693 if rf, ok := ret.Get(0).(func(uint64) stdlib.Block); ok { 694 r0 = rf(height) 695 } else { 696 r0 = ret.Get(0).(stdlib.Block) 697 } 698 699 if rf, ok := ret.Get(1).(func(uint64) bool); ok { 700 r1 = rf(height) 701 } else { 702 r1 = ret.Get(1).(bool) 703 } 704 705 if rf, ok := ret.Get(2).(func(uint64) error); ok { 706 r2 = rf(height) 707 } else { 708 r2 = ret.Error(2) 709 } 710 711 return r0, r1, r2 712 } 713 714 // GetCode provides a mock function with given fields: location 715 func (_m *Environment) GetCode(location common.Location) ([]byte, error) { 716 ret := _m.Called(location) 717 718 var r0 []byte 719 var r1 error 720 if rf, ok := ret.Get(0).(func(common.Location) ([]byte, error)); ok { 721 return rf(location) 722 } 723 if rf, ok := ret.Get(0).(func(common.Location) []byte); ok { 724 r0 = rf(location) 725 } else { 726 if ret.Get(0) != nil { 727 r0 = ret.Get(0).([]byte) 728 } 729 } 730 731 if rf, ok := ret.Get(1).(func(common.Location) error); ok { 732 r1 = rf(location) 733 } else { 734 r1 = ret.Error(1) 735 } 736 737 return r0, r1 738 } 739 740 // GetCurrentBlockHeight provides a mock function with given fields: 741 func (_m *Environment) GetCurrentBlockHeight() (uint64, error) { 742 ret := _m.Called() 743 744 var r0 uint64 745 var r1 error 746 if rf, ok := ret.Get(0).(func() (uint64, error)); ok { 747 return rf() 748 } 749 if rf, ok := ret.Get(0).(func() uint64); ok { 750 r0 = rf() 751 } else { 752 r0 = ret.Get(0).(uint64) 753 } 754 755 if rf, ok := ret.Get(1).(func() error); ok { 756 r1 = rf() 757 } else { 758 r1 = ret.Error(1) 759 } 760 761 return r0, r1 762 } 763 764 // GetInterpreterSharedState provides a mock function with given fields: 765 func (_m *Environment) GetInterpreterSharedState() *interpreter.SharedState { 766 ret := _m.Called() 767 768 var r0 *interpreter.SharedState 769 if rf, ok := ret.Get(0).(func() *interpreter.SharedState); ok { 770 r0 = rf() 771 } else { 772 if ret.Get(0) != nil { 773 r0 = ret.Get(0).(*interpreter.SharedState) 774 } 775 } 776 777 return r0 778 } 779 780 // GetOrLoadProgram provides a mock function with given fields: location, load 781 func (_m *Environment) GetOrLoadProgram(location common.Location, load func() (*interpreter.Program, error)) (*interpreter.Program, error) { 782 ret := _m.Called(location, load) 783 784 var r0 *interpreter.Program 785 var r1 error 786 if rf, ok := ret.Get(0).(func(common.Location, func() (*interpreter.Program, error)) (*interpreter.Program, error)); ok { 787 return rf(location, load) 788 } 789 if rf, ok := ret.Get(0).(func(common.Location, func() (*interpreter.Program, error)) *interpreter.Program); ok { 790 r0 = rf(location, load) 791 } else { 792 if ret.Get(0) != nil { 793 r0 = ret.Get(0).(*interpreter.Program) 794 } 795 } 796 797 if rf, ok := ret.Get(1).(func(common.Location, func() (*interpreter.Program, error)) error); ok { 798 r1 = rf(location, load) 799 } else { 800 r1 = ret.Error(1) 801 } 802 803 return r0, r1 804 } 805 806 // GetSigningAccounts provides a mock function with given fields: 807 func (_m *Environment) GetSigningAccounts() ([]common.Address, error) { 808 ret := _m.Called() 809 810 var r0 []common.Address 811 var r1 error 812 if rf, ok := ret.Get(0).(func() ([]common.Address, error)); ok { 813 return rf() 814 } 815 if rf, ok := ret.Get(0).(func() []common.Address); ok { 816 r0 = rf() 817 } else { 818 if ret.Get(0) != nil { 819 r0 = ret.Get(0).([]common.Address) 820 } 821 } 822 823 if rf, ok := ret.Get(1).(func() error); ok { 824 r1 = rf() 825 } else { 826 r1 = ret.Error(1) 827 } 828 829 return r0, r1 830 } 831 832 // GetStorageCapacity provides a mock function with given fields: address 833 func (_m *Environment) GetStorageCapacity(address common.Address) (uint64, error) { 834 ret := _m.Called(address) 835 836 var r0 uint64 837 var r1 error 838 if rf, ok := ret.Get(0).(func(common.Address) (uint64, error)); ok { 839 return rf(address) 840 } 841 if rf, ok := ret.Get(0).(func(common.Address) uint64); ok { 842 r0 = rf(address) 843 } else { 844 r0 = ret.Get(0).(uint64) 845 } 846 847 if rf, ok := ret.Get(1).(func(common.Address) error); ok { 848 r1 = rf(address) 849 } else { 850 r1 = ret.Error(1) 851 } 852 853 return r0, r1 854 } 855 856 // GetStorageUsed provides a mock function with given fields: address 857 func (_m *Environment) GetStorageUsed(address common.Address) (uint64, error) { 858 ret := _m.Called(address) 859 860 var r0 uint64 861 var r1 error 862 if rf, ok := ret.Get(0).(func(common.Address) (uint64, error)); ok { 863 return rf(address) 864 } 865 if rf, ok := ret.Get(0).(func(common.Address) uint64); ok { 866 r0 = rf(address) 867 } else { 868 r0 = ret.Get(0).(uint64) 869 } 870 871 if rf, ok := ret.Get(1).(func(common.Address) error); ok { 872 r1 = rf(address) 873 } else { 874 r1 = ret.Error(1) 875 } 876 877 return r0, r1 878 } 879 880 // GetValue provides a mock function with given fields: owner, key 881 func (_m *Environment) GetValue(owner []byte, key []byte) ([]byte, error) { 882 ret := _m.Called(owner, key) 883 884 var r0 []byte 885 var r1 error 886 if rf, ok := ret.Get(0).(func([]byte, []byte) ([]byte, error)); ok { 887 return rf(owner, key) 888 } 889 if rf, ok := ret.Get(0).(func([]byte, []byte) []byte); ok { 890 r0 = rf(owner, key) 891 } else { 892 if ret.Get(0) != nil { 893 r0 = ret.Get(0).([]byte) 894 } 895 } 896 897 if rf, ok := ret.Get(1).(func([]byte, []byte) error); ok { 898 r1 = rf(owner, key) 899 } else { 900 r1 = ret.Error(1) 901 } 902 903 return r0, r1 904 } 905 906 // Hash provides a mock function with given fields: data, tag, hashAlgorithm 907 func (_m *Environment) Hash(data []byte, tag string, hashAlgorithm sema.HashAlgorithm) ([]byte, error) { 908 ret := _m.Called(data, tag, hashAlgorithm) 909 910 var r0 []byte 911 var r1 error 912 if rf, ok := ret.Get(0).(func([]byte, string, sema.HashAlgorithm) ([]byte, error)); ok { 913 return rf(data, tag, hashAlgorithm) 914 } 915 if rf, ok := ret.Get(0).(func([]byte, string, sema.HashAlgorithm) []byte); ok { 916 r0 = rf(data, tag, hashAlgorithm) 917 } else { 918 if ret.Get(0) != nil { 919 r0 = ret.Get(0).([]byte) 920 } 921 } 922 923 if rf, ok := ret.Get(1).(func([]byte, string, sema.HashAlgorithm) error); ok { 924 r1 = rf(data, tag, hashAlgorithm) 925 } else { 926 r1 = ret.Error(1) 927 } 928 929 return r0, r1 930 } 931 932 // ImplementationDebugLog provides a mock function with given fields: message 933 func (_m *Environment) ImplementationDebugLog(message string) error { 934 ret := _m.Called(message) 935 936 var r0 error 937 if rf, ok := ret.Get(0).(func(string) error); ok { 938 r0 = rf(message) 939 } else { 940 r0 = ret.Error(0) 941 } 942 943 return r0 944 } 945 946 // InteractionUsed provides a mock function with given fields: 947 func (_m *Environment) InteractionUsed() (uint64, error) { 948 ret := _m.Called() 949 950 var r0 uint64 951 var r1 error 952 if rf, ok := ret.Get(0).(func() (uint64, error)); ok { 953 return rf() 954 } 955 if rf, ok := ret.Get(0).(func() uint64); ok { 956 r0 = rf() 957 } else { 958 r0 = ret.Get(0).(uint64) 959 } 960 961 if rf, ok := ret.Get(1).(func() error); ok { 962 r1 = rf() 963 } else { 964 r1 = ret.Error(1) 965 } 966 967 return r0, r1 968 } 969 970 // IsServiceAccountAuthorizer provides a mock function with given fields: 971 func (_m *Environment) IsServiceAccountAuthorizer() bool { 972 ret := _m.Called() 973 974 var r0 bool 975 if rf, ok := ret.Get(0).(func() bool); ok { 976 r0 = rf() 977 } else { 978 r0 = ret.Get(0).(bool) 979 } 980 981 return r0 982 } 983 984 // LimitAccountStorage provides a mock function with given fields: 985 func (_m *Environment) LimitAccountStorage() bool { 986 ret := _m.Called() 987 988 var r0 bool 989 if rf, ok := ret.Get(0).(func() bool); ok { 990 r0 = rf() 991 } else { 992 r0 = ret.Get(0).(bool) 993 } 994 995 return r0 996 } 997 998 // Logger provides a mock function with given fields: 999 func (_m *Environment) Logger() zerolog.Logger { 1000 ret := _m.Called() 1001 1002 var r0 zerolog.Logger 1003 if rf, ok := ret.Get(0).(func() zerolog.Logger); ok { 1004 r0 = rf() 1005 } else { 1006 r0 = ret.Get(0).(zerolog.Logger) 1007 } 1008 1009 return r0 1010 } 1011 1012 // Logs provides a mock function with given fields: 1013 func (_m *Environment) Logs() []string { 1014 ret := _m.Called() 1015 1016 var r0 []string 1017 if rf, ok := ret.Get(0).(func() []string); ok { 1018 r0 = rf() 1019 } else { 1020 if ret.Get(0) != nil { 1021 r0 = ret.Get(0).([]string) 1022 } 1023 } 1024 1025 return r0 1026 } 1027 1028 // MemoryUsed provides a mock function with given fields: 1029 func (_m *Environment) MemoryUsed() (uint64, error) { 1030 ret := _m.Called() 1031 1032 var r0 uint64 1033 var r1 error 1034 if rf, ok := ret.Get(0).(func() (uint64, error)); ok { 1035 return rf() 1036 } 1037 if rf, ok := ret.Get(0).(func() uint64); ok { 1038 r0 = rf() 1039 } else { 1040 r0 = ret.Get(0).(uint64) 1041 } 1042 1043 if rf, ok := ret.Get(1).(func() error); ok { 1044 r1 = rf() 1045 } else { 1046 r1 = ret.Error(1) 1047 } 1048 1049 return r0, r1 1050 } 1051 1052 // MeterComputation provides a mock function with given fields: operationType, intensity 1053 func (_m *Environment) MeterComputation(operationType common.ComputationKind, intensity uint) error { 1054 ret := _m.Called(operationType, intensity) 1055 1056 var r0 error 1057 if rf, ok := ret.Get(0).(func(common.ComputationKind, uint) error); ok { 1058 r0 = rf(operationType, intensity) 1059 } else { 1060 r0 = ret.Error(0) 1061 } 1062 1063 return r0 1064 } 1065 1066 // MeterEmittedEvent provides a mock function with given fields: byteSize 1067 func (_m *Environment) MeterEmittedEvent(byteSize uint64) error { 1068 ret := _m.Called(byteSize) 1069 1070 var r0 error 1071 if rf, ok := ret.Get(0).(func(uint64) error); ok { 1072 r0 = rf(byteSize) 1073 } else { 1074 r0 = ret.Error(0) 1075 } 1076 1077 return r0 1078 } 1079 1080 // MeterMemory provides a mock function with given fields: usage 1081 func (_m *Environment) MeterMemory(usage common.MemoryUsage) error { 1082 ret := _m.Called(usage) 1083 1084 var r0 error 1085 if rf, ok := ret.Get(0).(func(common.MemoryUsage) error); ok { 1086 r0 = rf(usage) 1087 } else { 1088 r0 = ret.Error(0) 1089 } 1090 1091 return r0 1092 } 1093 1094 // ProgramLog provides a mock function with given fields: _a0 1095 func (_m *Environment) ProgramLog(_a0 string) error { 1096 ret := _m.Called(_a0) 1097 1098 var r0 error 1099 if rf, ok := ret.Get(0).(func(string) error); ok { 1100 r0 = rf(_a0) 1101 } else { 1102 r0 = ret.Error(0) 1103 } 1104 1105 return r0 1106 } 1107 1108 // RandomSourceHistory provides a mock function with given fields: 1109 func (_m *Environment) RandomSourceHistory() ([]byte, error) { 1110 ret := _m.Called() 1111 1112 var r0 []byte 1113 var r1 error 1114 if rf, ok := ret.Get(0).(func() ([]byte, error)); ok { 1115 return rf() 1116 } 1117 if rf, ok := ret.Get(0).(func() []byte); ok { 1118 r0 = rf() 1119 } else { 1120 if ret.Get(0) != nil { 1121 r0 = ret.Get(0).([]byte) 1122 } 1123 } 1124 1125 if rf, ok := ret.Get(1).(func() error); ok { 1126 r1 = rf() 1127 } else { 1128 r1 = ret.Error(1) 1129 } 1130 1131 return r0, r1 1132 } 1133 1134 // ReadRandom provides a mock function with given fields: _a0 1135 func (_m *Environment) ReadRandom(_a0 []byte) error { 1136 ret := _m.Called(_a0) 1137 1138 var r0 error 1139 if rf, ok := ret.Get(0).(func([]byte) error); ok { 1140 r0 = rf(_a0) 1141 } else { 1142 r0 = ret.Error(0) 1143 } 1144 1145 return r0 1146 } 1147 1148 // RecordTrace provides a mock function with given fields: operation, location, duration, attrs 1149 func (_m *Environment) RecordTrace(operation string, location common.Location, duration time.Duration, attrs []attribute.KeyValue) { 1150 _m.Called(operation, location, duration, attrs) 1151 } 1152 1153 // RemoveAccountContractCode provides a mock function with given fields: location 1154 func (_m *Environment) RemoveAccountContractCode(location common.AddressLocation) error { 1155 ret := _m.Called(location) 1156 1157 var r0 error 1158 if rf, ok := ret.Get(0).(func(common.AddressLocation) error); ok { 1159 r0 = rf(location) 1160 } else { 1161 r0 = ret.Error(0) 1162 } 1163 1164 return r0 1165 } 1166 1167 // Reset provides a mock function with given fields: 1168 func (_m *Environment) Reset() { 1169 _m.Called() 1170 } 1171 1172 // ResolveLocation provides a mock function with given fields: identifiers, location 1173 func (_m *Environment) ResolveLocation(identifiers []ast.Identifier, location common.Location) ([]sema.ResolvedLocation, error) { 1174 ret := _m.Called(identifiers, location) 1175 1176 var r0 []sema.ResolvedLocation 1177 var r1 error 1178 if rf, ok := ret.Get(0).(func([]ast.Identifier, common.Location) ([]sema.ResolvedLocation, error)); ok { 1179 return rf(identifiers, location) 1180 } 1181 if rf, ok := ret.Get(0).(func([]ast.Identifier, common.Location) []sema.ResolvedLocation); ok { 1182 r0 = rf(identifiers, location) 1183 } else { 1184 if ret.Get(0) != nil { 1185 r0 = ret.Get(0).([]sema.ResolvedLocation) 1186 } 1187 } 1188 1189 if rf, ok := ret.Get(1).(func([]ast.Identifier, common.Location) error); ok { 1190 r1 = rf(identifiers, location) 1191 } else { 1192 r1 = ret.Error(1) 1193 } 1194 1195 return r0, r1 1196 } 1197 1198 // ResourceOwnerChanged provides a mock function with given fields: _a0, resource, oldOwner, newOwner 1199 func (_m *Environment) ResourceOwnerChanged(_a0 *interpreter.Interpreter, resource *interpreter.CompositeValue, oldOwner common.Address, newOwner common.Address) { 1200 _m.Called(_a0, resource, oldOwner, newOwner) 1201 } 1202 1203 // ReturnCadenceRuntime provides a mock function with given fields: _a0 1204 func (_m *Environment) ReturnCadenceRuntime(_a0 *runtime.ReusableCadenceRuntime) { 1205 _m.Called(_a0) 1206 } 1207 1208 // RevokeAccountKey provides a mock function with given fields: address, index 1209 func (_m *Environment) RevokeAccountKey(address common.Address, index int) (*stdlib.AccountKey, error) { 1210 ret := _m.Called(address, index) 1211 1212 var r0 *stdlib.AccountKey 1213 var r1 error 1214 if rf, ok := ret.Get(0).(func(common.Address, int) (*stdlib.AccountKey, error)); ok { 1215 return rf(address, index) 1216 } 1217 if rf, ok := ret.Get(0).(func(common.Address, int) *stdlib.AccountKey); ok { 1218 r0 = rf(address, index) 1219 } else { 1220 if ret.Get(0) != nil { 1221 r0 = ret.Get(0).(*stdlib.AccountKey) 1222 } 1223 } 1224 1225 if rf, ok := ret.Get(1).(func(common.Address, int) error); ok { 1226 r1 = rf(address, index) 1227 } else { 1228 r1 = ret.Error(1) 1229 } 1230 1231 return r0, r1 1232 } 1233 1234 // RevokeEncodedAccountKey provides a mock function with given fields: address, index 1235 func (_m *Environment) RevokeEncodedAccountKey(address common.Address, index int) ([]byte, error) { 1236 ret := _m.Called(address, index) 1237 1238 var r0 []byte 1239 var r1 error 1240 if rf, ok := ret.Get(0).(func(common.Address, int) ([]byte, error)); ok { 1241 return rf(address, index) 1242 } 1243 if rf, ok := ret.Get(0).(func(common.Address, int) []byte); ok { 1244 r0 = rf(address, index) 1245 } else { 1246 if ret.Get(0) != nil { 1247 r0 = ret.Get(0).([]byte) 1248 } 1249 } 1250 1251 if rf, ok := ret.Get(1).(func(common.Address, int) error); ok { 1252 r1 = rf(address, index) 1253 } else { 1254 r1 = ret.Error(1) 1255 } 1256 1257 return r0, r1 1258 } 1259 1260 // ServiceEvents provides a mock function with given fields: 1261 func (_m *Environment) ServiceEvents() flow.EventsList { 1262 ret := _m.Called() 1263 1264 var r0 flow.EventsList 1265 if rf, ok := ret.Get(0).(func() flow.EventsList); ok { 1266 r0 = rf() 1267 } else { 1268 if ret.Get(0) != nil { 1269 r0 = ret.Get(0).(flow.EventsList) 1270 } 1271 } 1272 1273 return r0 1274 } 1275 1276 // SetInterpreterSharedState provides a mock function with given fields: state 1277 func (_m *Environment) SetInterpreterSharedState(state *interpreter.SharedState) { 1278 _m.Called(state) 1279 } 1280 1281 // SetValue provides a mock function with given fields: owner, key, value 1282 func (_m *Environment) SetValue(owner []byte, key []byte, value []byte) error { 1283 ret := _m.Called(owner, key, value) 1284 1285 var r0 error 1286 if rf, ok := ret.Get(0).(func([]byte, []byte, []byte) error); ok { 1287 r0 = rf(owner, key, value) 1288 } else { 1289 r0 = ret.Error(0) 1290 } 1291 1292 return r0 1293 } 1294 1295 // StartChildSpan provides a mock function with given fields: name, options 1296 func (_m *Environment) StartChildSpan(name trace.SpanName, options ...oteltrace.SpanStartOption) tracing.TracerSpan { 1297 _va := make([]interface{}, len(options)) 1298 for _i := range options { 1299 _va[_i] = options[_i] 1300 } 1301 var _ca []interface{} 1302 _ca = append(_ca, name) 1303 _ca = append(_ca, _va...) 1304 ret := _m.Called(_ca...) 1305 1306 var r0 tracing.TracerSpan 1307 if rf, ok := ret.Get(0).(func(trace.SpanName, ...oteltrace.SpanStartOption) tracing.TracerSpan); ok { 1308 r0 = rf(name, options...) 1309 } else { 1310 r0 = ret.Get(0).(tracing.TracerSpan) 1311 } 1312 1313 return r0 1314 } 1315 1316 // TotalEmittedEventBytes provides a mock function with given fields: 1317 func (_m *Environment) TotalEmittedEventBytes() uint64 { 1318 ret := _m.Called() 1319 1320 var r0 uint64 1321 if rf, ok := ret.Get(0).(func() uint64); ok { 1322 r0 = rf() 1323 } else { 1324 r0 = ret.Get(0).(uint64) 1325 } 1326 1327 return r0 1328 } 1329 1330 // TransactionFeesEnabled provides a mock function with given fields: 1331 func (_m *Environment) TransactionFeesEnabled() bool { 1332 ret := _m.Called() 1333 1334 var r0 bool 1335 if rf, ok := ret.Get(0).(func() bool); ok { 1336 r0 = rf() 1337 } else { 1338 r0 = ret.Get(0).(bool) 1339 } 1340 1341 return r0 1342 } 1343 1344 // TxID provides a mock function with given fields: 1345 func (_m *Environment) TxID() flow.Identifier { 1346 ret := _m.Called() 1347 1348 var r0 flow.Identifier 1349 if rf, ok := ret.Get(0).(func() flow.Identifier); ok { 1350 r0 = rf() 1351 } else { 1352 if ret.Get(0) != nil { 1353 r0 = ret.Get(0).(flow.Identifier) 1354 } 1355 } 1356 1357 return r0 1358 } 1359 1360 // TxIndex provides a mock function with given fields: 1361 func (_m *Environment) TxIndex() uint32 { 1362 ret := _m.Called() 1363 1364 var r0 uint32 1365 if rf, ok := ret.Get(0).(func() uint32); ok { 1366 r0 = rf() 1367 } else { 1368 r0 = ret.Get(0).(uint32) 1369 } 1370 1371 return r0 1372 } 1373 1374 // UpdateAccountContractCode provides a mock function with given fields: location, code 1375 func (_m *Environment) UpdateAccountContractCode(location common.AddressLocation, code []byte) error { 1376 ret := _m.Called(location, code) 1377 1378 var r0 error 1379 if rf, ok := ret.Get(0).(func(common.AddressLocation, []byte) error); ok { 1380 r0 = rf(location, code) 1381 } else { 1382 r0 = ret.Error(0) 1383 } 1384 1385 return r0 1386 } 1387 1388 // ValidatePublicKey provides a mock function with given fields: key 1389 func (_m *Environment) ValidatePublicKey(key *stdlib.PublicKey) error { 1390 ret := _m.Called(key) 1391 1392 var r0 error 1393 if rf, ok := ret.Get(0).(func(*stdlib.PublicKey) error); ok { 1394 r0 = rf(key) 1395 } else { 1396 r0 = ret.Error(0) 1397 } 1398 1399 return r0 1400 } 1401 1402 // ValueExists provides a mock function with given fields: owner, key 1403 func (_m *Environment) ValueExists(owner []byte, key []byte) (bool, error) { 1404 ret := _m.Called(owner, key) 1405 1406 var r0 bool 1407 var r1 error 1408 if rf, ok := ret.Get(0).(func([]byte, []byte) (bool, error)); ok { 1409 return rf(owner, key) 1410 } 1411 if rf, ok := ret.Get(0).(func([]byte, []byte) bool); ok { 1412 r0 = rf(owner, key) 1413 } else { 1414 r0 = ret.Get(0).(bool) 1415 } 1416 1417 if rf, ok := ret.Get(1).(func([]byte, []byte) error); ok { 1418 r1 = rf(owner, key) 1419 } else { 1420 r1 = ret.Error(1) 1421 } 1422 1423 return r0, r1 1424 } 1425 1426 // VerifySignature provides a mock function with given fields: signature, tag, signedData, publicKey, signatureAlgorithm, hashAlgorithm 1427 func (_m *Environment) VerifySignature(signature []byte, tag string, signedData []byte, publicKey []byte, signatureAlgorithm sema.SignatureAlgorithm, hashAlgorithm sema.HashAlgorithm) (bool, error) { 1428 ret := _m.Called(signature, tag, signedData, publicKey, signatureAlgorithm, hashAlgorithm) 1429 1430 var r0 bool 1431 var r1 error 1432 if rf, ok := ret.Get(0).(func([]byte, string, []byte, []byte, sema.SignatureAlgorithm, sema.HashAlgorithm) (bool, error)); ok { 1433 return rf(signature, tag, signedData, publicKey, signatureAlgorithm, hashAlgorithm) 1434 } 1435 if rf, ok := ret.Get(0).(func([]byte, string, []byte, []byte, sema.SignatureAlgorithm, sema.HashAlgorithm) bool); ok { 1436 r0 = rf(signature, tag, signedData, publicKey, signatureAlgorithm, hashAlgorithm) 1437 } else { 1438 r0 = ret.Get(0).(bool) 1439 } 1440 1441 if rf, ok := ret.Get(1).(func([]byte, string, []byte, []byte, sema.SignatureAlgorithm, sema.HashAlgorithm) error); ok { 1442 r1 = rf(signature, tag, signedData, publicKey, signatureAlgorithm, hashAlgorithm) 1443 } else { 1444 r1 = ret.Error(1) 1445 } 1446 1447 return r0, r1 1448 } 1449 1450 type mockConstructorTestingTNewEnvironment interface { 1451 mock.TestingT 1452 Cleanup(func()) 1453 } 1454 1455 // NewEnvironment creates a new instance of Environment. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. 1456 func NewEnvironment(t mockConstructorTestingTNewEnvironment) *Environment { 1457 mock := &Environment{} 1458 mock.Mock.Test(t) 1459 1460 t.Cleanup(func() { mock.AssertExpectations(t) }) 1461 1462 return mock 1463 }