def test_vxuserapi_many_samples():
kwargs = dict(count=4,
offset=2,
md5=None,
sha1="sha1",
sha256=None,
format_name=None,
format_group_name=None)
request = mock.Mock(return_value=[{"id": 23}, {"id": 42}])
with mock.patch("vxcube_api.raw_api.VxCubeApiRequest.request",
new=request):
api = VxCubeApi(base_url="http://test", version=2.0)
samples = api.samples(**kwargs)
assert isinstance(samples, list)
assert len(samples) == 2
assert samples[0].id == 23
assert samples[0]._raw_api is api._raw_api
assert samples[1].id == 42
assert samples[1]._raw_api is api._raw_api
request.assert_called_with(method="get",
url="http://test/api-2.0/samples",
params={},
headers={},
json={
"count": 4,
"offset": 2,
"sha1": "sha1"
})
def test_vxuserapi_one_sample():
request = mock.Mock(return_value={"id": 42})
with mock.patch("vxcube_api.raw_api.VxCubeApiRequest.request",
new=request):
api = VxCubeApi(base_url="http://test", version=2.0)
sample = api.samples(sample_id=42)
assert isinstance(sample, Sample)
assert sample.id == 42
assert sample._raw_api is api._raw_api
request.assert_called_with(method="get",
url="http://test/api-2.0/samples/42",
params={},
headers={})
