github.com/chalford/terraform@v0.3.7-0.20150113080010-a78c69a8c81f/terraform/hook_mock.go (about)

     1  package terraform
     2  
     3  // MockHook is an implementation of Hook that can be used for tests.
     4  // It records all of its function calls.
     5  type MockHook struct {
     6  	PreApplyCalled bool
     7  	PreApplyInfo   *InstanceInfo
     8  	PreApplyDiff   *InstanceDiff
     9  	PreApplyState  *InstanceState
    10  	PreApplyReturn HookAction
    11  	PreApplyError  error
    12  
    13  	PostApplyCalled      bool
    14  	PostApplyInfo        *InstanceInfo
    15  	PostApplyState       *InstanceState
    16  	PostApplyError       error
    17  	PostApplyReturn      HookAction
    18  	PostApplyReturnError error
    19  
    20  	PreDiffCalled bool
    21  	PreDiffInfo   *InstanceInfo
    22  	PreDiffState  *InstanceState
    23  	PreDiffReturn HookAction
    24  	PreDiffError  error
    25  
    26  	PostDiffCalled bool
    27  	PostDiffInfo   *InstanceInfo
    28  	PostDiffDiff   *InstanceDiff
    29  	PostDiffReturn HookAction
    30  	PostDiffError  error
    31  
    32  	PreProvisionResourceCalled bool
    33  	PreProvisionResourceInfo   *InstanceInfo
    34  	PreProvisionInstanceState  *InstanceState
    35  	PreProvisionResourceReturn HookAction
    36  	PreProvisionResourceError  error
    37  
    38  	PostProvisionResourceCalled bool
    39  	PostProvisionResourceInfo   *InstanceInfo
    40  	PostProvisionInstanceState  *InstanceState
    41  	PostProvisionResourceReturn HookAction
    42  	PostProvisionResourceError  error
    43  
    44  	PreProvisionCalled        bool
    45  	PreProvisionInfo          *InstanceInfo
    46  	PreProvisionProvisionerId string
    47  	PreProvisionReturn        HookAction
    48  	PreProvisionError         error
    49  
    50  	PostProvisionCalled        bool
    51  	PostProvisionInfo          *InstanceInfo
    52  	PostProvisionProvisionerId string
    53  	PostProvisionReturn        HookAction
    54  	PostProvisionError         error
    55  
    56  	ProvisionOutputCalled        bool
    57  	ProvisionOutputInfo          *InstanceInfo
    58  	ProvisionOutputProvisionerId string
    59  	ProvisionOutputMessage       string
    60  
    61  	PostRefreshCalled bool
    62  	PostRefreshInfo   *InstanceInfo
    63  	PostRefreshState  *InstanceState
    64  	PostRefreshReturn HookAction
    65  	PostRefreshError  error
    66  
    67  	PreRefreshCalled bool
    68  	PreRefreshInfo   *InstanceInfo
    69  	PreRefreshState  *InstanceState
    70  	PreRefreshReturn HookAction
    71  	PreRefreshError  error
    72  }
    73  
    74  func (h *MockHook) PreApply(n *InstanceInfo, s *InstanceState, d *InstanceDiff) (HookAction, error) {
    75  	h.PreApplyCalled = true
    76  	h.PreApplyInfo = n
    77  	h.PreApplyDiff = d
    78  	h.PreApplyState = s
    79  	return h.PreApplyReturn, h.PreApplyError
    80  }
    81  
    82  func (h *MockHook) PostApply(n *InstanceInfo, s *InstanceState, e error) (HookAction, error) {
    83  	h.PostApplyCalled = true
    84  	h.PostApplyInfo = n
    85  	h.PostApplyState = s
    86  	h.PostApplyError = e
    87  	return h.PostApplyReturn, h.PostApplyReturnError
    88  }
    89  
    90  func (h *MockHook) PreDiff(n *InstanceInfo, s *InstanceState) (HookAction, error) {
    91  	h.PreDiffCalled = true
    92  	h.PreDiffInfo = n
    93  	h.PreDiffState = s
    94  	return h.PreDiffReturn, h.PreDiffError
    95  }
    96  
    97  func (h *MockHook) PostDiff(n *InstanceInfo, d *InstanceDiff) (HookAction, error) {
    98  	h.PostDiffCalled = true
    99  	h.PostDiffInfo = n
   100  	h.PostDiffDiff = d
   101  	return h.PostDiffReturn, h.PostDiffError
   102  }
   103  
   104  func (h *MockHook) PreProvisionResource(n *InstanceInfo, s *InstanceState) (HookAction, error) {
   105  	h.PreProvisionResourceCalled = true
   106  	h.PreProvisionResourceInfo = n
   107  	h.PreProvisionInstanceState = s
   108  	return h.PreProvisionResourceReturn, h.PreProvisionResourceError
   109  }
   110  
   111  func (h *MockHook) PostProvisionResource(n *InstanceInfo, s *InstanceState) (HookAction, error) {
   112  	h.PostProvisionResourceCalled = true
   113  	h.PostProvisionResourceInfo = n
   114  	h.PostProvisionInstanceState = s
   115  	return h.PostProvisionResourceReturn, h.PostProvisionResourceError
   116  }
   117  
   118  func (h *MockHook) PreProvision(n *InstanceInfo, provId string) (HookAction, error) {
   119  	h.PreProvisionCalled = true
   120  	h.PreProvisionInfo = n
   121  	h.PreProvisionProvisionerId = provId
   122  	return h.PreProvisionReturn, h.PreProvisionError
   123  }
   124  
   125  func (h *MockHook) PostProvision(n *InstanceInfo, provId string) (HookAction, error) {
   126  	h.PostProvisionCalled = true
   127  	h.PostProvisionInfo = n
   128  	h.PostProvisionProvisionerId = provId
   129  	return h.PostProvisionReturn, h.PostProvisionError
   130  }
   131  
   132  func (h *MockHook) ProvisionOutput(
   133  	n *InstanceInfo,
   134  	provId string,
   135  	msg string) {
   136  	h.ProvisionOutputCalled = true
   137  	h.ProvisionOutputInfo = n
   138  	h.ProvisionOutputProvisionerId = provId
   139  	h.ProvisionOutputMessage = msg
   140  }
   141  
   142  func (h *MockHook) PreRefresh(n *InstanceInfo, s *InstanceState) (HookAction, error) {
   143  	h.PreRefreshCalled = true
   144  	h.PreRefreshInfo = n
   145  	h.PreRefreshState = s
   146  	return h.PreRefreshReturn, h.PreRefreshError
   147  }
   148  
   149  func (h *MockHook) PostRefresh(n *InstanceInfo, s *InstanceState) (HookAction, error) {
   150  	h.PostRefreshCalled = true
   151  	h.PostRefreshInfo = n
   152  	h.PostRefreshState = s
   153  	return h.PostRefreshReturn, h.PostRefreshError
   154  }