#!/usr/bin/python

# Author: Athmane Madjoudj <athmanem@gmail.com>

#         Karanbir Singh <kbsingh@karan.org>

# Test default CentOS repos

# Note: since the -qa and CI setup will modify the

#       local repos, we need to run this tests

#       before those changes are made

import yum

import sys 

import datetime

import os

yb = yum.YumBase()

try:

    fasttrack = int(os.environ['FASTTRACK'])

except KeyError:

    fasttrack = 0

try:

    centosplus = int(os.environ['CENTOSPLUS'])

except KeyError:

    centosplus = 0

if fasttrack:

    centos_default_repos = ['base','extras','updates','cr','fasttrack','centos-kernel']

elif centosplus:

    centos_default_repos = ['base','extras','updates','cr','centosplus','centos-kernel']

else:

    centos_default_repos = ['base','extras','updates','cr','centos-kernel']

now = lambda: datetime.datetime.today().strftime("%c")

print "[+] %s -> Check if non default repo is enabled" % now() 

for repo in yb.repos.listEnabled():

    if not repo.id in centos_default_repos:

        print '%s is enabled, should be disabled at this stage' % repo.id

        print '[+] %s -> FAIL' % now()

        sys.exit(1)

print '[+] %s -> PASS' % now()

sys.exit(0)