def test_basic(self, context, Task, get_build, get_archive_type, import_archive, work): task = mock.MagicMock() task.assertHost = mock.MagicMock() Task.return_value = task imgdata = { 'arch': 'x86_64', 'task_id': 1, 'files': [ 'some_file', ], 'rpmlist': [ ], } cursor = mock.MagicMock() context.cnx.cursor.return_value = cursor context.session.host_id = 42 get_build.return_value = { 'id': 2, 'name': 'name', 'version': 'version', 'release': 'release', } get_archive_type.return_value = 4 work.return_value = self.tempdir os.makedirs(self.tempdir + "/tasks/1/1") kojihub.importImageInternal(task_id=1, build_info=get_build.return_value, imgdata=imgdata)
def test_with_rpm(self, context, Task, get_build, get_archive_type, import_archive, build, work, get_rpm): task = mock.MagicMock() task.assertHost = mock.MagicMock() Task.return_value = task rpm = { #'location': 'foo', 'id': 6, 'name': 'foo', 'version': '3.1', 'release': '2', 'epoch': 0, 'arch': 'noarch', 'payloadhash': 'laksjdflkasjdf', 'size': 42, 'buildtime': 12345, } imgdata = { 'arch': 'x86_64', 'task_id': 1, 'files': [ 'some_file', ], 'rpmlist': [rpm], } cursor = mock.MagicMock() context.cnx.cursor.return_value = cursor context.session.host_id = 42 get_build.return_value = {'id': 2} get_rpm.return_value = rpm get_archive_type.return_value = 4 work.return_value = self.tempdir build.return_value = self.tempdir import_archive.return_value = { 'id': 9, 'filename': self.tempdir + '/foo.archive', } workdir = self.tempdir + "/tasks/1/1" os.makedirs(workdir) # Create a log file to exercise that code path with open(workdir + '/foo.log', 'w'): pass kojihub.importImageInternal(task_id=1, build_id=2, imgdata=imgdata) # Check that the log symlink made it to where it was supposed to. dest = os.readlink(workdir + '/foo.log') dest = os.path.abspath(os.path.join(workdir, dest)) self.assertEquals(dest, self.tempdir + '/data/logs/image/foo.log') # And.. check all the sql statements self.assertEquals(len(cursor.execute.mock_calls), 1) expression, kwargs = cursor.execute.mock_calls[0][1] expression = " ".join(expression.split()) expected = 'INSERT INTO archive_rpm_components (archive_id, rpm_id) ' + \ 'VALUES (%(archive_id)s, %(rpm_id)s)' self.assertEquals(expression, expected) self.assertEquals(kwargs, {'archive_id': 9, 'rpm_id': 6})
def test_with_rpm(self, context, Task, get_build, get_archive_type, import_archive, build, work, get_rpm): task = mock.MagicMock() task.assertHost = mock.MagicMock() Task.return_value = task rpm = { #'location': 'foo', 'id': 6, 'name': 'foo', 'version': '3.1', 'release': '2', 'epoch': 0, 'arch': 'noarch', 'payloadhash': 'laksjdflkasjdf', 'size': 42, 'buildtime': 12345, } imgdata = { 'arch': 'x86_64', 'task_id': 1, 'files': [ 'some_file', ], 'rpmlist': [rpm], } cursor = mock.MagicMock() context.cnx.cursor.return_value = cursor context.session.host_id = 42 get_build.return_value = {'id': 2} get_rpm.return_value = rpm get_archive_type.return_value = 4 work.return_value = self.tempdir build.return_value = self.tempdir import_archive.return_value = { 'id': 9, 'filename': self.tempdir + '/foo.archive', } workdir = self.tempdir + "/tasks/1/1" os.makedirs(workdir) # Create a log file to exercise that code path with open(workdir + '/foo.log', 'w'): pass kojihub.importImageInternal(task_id=1, build_id=2, imgdata=imgdata) # Check that the log symlink made it to where it was supposed to. dest = os.readlink(workdir + '/foo.log') dest = os.path.abspath(os.path.join(workdir, dest)) self.assertEquals(dest, self.tempdir + '/data/logs/image/foo.log') # And.. check all the sql statements self.assertEquals(len(cursor.execute.mock_calls), 1) expression, kwargs = cursor.execute.mock_calls[0][1] expression = " ".join(expression.split()) expected = 'INSERT INTO archive_rpm_components (archive_id, rpm_id) ' + \ 'VALUES (%(archive_id)s, %(rpm_id)s)' self.assertEquals(expression, expected) self.assertEquals(kwargs, {'archive_id': 9, 'rpm_id': 6})
def test_basic(self, context, Task, get_build, get_archive_type, import_archive, work): task = mock.MagicMock() task.assertHost = mock.MagicMock() Task.return_value = task imgdata = {"arch": "x86_64", "task_id": 1, "files": ["some_file"], "rpmlist": []} cursor = mock.MagicMock() context.cnx.cursor.return_value = cursor context.session.host_id = 42 get_build.return_value = {"id": 2} get_archive_type.return_value = 4 work.return_value = self.tempdir os.makedirs(self.tempdir + "/tasks/1/1") kojihub.importImageInternal(task_id=1, build_id=2, imgdata=imgdata)
def test_with_rpm(self, context, Task, get_build, get_archive_type, import_archive, build, work, get_rpm): task = mock.MagicMock() task.assertHost = mock.MagicMock() Task.return_value = task rpm = { #'location': 'foo', "id": 6, "name": "foo", "version": "3.1", "release": "2", "epoch": 0, "arch": "noarch", "payloadhash": "laksjdflkasjdf", "size": 42, "buildtime": 12345, } imgdata = {"arch": "x86_64", "task_id": 1, "files": ["some_file"], "rpmlist": [rpm]} cursor = mock.MagicMock() context.cnx.cursor.return_value = cursor context.session.host_id = 42 get_build.return_value = {"id": 2} get_rpm.return_value = rpm get_archive_type.return_value = 4 work.return_value = self.tempdir build.return_value = self.tempdir import_archive.return_value = {"id": 9, "filename": self.tempdir + "/foo.archive"} workdir = self.tempdir + "/tasks/1/1" os.makedirs(workdir) # Create a log file to exercise that code path with open(workdir + "/foo.log", "w"): pass kojihub.importImageInternal(task_id=1, build_id=2, imgdata=imgdata) # Check that the log symlink made it to where it was supposed to. dest = os.readlink(workdir + "/foo.log") self.assertEquals(dest, self.tempdir + "/data/logs/image/foo.log") # And.. check all the sql statements self.assertEquals(len(cursor.execute.mock_calls), 1) expression, kwargs = cursor.execute.mock_calls[0][1] expression = " ".join(expression.split()) expected = "INSERT INTO archive_rpm_components (archive_id,rpm_id) " + "VALUES (%(archive_id)i,%(rpm_id)i)" self.assertEquals(expression, expected) self.assertEquals(kwargs, {"archive_id": 9, "rpm_id": 6})
def test_basic(self, context, Task, get_build, get_archive_type, import_archive, work): imgdata = { 'arch': 'x86_64', 'task_id': 1, 'files': [ 'some_file', ], 'rpmlist': [ ], } cursor = mock.MagicMock() context.cnx.cursor.return_value = cursor context.session.host_id = 42 get_build.return_value = { 'id': 2, } get_archive_type.return_value = 4 work.return_value = self.tempdir os.makedirs(self.tempdir + "/tasks/1/1") kojihub.importImageInternal(task_id=1, build_id=2, imgdata=imgdata)