def test_inject_mirrors_rhel(self): start_repo = ''' [delorean] name=delorean baseurl=https://trunk.rdoproject.org/centos7/some-repo-hash enabled=1 [rhel] name=rhel baseurl=https://some/stuff enabled=1 ''' expected = ''' [delorean] name=delorean baseurl=http://bar/centos7/some-repo-hash enabled=1 [rhel] name=rhel baseurl=http://foo/stuff enabled=1 ''' mock_args = mock.Mock(mirror='http://foo', rdo_mirror='http://bar', distro='rhel', old_mirror='https://some') result = main._inject_mirrors(start_repo, mock_args) self.assertEqual(expected, result)
def test_inject_mirrors_centos(self): start_repo = ''' [delorean] name=delorean baseurl=https://trunk.rdoproject.org/centos7/some-repo-hash enabled=1 [centos] name=centos baseurl=http://mirror.centos.org/centos/7/virt/$basearch/kvm-common enabled=1 ''' expected = ''' [delorean] name=delorean baseurl=http://bar/centos7/some-repo-hash enabled=1 [centos] name=centos baseurl=http://foo/centos/7/virt/$basearch/kvm-common enabled=1 ''' mock_args = mock.Mock(mirror='http://foo', rdo_mirror='http://bar', distro='centos', old_mirror='http://mirror.centos.org') result = main._inject_mirrors(start_repo, mock_args) self.assertEqual(expected, result)
def test_inject_mirrors_no_match(self): start_repo = ''' [delorean] name=delorean baseurl=https://some.mirror.com/centos7/some-repo-hash enabled=1 ''' mock_args = mock.Mock(rdo_mirror='http://some.mirror.com', distro='centos') # If a user has a mirror whose repos already point at itself then # the _inject_mirrors call should be a noop. self.assertEqual(start_repo, main._inject_mirrors(start_repo, mock_args))