def test_upload_file__should_upload_file(self):
self.sdk.set_device_file_data.return_value = None
obj = ZKAccess(connstr=self.connstr, device_model=ZK400)
data_stream = io.BytesIO(b'file_data!')
obj.upload_file('test_file.dat', data_stream)
self.sdk.set_device_file_data.assert_called_once_with(
'test_file.dat', b'file_data!', 10)
def test_upload_file__should_preserve_stream_pointer_position(self):
self.sdk.set_device_file_data.return_value = None
obj = ZKAccess(connstr=self.connstr, device_model=ZK400)
data_stream = io.BytesIO(b'file_data!')
data_stream.seek(3)
obj.upload_file('test_file.dat', data_stream)
self.sdk.set_device_file_data.assert_called_once_with(
'test_file.dat', b'e_data!', 7)
assert data_stream.tell() == 3
