def test_token_auth(self):
token = self._connector._token
conn1 = newt.ClientNEWTConnector(self._site)
self.assertTrue(conn1.auth(username=self._username,
token=token))
self.assertTrue(conn1.status())
conn2 = newt.ClientNEWTConnector(self._site)
self.assertFalse(conn2.auth(username="******",
password="******"))
conn3 = newt.ClientNEWTConnector(self._site)
self.assertFalse(conn3.auth(username=self._username,
token="Loren Ipsum"))
def test_homedir(self):
self.assertIn(self._username, self._connector.get_home_dir())
new_connector = newt.ClientNEWTConnector(self._site)
self.assertTrue(new_connector.auth(username=self._username,
password=self._password,
home_dir="/tmp"))
self.assertEqual(new_connector.get_home_dir(), "/tmp")
def setUp(self):
self._username = os.getenv("QDO_NEWT_USER", None)
self._password = os.getenv("QDO_NEWT_PASSWORD", None)
self._site = os.getenv("QOD_NEWT_SITE", "edison")
if self._username is None:
raise ValueError, "Missing QDO_NEWT_USER env var"
if self._password is None:
raise ValueError, "Missing QDO_NEWT_PASSWORD env var"
self._connector = newt.ClientNEWTConnector(self._site)
self.assertTrue(self._connector.auth(username=self._username,
password=self._password))
def valid_queue_name_dict(value):
start_t = time.time()
return_value, out = client.do_remote_call("qdo",
"valid_queue_name",
args={"name": value})
run_time = time.time() - start_t
#print out
print "Total Execution time of call valid_queue_name (s):", run_time
return return_value
# Creation of a connector with Newt capacities,
connector = newt.ClientNEWTConnector(argv[1])
#connector.set_tmp_at_home_dir("private_qdo")
#connector.set_sremote_dir("/global/u1/g/gprodri/shared_qdo")
# Authentication, this call connects remotely and retrieves the default
# after login directory.
start_t = time.time()
if connector.auth(argv[2], argv[3]):
run_time = time.time() - start_t
print "Auth time (s):", run_time
# Client class that uses the connector to do the remote calls.
start_t = time.time()
connector.do_self_discovery()
run_time = time.time() - start_t
print "Self discovery time (s):", run_time
client = remote.RemoteClient(connector)
print valid_queue_name("juanito")