def test_all(self):
gen = DocumentGenerator()
def gen_contact():
return {'email': gen.email(), 'phone': gen.phone()}
def gen_rating():
return random.uniform(1, 5)
template = {
'id': 'guid',
'status': ['online', 'offline', 'dnd', 'anonymous'],
'age': 'small_int',
'homepage': 'url',
'name': 'name',
'headline': 'sentence',
'about': 'paragraph',
'email': {
'typemap': 'email',
'unique': True
},
'rating': gen_rating,
'contact': gen_contact,
'favpost': [gen.slug() for n in range(1000)]
}
gen.set_template(template)
documents = gen.documents(10)
print(documents[0])
self.assertEqual(len(documents), 10)
def generate_Documents():
gen = DocumentGenerator()
gen.init_word_cache(5)
gen.init_sentence_cache(5)
gen.word_cache = ['Is', 'this', 'working', 'pls', 'work']
gen.sentence_cache = [
'This is a great step in the right direction',
'Every moment is the paradox of now or never',
'World will judge you on this moment'
]
print(gen.word_cache)
print(gen.sentence_cache)
template = {
'id': 'guid',
'status': ['online', 'offline', 'dnd', 'anonymous'],
'age': 'small_int',
'homepage': 'url',
'name': 'name',
'headline': 'sentence',
'text': 'paragraph'
}
gen.set_template(template)
documents = gen.documents(10)
print(documents[0])
def test_template(self):
gen = DocumentGenerator()
template = {
'id': 'guid',
'status': ['online', 'offline', 'dnd', 'anonymous'],
'age': 'small_int',
'homepage': 'url',
'name': 'name',
'headline': 'sentence',
'about': 'paragraph'
}
gen.set_template(template)
documents = gen.documents(10)
self.assertEqual(len(documents), 10)
def gen_bugs(count):
gen = DocumentGenerator()
template = {
'_id': get_id,
'title': 'sentence',
'severity': [1, 2, 3, 4, 5, 6],
'engineer': get_name,
'headline': 'sentence',
'description': 'paragraph',
'url': 'url',
'age': 'small_int',
'component': ['frontend', 'backend', 'middleware', 'infra', 'others'],
}
gen.set_template(template)
documents = gen.documents(count)
return documents
def test_nested_field(self):
gen = DocumentGenerator()
def gen_contact():
return {'email': gen.email(), 'phone': gen.phone()}
template = {
'id': 'guid',
'status': ['online', 'offline', 'dnd', 'anonymous'],
'age': 'small_int',
'homepage': 'url',
'name': 'name',
'headline': 'sentence',
'about': 'paragraph',
'contact': gen_contact
}
gen.set_template(template)
documents = gen.documents(10)
self.assertEqual(len(documents), 10)
def test_custom_field(self):
gen = DocumentGenerator()
def gen_rating():
return random.uniform(1, 5)
template = {
'id': 'guid',
'status': ['online', 'offline', 'dnd', 'anonymous'],
'age': 'small_int',
'homepage': 'url',
'name': 'name',
'headline': 'sentence',
'about': 'paragraph',
'rating': gen_rating
}
gen.set_template(template)
documents = gen.documents(10)
self.assertEqual(len(documents), 10)
def make_users(count=1):
from essential_generators import DocumentGenerator
template = {
'_id': 'guid',
'login': {
'typemap': 'email',
'unique': True
},
'fname': 'word',
'mname': 'word',
'lname': 'word',
'email': 'email',
'dob': 'large_int',
'tags': ['admin', 'buildteam', 'dev', 'mgr'],
'desc': 'sentence'
}
gen = DocumentGenerator()
gen.set_template(template)
return gen.documents(count)
#!/usr/bin/python
import sys
import random
from essential_generators import DocumentGenerator
gen = DocumentGenerator()
N = int(sys.argv[1]) if (len(sys.argv) > 1 and sys.argv[1].isdigit()) else 10
def gen_partNumber():
partId = random.randrange(1000, 9999)
partCode = ''
while not (len(partCode) >= 4 and partCode.isalnum()):
partCode = gen.word()
partNumber = str(partId) + str('-') + str(partCode)
return str(partNumber)
template = {'PartNumber': gen_partNumber, 'Description': 'sentence'}
gen.set_template(template)
documents = gen.documents(N)
print(str(documents))
from User.models import User
from essential_generators import DocumentGenerator
gen = DocumentGenerator()
import random
import string
newUsersGen = 8000
template = {'username': '******', "email": "email"}
N = 4
gen.set_template(template)
documents = gen.documents(newUsersGen)
for doc in documents:
randomContraint = ''.join(
random.choices(string.ascii_uppercase + string.digits, k=N))
try:
user = User.objects.create_user(username=doc["username"] +
randomContraint,
email=doc["email"],
password='******')
except Exception as e:
print(e)
