add Tests and cleanup testing issues

This commit is contained in:
Evan J Felix 2017-03-17 13:10:51 -07:00
parent 564a82ca89
commit c78ebd0d85
4 changed files with 49 additions and 6 deletions

View file

@ -5,6 +5,13 @@ import sys
import util
def is_on_branch(protected):
retval = False
branch = util.cmd_output('git', 'symbolic-ref', 'HEAD')
chunks = branch.strip().split('/')
if chunks[2] == protected:
retval = True
return retval
def main(argv=None):
parser = argparse.ArgumentParser()
@ -12,12 +19,10 @@ def main(argv=None):
parser.add_argument('filenames', nargs='*', help='filenames to check.')
args = parser.parse_args(argv)
retval = 0
branch = util.cmd_output('git', 'symbolic-ref', 'HEAD')
chunks = branch.strip().split('/')
if chunks[2] == args.b:
retval = -1
return retval
if is_on_branch(args.b):
return 1
else:
return 0
if __name__ == '__main__':