def test_put_creds_in_file(self):
testfile = '~/test_picas'
pc2 = pc.PicasCred(usr=self.d_usr, pwd=self.d_pwd, dbn=self.d_dbn)
pc2.put_creds_in_file(testfile)
pc3 = pc.PicasCred(source=testfile)
self.assertTrue(pc3.user == self.d_usr)
self.assertTrue(pc3.password == self.d_pwd)
self.assertTrue(pc3.database == self.d_dbn)
self.orig_pc.put_creds_in_file()
def setUp(self):
##Saves the original credentials
if os.environ['PICAS_USR'] != "":
self.orig_dest = 'env'
self.orig_pc = pc.PicasCred(usr=os.environ['PICAS_USR'],
pwd=os.environ['PICAS_USR_PWD'],
dbn=os.environ['PICAS_DB'])
else:
self.orig_pc = pc.PicasCred(source='~/.picasrc')
self.orig_dest = 'file'
self.d_usr = "******"
self.d_pwd = "dammy"
self.d_dbn = "dommy"
def setUp(self):
vers=str(sys.version_info[0])+"."+str(sys.version_info[1])
if vers == '2.6':
vers = '2.6.0'
self.token_id='travis_getSBX_test'+vers
pc=gpc.PicasCred()
creds=pc.return_credentials()
self.usr=creds['user']
self.pwd=creds['password']
self.dbn=str('sksp_unittest')
self.test_tokvarile='GRID_PiCaS_Launcher/tests/test_tok_var.json'
server = couchdb.Server("https://picas-lofar.grid.surfsara.nl:6984")
server.resource.credentials = (self.usr,self.pwd)
self.db= server[self.dbn]
variable_data ={"_token_keys":{
"$STRING1":"string1",
"$TOKEN":"_id",
"$INT1":"integer1"},
"_attachments":{
"test_attachment":"test_attachment",
"ATTACH":"test_attachment2"}}
os.chdir(BASE_DIR)
os.chdir('..')
with open(self.test_tokvarile,'w') as f:
json.dump(variable_data, f)
set_token_field(self.token_id,'string1','test_string',pc)
set_token_field(self.token_id,'integer1',1234,pc)
if os.path.isfile('test_attachment'): os.remove('test_attachment')
def test_missing_tokvar(self):
from GRID_PiCaS_Launcher.tok_to_bash import export_tok_keys
pc = gpc.PicasCred(usr=self.usr, pwd=self.pwd, dbn=self.dbn)
try:
export_tok_keys("xtokvar.json", {"_id": self.token})
except Exception as e:
self.assertTrue("tokvar read error" in str(e))
self.assertTrue(get_token_field(self.token, "output", pc) == -2)
def test_put_creds_in_env(self):
os.environ['PICAS_USR'] = ""
os.environ['PICAS_USR_PWD'] = ""
os.environ['PICAS_DB'] = ""
pc2 = pc.PicasCred(usr=self.d_usr, pwd=self.d_pwd, dbn=self.d_dbn)
pc2.put_picas_creds_in_env()
self.assertTrue(os.environ['PICAS_USR'] == self.d_usr)
self.assertTrue(os.environ['PICAS_USR_PWD'] == self.d_pwd)
self.assertTrue(os.environ['PICAS_DB'] == self.d_dbn)
def test_readenv_vars(self):
os.environ['PICAS_USR'] = self.d_usr
os.environ['PICAS_USR_PWD'] = self.d_pwd
os.environ['PICAS_DB'] = self.d_dbn
pc1 = pc.PicasCred()
pc1.get_picas_creds_from_env()
self.assertTrue(pc1.user == self.d_usr)
self.assertTrue(pc1.password == self.d_pwd)
self.assertTrue(pc1.database == self.d_dbn)
def test_readfile_vars(self):
testfile = '~/test_picas'
with open(expanduser(testfile), 'w') as tf:
tf.write('user='******'password='******'database=' + self.d_dbn + "\n")
pc2 = pc.PicasCred(source=testfile)
pc2.get_picas_creds_from_file(testfile)
self.assertTrue(pc2.user == self.d_usr)
self.assertTrue(pc2.password == self.d_pwd)
self.assertTrue(pc2.database == self.d_dbn)
def setUp(self):
vers = str(sys.version_info[0]) + "." + str(sys.version_info[1])
if vers == '2.6':
vers = '2.6.0'
self.token_id = 'travis_getSBX_test' + vers
pc = gpc.PicasCred()
creds = pc.return_credentials()
self.usr = creds['user']
self.pwd = creds['password']
self.dbn = str('sksp_unittest')
server = couchdb.Server("https://picas-lofar.grid.surfsara.nl:6984")
server.resource.credentials = (self.usr, self.pwd)
def test_scrub(self):
pc = gpc.PicasCred(usr=self.usr, pwd=self.pwd, dbn=self.dbn)
scrubs = get_token_field(self.token, "scrub_count", pc)
self.client.modify_token(self.modifier.scrub(self.db[self.token]))
self.assertTrue(
scrubs + 1 == int(get_token_field(self.token, "scrub_count", pc))
)
_ = self.nestedmodifier.scrub(self.token, self.client.db)
self.assertTrue(
scrubs + 2 == int(get_token_field(self.token, "scrub_count", pc))
)
set_token_field(self.token, "scrub_count", scrubs, pc)
def test_failed_sbx(self):
pc = gpc.PicasCred(usr=self.usr, dbn=self.dbn, pwd=self.pwd)
set_token_field(
self.token,
"SBXloc",
"ftp://ftp.strw.leidenuniv.nl/pub/apmechev/travis_ci_tests/sanddbox_travis.tar",
pc,
)
tok = self.db[self.token]
try:
self.Ex.process_token(self.token, tok)
except Exception as e:
self.assertTrue(str(e) == "Sandbox failed to download!")
def tearDown(self):
pc = gpc.PicasCred(usr=self.usr, dbn=self.dbn, pwd=self.pwd)
set_token_field(self.token, "lock", 0, pc)
set_token_field(self.token, "done", 0, pc)
set_token_field(self.token, "status", "todo", pc)
set_token_field(self.token, "hostname", "", pc)
self.client.modify_token(
self.modifier.add_output(self.db[self.token], {"output": 0})
)
self.assertTrue(get_token_field(self.token, "output", pc) == 0)
set_token_field(self.token, "string1", "1234", pc)
self.find_and_delete("png")
self.client.modify_token(self.modifier.unlock(self.db[self.token]))
self.client.modify_token(self.modifier.unclose(self.db[self.token]))
update_status(self.dbn, self.usr, self.pwd, self.token, "done")
def setUp(self):
vers = str(sys.version_info[0]) + "." + str(sys.version_info[1])
if vers == "2.6":
vers = "2.6.0"
self.t_type = "travis_ci_test" + vers
pc = gpc.PicasCred()
creds = pc.return_credentials()
self.usr = creds["user"]
self.pwd = creds["password"]
self.dbn = "sksp_unittest"
sys.argv = ["dummy", self.dbn, self.usr, self.pwd]
self.token = "travis_getSBX_test" + vers
server = couchdb.Server("https://picas-lofar.grid.surfsara.nl:6984")
self.client = CouchClient(
url="https://picas-lofar.grid.surfsara.nl:6984",
db=self.dbn,
username=self.usr,
password=self.pwd,
)
server.resource.credentials = (self.usr, self.pwd)
self.db = server[self.dbn]
tok = {"type": self.token}
vers = ".".join([str(i) for i in sys.version_info])
tok["_id"] = self.token
if "_attachments" in tok.keys():
_ = tok.pop("_attachments")
self.db.update([tok])
set_token_field(self.token, "lock", 0, pc)
set_token_field(self.token, "string1", "1234", pc)
set_token_main(self.token, "string1", "1234", self.dbn, self.usr, self.pwd)
set_token_main(self.token, "string1", "1234")
set_token_field(self.token, "done", 0, pc)
update_status(self.dbn, self.usr, self.pwd, self.token, "todo")
set_token_field(
self.token,
"SBXloc",
"https://home.strw.leidenuniv.nl/~apmechev/sandbox_travis.tar",
pc,
)
self.modifier = BasicTokenModifier()
iterator = BasicViewIterator(self.client, self.token + "/todo", self.modifier)
self.TestActor = TestActor(iterator, self.modifier)
self.Ex = ExampleActor(iterator, self.modifier)
self.nestedmodifier = NestedTokenModifier()
def test_lock_token(self):
pc = gpc.PicasCred(usr=self.usr, pwd=self.pwd, dbn=self.dbn)
self.assertTrue(get_token_field(self.token, "lock", pc) == 0)
try:
self.TestActor.run()
except Exception as e:
self.assertTrue(e.args[0] == self.token)
self.key = e.args[0]
self.tok = e.args[1]
self.assertTrue(get_token_field(self.token, "lock", pc) > 0)
set_token_field(self.token, "lock", 0, pc)
self.nestedmodifier.lock(self.token, self.client.db)
self.assertTrue(get_token_field(self.token, "lock", pc) > 0)
self.nestedmodifier.unlock(self.token, self.client.db)
self.assertTrue(get_token_field(self.token, "lock", pc) == 0)
self.nestedmodifier.close(self.token, self.client.db)
self.assertTrue(get_token_field(self.token, "done", pc) > 0)
# self.nestedmodifier.add_output(self.token, self.client.db, 12) #This doesn't work??
# self.assertTrue(get_token_field(self.token,'output', pc)==12)
self.nestedmodifier.set_error(self.token, self.client.db)
self.assertTrue(get_token_field(self.token, "done", pc) == -1)
self.assertTrue(get_token_field(self.token, "lock", pc) == -1)
self.nestedmodifier.unclose(self.token, self.client.db)
self.assertTrue(get_token_field(self.token, "done", pc) == 0)
