def prep_db(): """ Prep the DB for testing. """ clean_db() # Create a new default config cripts_config = CRIPTsConfig() cripts_config.save() # Add Source handlers.add_new_source(TSRC, TRANDUSER) # Add User user = CRIPTsUser.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 cripts_config = CRIPTsConfig() cripts_config.save() # Add Source handlers.add_new_source(TSRC, TRANDUSER) # Add User user = CRIPTsUser.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))