Add schema2 manifest and schema2 config, along with tests

image/docker/schema2/test/test_manifest.py

@@ -0,0 +1,87 @@
+import json
+import pytest
+
+from app import docker_v2_signing_key
+from image.docker.schema1 import DockerSchema1ManifestBuilder
+from image.docker.schema2.manifest import MalformedSchema2Manifest, DockerSchema2Manifest
+from image.docker.schema2.test.test_config import CONFIG_BYTES
+
+@pytest.mark.parametrize('json_data', [
+  '',
+  '{}',
+  """
+  {
+    "unknown": "key"
+  }
+  """,
+])
+def test_malformed_manifests(json_data):
+  with pytest.raises(MalformedSchema2Manifest):
+    DockerSchema2Manifest(json_data)
+
+
+MANIFEST_BYTES = json.dumps({
+  "schemaVersion": 2,
+  "mediaType": "application/vnd.docker.distribution.manifest.v2+json",
+  "config": {
+    "mediaType": "application/vnd.docker.container.image.v1+json",
+    "size": 7023,
+    "digest": "sha256:b5b2b2c507a0944348e0303114d8d93aaaa081732b86451d9bce1f432a537bc7"
+  },
+  "layers": [
+    {
+      "mediaType": "application/vnd.docker.image.rootfs.foreign.diff.tar.gzip",
+      "size": 1234,
+      "digest": "sha256:ec4b8955958665577945c89419d1af06b5f7636b4ac3da7f12184802ad867736",
+      "urls": ['http://some/url'],
+    },
+    {
+      "mediaType": "application/vnd.docker.image.rootfs.diff.tar.gzip",
+      "size": 32654,
+      "digest": "sha256:e692418e4cbaf90ca69d05a66403747baa33ee08806650b51fab815ad7fc331f"
+    },
+    {
+      "mediaType": "application/vnd.docker.image.rootfs.diff.tar.gzip",
+      "size": 16724,
+      "digest": "sha256:3c3a4604a545cdc127456d94e421cd355bca5b528f4a9c1905b15da2eb4a4c6b"
+    },
+    {
+      "mediaType": "application/vnd.docker.image.rootfs.diff.tar.gzip",
+      "size": 73109,
+      "digest": "sha256:ec4b8955958665577945c89419d1af06b5f7636b4ac3da7f12184802ad867736"
+    },
+  ],
+})
+
+def test_valid_manifest():
+  manifest = DockerSchema2Manifest(MANIFEST_BYTES)
+  assert manifest.config.size == 7023
+  assert str(manifest.config.digest) == 'sha256:b5b2b2c507a0944348e0303114d8d93aaaa081732b86451d9bce1f432a537bc7'
+  
+  assert len(manifest.layers) == 4
+  assert manifest.layers[0].is_remote
+  assert manifest.layers[0].size == 1234
+  assert str(manifest.layers[0].digest) == 'sha256:ec4b8955958665577945c89419d1af06b5f7636b4ac3da7f12184802ad867736'
+  assert manifest.layers[0].urls
+
+  assert manifest.leaf_layer == manifest.layers[3]
+  assert not manifest.leaf_layer.is_remote
+  assert manifest.leaf_layer.size == 73109
+
+
+def test_build_schema1():
+  manifest = DockerSchema2Manifest(MANIFEST_BYTES)
+
+  builder = DockerSchema1ManifestBuilder('somenamespace', 'somename', 'sometag')
+  manifest.populate_schema1_builder(builder, lambda digest: CONFIG_BYTES)
+  schema1 = builder.build(docker_v2_signing_key)
+
+  assert len(schema1.layers) == len(manifest.layers)
+  assert set(schema1.image_ids) == set([l.v1_id for l in manifest.layers_with_v1_ids])
+  assert set(schema1.parent_image_ids) == set([l.v1_parent_id for l in manifest.layers_with_v1_ids if l.v1_parent_id])
+
+  manifest_layers = list(manifest.layers_with_v1_ids)
+  for index, layer in enumerate(schema1.layers):
+    assert layer.digest == manifest_layers[index].layer.digest
+    assert layer.v1_metadata.image_id == manifest_layers[index].v1_id
+    assert layer.v1_metadata.parent_image_id == manifest_layers[index].v1_parent_id