// Copyright 2016 Mender Software AS
//
//    Licensed under the Apache License, Version 2.0 (the "License");
//    you may not use this file except in compliance with the License.
//    You may obtain a copy of the License at
//
//        http://www.apache.org/licenses/LICENSE-2.0
//
//    Unless required by applicable law or agreed to in writing, software
//    distributed under the License is distributed on an "AS IS" BASIS,
//    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//    See the License for the specific language governing permissions and
//    limitations under the License.

package mocks

import (
	"github.com/mendersoftware/deployments/integration"
	"github.com/stretchr/testify/mock"
)

// APIClient is an autogenerated mock type for the APIClient type
type APIClient struct {
	mock.Mock
}

// GetDeviceInventory provides a mock function with given fields: device
func (_m *APIClient) GetDeviceInventory(device integration.DeviceID) (*integration.Device, error) {
	ret := _m.Called(device)

	var r0 *integration.Device
	if rf, ok := ret.Get(0).(func(integration.DeviceID) *integration.Device); ok {
		r0 = rf(device)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*integration.Device)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(integration.DeviceID) error); ok {
		r1 = rf(device)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

package mocks

import (
	"github.com/mendersoftware/deployments/integration"
	"github.com/stretchr/testify/mock"
)

// APIClient is an autogenerated mock type for the APIClient type
type APIClient struct {
	mock.Mock
}

// GetDeviceInventory provides a mock function with given fields: device
func (_m *APIClient) GetDeviceInventory(device integration.DeviceID) (*integration.Device, error) {
	ret := _m.Called(device)

	var r0 *integration.Device
	if rf, ok := ret.Get(0).(func(integration.DeviceID) *integration.Device); ok {
		r0 = rf(device)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*integration.Device)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(integration.DeviceID) error); ok {
		r1 = rf(device)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

--- 
+++ 
@@ -1,3 +1,17 @@
+// Copyright 2016 Mender Software AS
+//
+//    Licensed under the Apache License, Version 2.0 (the "License");
+//    you may not use this file except in compliance with the License.
+//    You may obtain a copy of the License at
+//
+//        http://www.apache.org/licenses/LICENSE-2.0
+//
+//    Unless required by applicable law or agreed to in writing, software
+//    distributed under the License is distributed on an "AS IS" BASIS,
+//    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+//    See the License for the specific language governing permissions and
+//    limitations under the License.
+
 package mocks
 
 import (
Add missing license statement in mock file.

Signed-off-by: Maciej Mrowiec <0c5b7bd73e5de492f16f5039ef275802502d5ce2@gmail.com>
