def testCredentials_user(self):
stored = QuickConfig.readConfiguration(QuickConfig.config_file_path)
cfg = QuickConfig.configForUser(stored["server"],
stored["standarduser"],
stored["standarduserpwd"], 1)
token = cfg.getToken(cfg.getUserCredential(UserType.CustomUser))
assert len(token) > 10
def testEmptyUser(self):
loCfg = None
try:
loCfg = QuickConfig.configForUser("https://", None, "2", False, 1)
except Exception as e:
pass
self.assertIsNone(loCfg)
try:
loCfg = QuickConfig.configForUser("https://", "", "2", False, 1)
except Exception as e:
pass
self.assertIsNone(loCfg)
def testDefaultAdminConfig(self):
loCfg = QuickConfig.fromConfigfile()
self.assertIsNotNone(
loCfg.getToken(loCfg.getUserCredential(UserType.PacteAdmin)))
self.assertIsNotNone(
loCfg.getToken(loCfg.getUserCredential(UserType.PSCAdmin)))
def testLinkUsers(self):
username1 = str(uuid.uuid4())
password1 = str(uuid.uuid4())
username2 = str(uuid.uuid4())
password2 = str(uuid.uuid4())
cfg = QuickConfig.fromConfigfile()
admin = Admin(cfg)
user1 = admin.createUser(username1, password1, str(uuid.uuid4()),
str(uuid.uuid4()))
user2 = admin.createUser(username2, password2, str(uuid.uuid4()),
str(uuid.uuid4()))
cfg.setCustomUser(username1, password1)
admin.removeContact(user2.userProfileId)
self.assertTrue(admin.addContact(user2.userProfileId))
self.assertTrue(admin.removeContact(user2.userProfileId))
admin.deleteUser(user1.userId)
self.assertIsNone(admin.checkUser(username1, password1))
cfg.setCustomUser(username2, password2)
admin.deleteUser(user2.userId)
self.assertIsNone(admin.checkUser(username2, password2))
def testCorpusLifeCycle(self):
corpus_manager = CorpusManager(QuickConfig.fromConfigfile())
new_corpus_name = str(uuid.uuid4())
# Creating new corpus
print("Creating new corpus... ")
corpus_id = corpus_manager.createCorpus(new_corpus_name, ["fr-fr"])
self.assertIsNotNone(corpus_id)
print("created")
# Populate
print("Adding document...")
doc_id = corpus_manager.addDocument(corpus_id, "bla bla bla", "bla",
None, "fr-fr")
self.assertIsNotNone(doc_id)
print("Added !")
# Create annotation group
print("Creating annotation group...")
group_id = corpus_manager.createBucket(corpus_id, str(uuid.uuid4()))
self.assertIsNotNone(group_id)
print("Created !")
# Register schema
print("Generate and add schema...")
feat_desc = FeatureDefinition("description", "description",
"description of description", "", True,
["noop"], False)
schemaTest = SchemaData(TARGET.document_surface1d, "TestSchema",
{"description": feat_desc})
schemaId = corpus_manager.registerSchema(schemaTest.to_string())
print("Created ! (" + schemaId + ")")
self.assertTrue(
corpus_manager.copySchemaToGroup(schemaId, corpus_id, group_id))
# Remove schema
print("Deleting schema " + schemaId + " ...")
corpus_manager.deleteSchema(schemaId)
print("Schema deleted!")
# TODO delete annotation group
# Remove document
time.sleep(1)
print("Getting document...")
self.assertTrue(
corpus_manager.getDocument(corpus_id, doc_id).title == "bla")
print("Document retrieved")
# Deleting corpus
print("Deleting corpus...")
self.assertTrue(corpus_manager.deleteCorpus(corpus_id))
r = corpus_manager.getcorpus_id(new_corpus_name)
self.assertIsNone(r)
print("Deleted !")
print("Done !")
def createTestingUser():
QuickConfig.config_file_path = "config.properties"
config = QuickConfig.fromConfigfile()
admin = Admin(config)
user = config.getUserCredential(UserType.CustomUser)
userId = admin.checkUser(user.username, user.password)
if not userId:
admin.createUser(user.username, user.password, "TestUser", "011")
userId = admin.checkUser(user.username, user.password)
return userId is not None
def setUp(self):
username1 = "*****@*****.**"
pswd1 = "secret"
username2 = "*****@*****.**"
pswd2 = "secret"
admin = Admin(QuickConfig.fromConfigfile())
id1 = admin.checkUser(username1, pswd1)
id2 = admin.checkUser(username2, pswd2)
if not id1:
admin.createUser(username1, pswd1, "testingUser1", "testingUser1")
if not id2:
admin.createUser(username2, pswd2, "testingUser2", "testingUser2")
def testExportCorpus(self):
corpus_manager = CorpusManager(QuickConfig.fromConfigfile())
corpus_id = createSmallCorpus(corpus_manager)
self.assertIsNotNone(corpus_id)
print(corpus_id)
print(corpus_manager.getSize(corpus_id))
export_completed = corpus_manager.exportToDisk(corpus_id,
self.tempdir.name)
print(export_completed)
self.assertTrue(export_completed)
self.assertTrue(len(list(Path(self.tempdir.name).iterdir())) > 0)
self.assertEqual(
2,
len(list(Path(self.tempdir.name).joinpath("documents").iterdir())))
def testCredentials(self):
loCfg = QuickConfig.configForAdminAndUser("https://", "1", "2", "3",
"4", "5", "6", False, 1, "")
self.assertIsNotNone((loCfg.credential.get(UserType.CustomUser)))
self.assertIsNotNone((loCfg.credential.get(UserType.PacteAdmin)))
self.assertIsNotNone((loCfg.credential.get(UserType.PSCAdmin)))
self.assertEqual("1", loCfg.credential.get(UserType.PSCAdmin).username)
self.assertEqual("2", loCfg.credential.get(UserType.PSCAdmin).password)
self.assertEqual("3",
loCfg.credential.get(UserType.PacteAdmin).username)
self.assertEqual("4",
loCfg.credential.get(UserType.PacteAdmin).password)
self.assertEqual("5",
loCfg.credential.get(UserType.CustomUser).username)
self.assertEqual("6",
loCfg.credential.get(UserType.CustomUser).password)