29 lines
738 B
Python
29 lines
738 B
Python
from unittest import TestCase
|
|
|
|
import shutil
|
|
import tempfile
|
|
|
|
from cloud_build import CB
|
|
|
|
|
|
class TestUtils(TestCase):
|
|
def setUp(self):
|
|
kwargs = {
|
|
'data_dir': tempfile.mkdtemp(prefix='cloud_build'),
|
|
'config': 'tests/minimal_config.yaml',
|
|
}
|
|
self.kwargs = kwargs
|
|
self.cb = CB(**kwargs)
|
|
|
|
def tearDown(self):
|
|
shutil.rmtree(self.kwargs['data_dir'])
|
|
|
|
def test_conver_size_lower_case(self):
|
|
self.assertEqual(self.cb.convert_size('200k'), '204800')
|
|
|
|
def test_conver_size_upper_case(self):
|
|
self.assertEqual(self.cb.convert_size('1M'), '1048576')
|
|
|
|
def test_conver_size_real(self):
|
|
self.assertEqual(self.cb.convert_size('0.1G'), '107374182')
|