Exemplo n.º 1
0
    def test_uri(self):
        '''test_uri will make sure that the endpoint returns the equivalent
        image for all different uri options
        '''
        from shub.api import get_image_name, get_manifest
        manifest = get_manifest(image=self.image_id)
        image_name = get_image_name(manifest)

        print("Case 1: ask for image and ask for master branch (tag)")
        manifest = get_manifest(image="%s/%s:master" %
                                (self.user_name, self.repo_name))
        self.assertEqual(image_name, get_image_name(manifest))

        print("Case 2: ask for different tag (mongo)")
        manifest = get_manifest(image="%s/%s:mongo" %
                                (self.user_name, self.repo_name))
        mongo = get_image_name(manifest)
        self.assertFalse(image_name == mongo)

        print(
            "Case 3: ask for image without tag (should be latest across tags, mongo)"
        )
        manifest = get_manifest(image="%s/%s" %
                                (self.user_name, self.repo_name))
        self.assertEqual(mongo, get_image_name(manifest))

        print(
            "Case 4: ask for latest tag (should be latest across tags, mongo)")
        manifest = get_manifest(image="%s/%s:latest" %
                                (self.user_name, self.repo_name))
        self.assertEqual(mongo, get_image_name(manifest))
Exemplo n.º 2
0
    def test_download_image(self):
        '''test_download_image will ensure that an image is downloaded to an
        appropriate location (tmpdir) or cache
        '''
        from shub.api import download_image, get_manifest
        print("Case 1: Specifying a directory downloads to it")
        manifest = get_manifest(image=self.image_id)
        image = download_image(manifest, download_folder=self.tmpdir)
        self.assertEqual(os.path.dirname(image), self.tmpdir)

        print("Case 2: Image should be named based on commit.")
        image_name = os.path.splitext(os.path.basename(image))[0]
        self.assertEqual(image_name, manifest['version'])
        os.remove(image)

        print("Case 3: Not specifying a directory downloads to PWD")
        os.chdir(self.tmpdir)
        image = download_image(manifest)
        self.assertEqual(os.getcwd(), self.tmpdir)
        self.assertTrue(image in glob("*"))
        os.remove(image)

        print("Case 4: Image should not be extracted.")
        image = download_image(manifest, extract=False)
        self.assertTrue(image.endswith('.img.gz'))
Exemplo n.º 3
0
 def test_get_manifest(self):
     '''test_get_manifest should return the shub manifest
     '''
     from shub.api import get_manifest
     print("Case 1: Testing retrieval of singularity-hub manifest")
     manifest = get_manifest(self.image_id)
     keys = ['files', 'name', 'image', 'collection', 
             'id', 'version', 'spec']
     [self.assertTrue(x in manifest) for x in keys]
     self.assertTrue(manifest['id']==self.image_id)
Exemplo n.º 4
0
 def test_get_manifest(self):
     '''test_get_manifest should return the shub manifest
     '''
     from shub.api import get_manifest
     print("Case 1: Testing retrieval of singularity-hub manifest")
     manifest = get_manifest(self.image_id)
     keys = [
         'files', 'name', 'image', 'collection', 'id', 'version', 'spec'
     ]
     [self.assertTrue(x in manifest) for x in keys]
     self.assertTrue(manifest['id'] == self.image_id)
Exemplo n.º 5
0
    def test_get_image_name(self):
        '''test_get_image_name will return the image name from the manifest
        '''
        from shub.api import get_image_name, get_manifest
        manifest = get_manifest(image_id=self.image_id)

        print("Case 1: return an image name using the commit id")
        image_name = get_image_name(manifest)
        self.assertEqual('f57e631a0434c31f0b4fa5276a314a6d8a672a55.img.gz',
                         image_name)

        print("Case 2: ask for invalid extension")
        with self.assertRaises(SystemExit) as cm:
            image_name = get_image_name(manifest, extension='.bz2')
        self.assertEqual(cm.exception.code, 1)

        print("Case 3: don't use commit (use md5 sum on generation)")
        image_name = get_image_name(manifest, use_commit=False)
        print(image_name)
        self.assertEqual('be4b9ba8fc22525d2ee2b27846513d42.img.gz', image_name)
Exemplo n.º 6
0
    def test_get_image_name(self):
        '''test_get_image_name will return the image name from the manifest
        '''
        from shub.api import get_image_name, get_manifest
        manifest = get_manifest(image=self.image_id)

        print("Case 1: return an image name using the commit id")
        image_name = get_image_name(manifest)
        self.assertEqual('6d3715a982865863ff20e8783014522edf1240e4.img.gz',
                         image_name)

        print("Case 2: ask for invalid extension")
        with self.assertRaises(SystemExit) as cm:
            image_name = get_image_name(manifest, extension='.bz2')
        self.assertEqual(cm.exception.code, 1)

        print("Case 3: don't use commit (use md5 sum on generation)")
        image_name = get_image_name(manifest, use_commit=False)
        print(image_name)
        self.assertEqual('9e46ba8be1e10b1a2812844ac8072259.img.gz', image_name)
Exemplo n.º 7
0
    def test_download_image(self):
        '''test_download_image will ensure that an image is downloaded to an
        appropriate location (tmpdir) or cache
        '''
        from shub.api import download_image, get_manifest
        print("Case 1: Specifying a directory downloads to it")
        manifest = get_manifest(image_id=self.image_id)
        image = download_image(manifest,
                               download_folder=self.tmpdir)
        self.assertEqual(os.path.dirname(image),self.tmpdir)
        os.remove(image)

        print("Case 2: Not specifying a directory downloads to PWD")
        os.chdir(self.tmpdir)
        image = download_image(manifest)
        self.assertEqual(os.getcwd(),self.tmpdir)
        os.remove(image)

        print("Case 3: Image should not be extracted.")
        image = download_image(manifest,extract=False)
        self.assertTrue(image.endswith('.img.gz'))        
Exemplo n.º 8
0
    def test_get_image_name(self):
        '''test_get_image_name will return the image name from the manifest
        '''
        from shub.api import get_image_name, get_manifest
        manifest = get_manifest(image_id=self.image_id)
        
        print("Case 1: return an image name using the commit id")
        image_name = get_image_name(manifest)
        self.assertEqual('f57e631a0434c31f0b4fa5276a314a6d8a672a55.img.gz',
                         image_name)

        print("Case 2: ask for invalid extension")
        with self.assertRaises(SystemExit) as cm:
            image_name = get_image_name(manifest,
                                        extension='.bz2')
        self.assertEqual(cm.exception.code, 1)

        print("Case 3: don't use commit (use md5 sum on generation)")
        image_name = get_image_name(manifest,
                                    use_commit=False)
        print(image_name)
        self.assertEqual('be4b9ba8fc22525d2ee2b27846513d42.img.gz',image_name)