Python Testing.Test Examples

Example #1

File: app.py Project: GameFreat/Website-risk-prediction-and-prevention

def predict(url):
    f = Features()
    features = f.Create_features(url)
    # print(features)
    test = Testing()
    r = test.Test(rules, features)
    label = eval(r[-1])
    d = dict(label)
    # print(d.keys())
    if not url:
        return ("Empty string")
    else:
        for key in d.keys():
            if (key == '0'):
                return (
                    "\nYou are safe!This website is not vulnerable to XSS attack\n"
                )
            else:
                return ("\nAlert!This website is vulnerable to XSS attack\n")

Example #2

File: prediction.py Project: GameFreat/Website-risk-prediction-and-prevention

            return 1
        else:
            return 0
    
    #.........Returning count of spcl char............
    def Spcl_char(self,url):
        spcl = re.findall("(<)|(>)|(/)|(=)",url)
        return len(spcl)

    #.........Create feature set...............
    def Create_features(self,url):
        features = [self.url_length(url),self.Keywords(url),self.url_count(url),self.Spcl_char(url)]
        return features


url = input('Enter the website URL : ')
f = Features()
features = f.Create_features(url)
#print(features)
test = Testing()
r = test.Test(rules,features)
label = eval(r[-1])
d = dict(label)
#print(d.keys())

for key in d.keys() :
    
    if(key == '0'):
        print("\nYou are safe!This website is not vulnerable to XSS attack\n")
    else:
        print("\nAlert!This website is vulnerable to XSS attack\n")