def get_gadget_snap():
model = next(decode(Snapd().get_assertions('model')), None)
if model:
# convert to resource to handle presence of track info
resource = model_to_resource(model)
return resource.get('gadget')
return None
def test_model_to_resource_focal_server(self):
assertion_stream = create_mock_response(model_focal_server)
iter = decode(assertion_stream)
a = next(iter)
correct_resource = {
'type':
'model',
'authority-id':
'generic',
'brand-id':
'generic',
'model':
'generic-classic',
'sign-key-sha3-384':
'd-JcZF9nD9eBw7bwMnH61x-bklnQOhQud1Is6o_cn2wTj8EYDi9musrIT9z2MdAa'
}
self.assertDictEqual(correct_resource, model_to_resource(a))
self.assertNotIn('kernel', model_to_resource(a))
with self.assertRaises(StopIteration):
next(iter)
def test_model_to_resource_uc16_brandstore(self):
assertion_stream = create_mock_response(model_uc16_brandstore)
iter = decode(assertion_stream)
a = next(iter)
correct_resource = {
'type': 'model',
'authority-id': 'acmeinc',
'brand-id': 'acmeinc',
'model': 'devicemodel',
'sign-key-sha3-384':
'aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa',
'architecture': 'amd64',
'gadget': 'device',
'kernel': 'device-kernel',
'store': 'aaaaabbbbbccccc12345'
}
self.assertDictEqual(correct_resource, model_to_resource(a))
with self.assertRaises(StopIteration):
next(iter)
def test_model_to_resource_uc18(self):
assertion_stream = create_mock_response(model_uc18)
iter = decode(assertion_stream)
a = next(iter)
correct_resource = {
'type': 'model',
'authority-id': 'canonical',
'brand-id': 'canonical',
'model': 'ubuntu-core-18-amd64',
'architecture': 'amd64',
'base': 'core18',
'sign-key-sha3-384':
'9tydnLa6MTJ-jaQTFUXEwHl1yRx7ZS4K5cyFDhYDcPzhS7uyEkDxdUjg9g08BtNn',
'kernel': 'pc-kernel',
'kernel_track': '18',
'gadget': 'pc',
'gadget_track': '18'
}
self.assertDictEqual(correct_resource, model_to_resource(a))
with self.assertRaises(StopIteration):
next(iter)