def prep_db(): """ Prep database for test. """ clean_db() # Add User user = CRITsUser.create_user(username=TUSER_NAME, password=TUSER_PASS, email=TUSER_EMAIL) user.save() # Add Source add_new_source(TSRC, TUSER_NAME) # Add Data Type add_new_signature_type(TDT, TUSER_NAME)
def prep_db(): """ Prep database for test. """ clean_db() # Add Source add_new_source(TSRC, "RandomUser") # Add User user = CRITsUser.create_user( username=TUSER_NAME, password=TUSER_PASS, email=TUSER_EMAIL, ) user.save()
def prep_db(): """ Prep the DB for testing. """ clean_db() # Create a new default config crits_config = CRITsConfig() crits_config.save() # Add Source handlers.add_new_source(TSRC, TRANDUSER) # Add User user = CRITsUser.create_user(username=TUSER_NAME, password=TUSER_PASS, email=TUSER_EMAIL) user.first_name = TUSER_FNAME user.last_name = TUSER_LNAME user.save() # Add test source object '''
def prep_db(): """ Prep the DB for testing. """ clean_db() # Create a new default config crits_config = CRITsConfig() crits_config.save() # Add Source handlers.add_new_source(TSRC, TRANDUSER) # Add User user = CRITsUser.create_user(username=TUSER_NAME, password=TUSER_PASS, email=TUSER_EMAIL ) user.first_name = TUSER_FNAME user.last_name = TUSER_LNAME user.save() # Add test source object '''
def prep_db(): """ Prep the DB for testing. """ clean_db() # Create a new default config crits_config = CRITsConfig() crits_config.save() # Add Source handlers.add_new_source(TSRC, TRANDUSER) # Add User user = CRITsUser.create_user( username=TUSER_NAME, password=TUSER_PASS, email=TUSER_EMAIL, ) user.first_name = TUSER_FNAME user.last_name = TUSER_LNAME user.save() # Add test source object obj = TestSourceObject() obj.name = TOBJS_NAME obj.value = TOBJS_VALUE obj.add_source(source=TSRC, analyst=TUSER_NAME) obj.save() # Add another with Different source obj = TestSourceObject() obj.name = TOBJS_NAME obj.value = TOBJS_VALUE obj.add_source(source=TUNKSRC, analyst=TRANDUSER) obj.save() # Add test non-source object obj = TestObject() obj.name = TOBJ_NAME obj.value = TOBJ_VALUE obj.save()
def prep_db(): """ Prep the DB for testing. """ clean_db() # Create a new default config crits_config = CRITsConfig() crits_config.save() # Add Source handlers.add_new_source(TSRC, TRANDUSER) # Add User user = CRITsUser.create_user(username=TUSER_NAME, password=TUSER_PASS, email=TUSER_EMAIL, ) user.first_name = TUSER_FNAME user.last_name = TUSER_LNAME user.save() # Add test source object obj = TestSourceObject() obj.name = TOBJS_NAME obj.value = TOBJS_VALUE obj.add_source(source=TSRC, analyst=TUSER_NAME) obj.save() # Add another with Different source obj = TestSourceObject() obj.name = TOBJS_NAME obj.value = TOBJS_VALUE obj.add_source(source=TUNKSRC, analyst=TRANDUSER) obj.save() # Add test non-source object obj = TestObject() obj.name = TOBJ_NAME obj.value = TOBJ_VALUE obj.save()
def AddSource(self): self.assertTrue(handlers.add_new_source(TSRC, TUSER_NAME))