コード例 #1
0
ファイル: tests.py プロジェクト: lakiw/cripts
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()
コード例 #2
0
ファイル: tests.py プロジェクト: lakiw/cripts
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()
コード例 #3
0
ファイル: tests.py プロジェクト: lakiw/cripts
 def AddSource(self):
     self.assertTrue(handlers.add_new_source(TSRC, TUSER_NAME))
コード例 #4
0
ファイル: tests.py プロジェクト: lakiw/cripts
 def AddSource(self):
     self.assertTrue(handlers.add_new_source(TSRC, TUSER_NAME))