test_helpers.py

"""
:copyright: (c) 2014 Building Energy Inc
:license: see LICENSE for more details.
"""
import random
from django.contrib.webdesign import lorem_ipsum
import datetime
import base64
from decimal import getcontext, Decimal
getcontext().prec = 7
from localflavor.us.us_states import STATE_CHOICES

import csv
import datetime
import os
from tempfile import NamedTemporaryFile, mkdtemp

from django.contrib.auth.models import User
from django.core.files import File

from data_importer.models import ImportRecord, ImportFile

from django.test import TestCase


class BETestCase(TestCase):
    maxDiff = None

    def assertQuerySetEqualByPK(self, qs1, qs2):
        self.assertEqual([i.pk for i in qs1], [i.pk for i in qs2])

class DjangoFunctionalFactory:

    @classmethod
    def rand_int(cls, start=0, end=100):
        return random.randint(start, end)

    @classmethod
    def rand_float(cls, start=0, end=100):
        return random.uniform(start, end)

    @classmethod
    def rand_str(cls, length=None):
        # from http://stackoverflow.com/questions/785058/random-strings-in-python-2-6-is-this-ok
        if not length:
            length = cls.rand_int(end=10)
        nbits = length * 6 + 1
        bits = random.getrandbits(nbits)
        uc = u"%0x" % bits
        newlen = int(len(uc) / 2) * 2
        ba = bytearray.fromhex(uc[:newlen])
        return base64.urlsafe_b64encode(str(ba))[:length]

    @classmethod
    def rand_text(cls, num_paragraphs=None):
        if not num_paragraphs:
            num_paragraphs = cls.rand_int(start=1, end=5)
        return lorem_ipsum.paragraphs(num_paragraphs, common=False)

    @classmethod
    def rand_phone(cls):
        area = cls.rand_int(start=100, end=999)
        first = cls.rand_int(start=100, end=999)
        last = cls.rand_int(start=1000, end=9999)
        return "%s-%s-%s" % (area, first, last)

    @classmethod
    def rand_street_address(cls):
        s = "%s %s %s" % (cls.rand_int(end=10000), cls.rand_plant_name(), cls.rand_street_suffix())
        return s[:63]

    @classmethod
    def rand_city(cls):
        return "%s%s" % (cls.rand_plant_name(), cls.rand_city_suffix())

    @classmethod
    def rand_bool(cls):
        return cls.rand_int(0, 1) == 0

    @classmethod
    def rand_name(cls):
        return RANDOM_NAME_SOURCE[cls.rand_int(0, len(RANDOM_NAME_SOURCE) - 1)]

    @classmethod
    def rand_plant_name(cls):
        return RANDOM_PLANT_NAME_SOURCE[cls.rand_int(0, len(RANDOM_PLANT_NAME_SOURCE) - 1)]

    @classmethod
    def rand_street_suffix(cls):
        return RANDOM_STREET_SUFFIX_SOURCE[cls.rand_int(0, len(RANDOM_STREET_SUFFIX_SOURCE) - 1)]

    @classmethod
    def rand_city_suffix(cls):
        return RANDOM_CITY_SUFFIX_SOURCE[cls.rand_int(0, len(RANDOM_CITY_SUFFIX_SOURCE) - 1)]

    @classmethod
    def rand_date(cls, start_year=1900, end_year=2011):
        return datetime.date(year=cls.rand_int(start_year, end_year), month=cls.rand_int(1, 12), day=cls.rand_int(1, 28))

    @classmethod
    def rand_currency(cls, start=0, end=100):
        return Decimal(cls.rand_int(start=start * 100, end=end * 100)) / 100

    @classmethod
    def rand_email(cls):
        return "%s@%s" % (cls.rand_name().lower(), cls.rand_domain())

    @classmethod
    def rand_domain(cls):
        return RANDOM_EMAIL_DOMAINS[cls.rand_int(0, len(RANDOM_EMAIL_DOMAINS) - 1)]

    @classmethod
    def rand_us_state(cls):
        return STATE_CHOICES[cls.rand_int(0, len(STATE_CHOICES) - 1)][0]

    @classmethod
    def valid_test_cc_number(cls):
        return "4242424242424242"

    @classmethod
    def invalid_test_cc_number(cls):
        return "4242424242424241"

    @classmethod
    def test_cc_number(cls, valid=True):
        if valid:
            return cls.valid_test_cc_number()
        else:
            return cls.invalid_test_cc_number()

    @classmethod
    def random_conversation(cls, paragraphs=3):
        return generate_chomsky(paragraphs)


RANDOM_NAME_SOURCE = ["Atricia", "Linda", "Barbara", "Elizabeth", "Jennifer",
"Maria", "Susan", "Margaret", "Dorothy", "Lisa", "Nancy", "Karen", "Betty",
"Helen", "Sandra", "Donna", "Carol", "Ruth", "Sharon", "Michelle", "Laura",
"Sarah", "Kimberly", "Deborah", "Jessica", "Shirley", "Cynthia", "Angela",
"Melissa", "Brenda", "Amy", "Anna", "Rebecca", "Virginia", "Kathleen",
"Pamela", "Martha", "Debra", "Amanda", "Stephanie", "Carolyn", "Christine",
"Marie", "Janet", "Catherine", "Frances", "Ann", "Joyce", "Diane", "Alice",
"Julie", "Heather", "Teresa", "Doris", "Gloria", "Evelyn", "Jean", "Cheryl",
"Mildred", "Katherine", "Joan", "Ashley", "Judith", "Rose", "Janice", "Kelly",
"Nicole", "Judy", "Christina", "Kathy", "Theresa", "Beverly", "Denise",
"Tammy", "Irene", "Jane", "Lori", "Rachel", "Marilyn", "Andrea", "Kathryn",
"Louise", "Sara", "Anne", "Jacquelin", "Wanda", "Bonnie", "Julia", "Ruby",
"Lois", "Tina", "Phyllis", "Norma", "Paula", "Diana", "Annie", "Lillian",
"Emily", "Robin", "Peggy", "Crystal", "Gladys", "Rita", "Dawn", "Connie",
"Florence", "Tracy", "Edna", "Tiffany", "Carmen", "Rosa", "Cindy", "Grace",
"Wendy", "Victoria", "Edith", "Kim", "Sherry", "Sylvia", "Josephine", "Thelma",
"Shannon", "Sheila", "Ethel", "Ellen", "Elaine", "Marjorie", "Carrie",
"Charlotte", "Monica", "Esther", "Pauline", "Emma", "Juanita", "Anita",
"Rhonda", "Hazel", "Amber", "Eva", "Debbie", "April", "Leslie", "Clara",
"Lucille", "Jamie", "Joanne", "Eleanor", "Valerie", "Danielle", "Megan",
"Alicia", "Suzanne", "Michele", "Gail", "Bertha", "Darlene", "Veronica",
"Jill", "Erin", "Geraldine", "Lauren", "Cathy", "Joann", "Lorraine", "Lynn",
"Sally", "Regina", "Erica", "Beatrice", "Dolores", "Bernice", "Audrey",
"Yvonne", "Annette", "June", "Samantha", "Marion", "Dana", "Stacy", "Ana",
"Renee", "Ida", "Vivian", "Roberta", "Holly", "Brittany", "Melanie",
"Loretta", "Yolanda", "Jeanette", "Laurie", "Katie", "Kristen", "Vanessa",
"Alma", "Sue", "Elsie", "Beth", "Jeanne", "Vicki", "Carla", "Tara",
"Rosemary", "Eileen", "Terri", "Gertrude", "Lucy", "Tonya", "Ella",
"Stacey", "Wilma", "Gina", "Kristin", "Jessie", "Natalie", "Agnes", "Vera",
"Willie", "Charlene", "Bessie", "Delores", "Melinda", "Pearl", "Arlene",
"Maureen", "Colleen", "Allison", "Tamara", "Joy", "Georgia", "Constance",
"Lillie", "Claudia", "Jackie", "Marcia", "Tanya", "Nellie", "Minnie",
"Marlene", "Heidi", "Glenda", "Lydia", "Viola", "Courtney", "Marian",
"Stella", "Caroline", "Dora", "Jo", "Vickie", "Mattie", "Terry", "Maxine",
"Irma", "Mabel", "Marsha", "Myrtle", "Lena", "Christy", "Deanna", "Patsy",
"Hilda", "Gwendolyn", "Jennie", "Nora", "Margie", "Nina", "Cassandra",
"Leah", "Penny", "Kay", "Priscilla", "Naomi", "Carole", "Brandy", "Olga",
"Billie", "Dianne", "Tracey", "Leona", "Jenny", "Felicia", "Sonia", "Miriam",
"Velma", "Becky", "Bobbie", "Violet", "Kristina", "Toni", "Misty", "Mae",
"Shelly", "Daisy", "Ramona", "Sherri", "Erika", "Katrina", "Claire",
"Lindsey", "Lindsay", "Geneva", "Guadalupe", "Belinda", "Margarita", "Sheryl",
"Cora", "Faye", "Ada", "Natasha", "Sabrina", "Isabel", "Marguerit", "Hattie",
"Harriet", "Molly", "Cecilia", "Kristi", "Brandi", "Blanche", "Sandy", "Rosie",
"Joanna", "Iris", "Eunice", "Angie", "Inez", "Lynda", "Madeline", "Amelia",
"Alberta", "Genevieve", "Monique", "Jodi", "Janie", "Maggie", "Kayla", "Sonya",
"Jan", "Lee", "Kristine", "Candace", "Fannie", "Maryann", "Opal", "Alison",
"Yvette", "Melody", "Luz", "Susie", "Olivia", "Flora", "Shelley", "Kristy",
"Mamie", "Lula", "Lola", "Verna", "Beulah", "Antoinett", "Candice", "Juana",
"Jeannette", "Pam", "Kelli", "Hannah", "Whitney", "Bridget", "Karla", "Celia",
"Latoya", "Patty", "Shelia", "Gayle", "Della", "Vicky", "Lynne", "Sheri",
"Marianne", "Kara", "Jacquelyn", "Erma", "Blanca", "Myra", "Leticia", "Pat",
"Krista", "Roxanne", "Angelica", "Johnnie", "Robyn", "Francis", "Adrienne",
"Rosalie", "Alexandra", "Brooke", "Bethany", "Sadie", "Bernadett", "Traci",
"Jody", "Kendra", "Jasmine", "Nichole", "Rachael", "Chelsea", "Mable",
"Ernestine", "Muriel", "Marcella", "Elena", "Krystal", "Angelina", "Nadine",
"Kari", "Estelle", "Dianna", "Paulette", "Lora", "Mona", "Doreen", "Rosemarie",
"Angel", "Desiree", "Antonia", "Hope", "Ginger", "Janis", "Betsy", "Christie",
"Freda", "Mercedes", "Meredith", "Lynette", "Teri", "Cristina", "Eula",
"Leigh", "Meghan", "Sophia", "Eloise", "James", "John", "Robert",
"Michael", "William", "David", "Richard", "Charles", "Joseph", "Thomas",
"Christoph", "Daniel", "Paul", "Mark", "Donald", "George", "Kenneth", "Steven",
"Edward", "Brian", "Ronald", "Anthony", "Kevin", "Jason", "Matthew", "Gary",
"Timothy", "Jose", "Larry", "Jeffrey", "Frank", "Scott", "Eric", "Stephen",
"Andrew", "Raymond", "Gregory", "Joshua", "Jerry", "Dennis", "Walter",
"Patrick", "Peter", "Harold", "Douglas", "Henry", "Carl", "Arthur", "Ryan",
"Roger", "Joe", "Juan", "Jack", "Albert", "Jonathan", "Justin", "Terry",
"Gerald", "Keith", "Samuel", "Willie", "Ralph", "Lawrence", "Nicholas", "Roy",
"Benjamin", "Bruce", "Brandon", "Adam", "Harry", "Fred", "Wayne", "Billy",
"Steve", "Louis", "Jeremy", "Aaron", "Randy", "Howard", "Eugene", "Carlos",
"Russell", "Bobby", "Victor", "Martin", "Ernest", "Phillip", "Todd", "Jesse",
"Craig", "Alan", "Shawn", "Clarence", "Sean", "Philip", "Chris", "Johnny",
"Earl", "Jimmy", "Antonio", "Danny", "Bryan", "Tony", "Luis", "Mike",
"Stanley", "Leonard", "Nathan", "Dale", "Manuel", "Rodney", "Curtis", "Norman",
"Allen", "Marvin", "Vincent", "Glenn", "Jeffery", "Travis", "Jeff", "Chad",
"Jacob", "Lee", "Melvin", "Alfred", "Kyle", "Francis", "Bradley", "Jesus",
"Herbert", "Frederick", "Ray", "Joel", "Edwin", "Don", "Eddie", "Ricky",
"Troy", "Randall", "Barry", "Alexander", "Bernard", "Mario", "Leroy",
"Francisco", "Marcus", "Micheal", "Theodore", "Clifford", "Miguel", "Oscar",
"Jay", "Jim", "Tom", "Calvin", "Alex", "Jon", "Ronnie", "Bill", "Lloyd",
"Tommy", "Leon", "Derek", "Warren", "Darrell", "Jerome", "Floyd", "Leo",
"Alvin", "Tim", "Wesley", "Gordon", "Dean", "Greg", "Jorge", "Dustin", "Pedro",
"Derrick", "Dan", "Lewis", "Zachary", "Corey", "Herman", "Maurice", "Vernon",
"Roberto", "Clyde", "Glen", "Hector", "Shane", "Ricardo", "Sam", "Rick",
"Lester", "Brent", "Ramon", "Charlie", "Tyler", "Gilbert", "Gene", "Marc",
"Reginald", "Ruben", "Brett", "Angel", "Nathaniel", "Rafael", "Leslie",
"Edgar", "Milton", "Raul", "Ben", "Chester", "Cecil", "Duane", "Franklin",
"Andre", "Elmer", "Brad", "Gabriel", "Ron", "Mitchell", "Roland", "Arnold",
"Harvey", "Jared", "Adrian", "Karl", "Cory", "Claude", "Erik", "Darryl",
"Jamie", "Neil", "Jessie", "Christian", "Javier", "Fernando", "Clinton", "Ted",
"Mathew", "Tyrone", "Darren", "Lonnie", "Lance", "Cody", "Julio", "Kelly",
"Kurt", "Allan", "Nelson", "Guy", "Clayton", "Hugh", "Max", "Dwayne", "Dwight",
"Armando", "Felix", "Jimmie", "Everett", "Jordan", "Ian", "Wallace", "Ken",
"Bob", "Jaime", "Casey", "Alfredo", "Alberto", "Dave", "Ivan", "Johnnie",
"Sidney", "Byron", "Julian", "Isaac", "Morris", "Clifton", "Willard", "Daryl",
"Ross", "Virgil", "Andy", "Marshall", "Salvador", "Perry", "Kirk", "Sergio",
"Marion", "Tracy", "Seth", "Kent", "Terrance", "Rene", "Eduardo", "Terrence",
"Enrique", "Freddie", "Wade", "Austin", "Stuart", "Fredrick", "Arturo",
"Alejandro", "Jackie", "Joey", "Nick", "Luther", "Wendell", "Jeremiah", "Evan",
"Julius", "Dana", "Donnie", "Otis", "Shannon", "Trevor", "Oliver", "Luke",
"Homer", "Gerard", "Doug", "Kenny", "Hubert", "Angelo", "Shaun", "Lyle",
"Matt", "Lynn", "Alfonso", "Orlando", "Rex", "Carlton", "Ernesto", "Cameron",
"Neal", "Pablo", "Lorenzo", "Omar", "Wilbur", "Blake", "Grant", "Horace",
"Roderick", "Kerry", "Abraham", "Willis", "Rickey", "Jean", "Ira", "Andres",
"Cesar", "Johnathan", "Malcolm", "Rudolph", "Damon", "Kelvin", "Rudy",
"Preston", "Alton", "Archie", "Marco", "Wm", "Pete", "Randolph", "Garry",
"Geoffrey", "Jonathon", "Felipe", "Bennie", "Gerardo", "Ed", "Dominic",
"Robin", "Loren", "Delbert", "Colin", "Guillermo", "Earnest", "Lucas", "Benny",
"Noel", "Spencer", "Rodolfo", "Myron", "Edmund", "Garrett", "Salvatore",
"Cedric", "Lowell", "Gregg", "Sherman", "Wilson", "Devin", "Sylvester", "Kim",
"Roosevelt", "Israel", "Jermaine", "Forrest", "Wilbert", "Leland", "Simon",
"Guadalupe", "Clark", "Irving", "Carroll", "Bryant", "Owen", "Rufus",
"Woodrow", "Sammy", "Kristophe", "Mack", "Levi", "Marcos", "Gustavo", "Jake",
"Lionel", "Marty", "Taylor", "Ellis", "Dallas", "Gilberto", "Clint", "Nicolas",
"Laurence", "Ismael", "Orville", "Drew", "Jody", "Ervin", "Dewey", "Al",
"Wilfred", "Josh", "Hugo", "Ignacio", "Caleb", "Tomas", "Sheldon", "Erick",
"Frankie", "Stewart", "Doyle", "Darrel", "Rogelio", "Terence", "Santiago",
"Alonzo", "Elias", "Bert", "Elbert", "Ramiro", "Conrad", "Pat", "Noah",
"Grady", "Phil", "Cornelius", "Lamar", "Rolando", "Clay", "Percy", "Dexter",
"Bradford", "Merle", "Darin", "Amos", "Terrell", "Moses", "Irvin", "Saul",
"Roman", "Darnell", "Randal", "Tommie", "Timmy", "Darrin", "Winston",
"Brendan", "Toby", "Van", "Abel", "Dominick", "Boyd", "Courtney", "Jan",
"Emilio", "Elijah", "Cary", "Domingo", "Santos", "Aubrey", "Emmett", "Marlon",
"Emanuel", "Jerald", "Edmond", "Emil", "Dewayne", "Will", "Otto", "Teddy",
"Reynaldo", "Bret", "Morgan", "Jess", "Trent", "Humberto", "Emmanuel",
"Stephan", "Louie", "Vicente", "Lamont", "Stacy", "Garland", "Miles", "Micah",
"Efrain", "Billie", "Logan", "Heath", "Rodger", "Harley", "Demetrius",
"Ethan", "Eldon", "Rocky", "Pierre", "Junior", "Freddy", "Eli", "Bryce",
"Antoine", "Robbie", "Kendall", "Royce", "Sterling", "Mickey", "Chase",
"Grover", "Elton", "Cleveland", "Dylan", "Chuck", "Damian", "Reuben", "Stan",
"August", "Leonardo", "Jasper", "Russel", "Erwin", "Benito", "Hans", "Monte",
"Blaine", "Ernie", "Curt", "Quentin", "Agustin", "Murray", "Jamal", "Devon",
"Adolfo", "Harrison", "Tyson", "Burton", "Brady", "Elliott", "Wilfredo",
"Bart", "Jarrod", "Vance", "Denis", "Damien", "Joaquin", "Harlan", "Desmond",
"Elliot", "Darwin", "Ashley", "Gregorio", "Buddy", "Xavier", "Kermit",
"Roscoe", "Esteban", "Anton", "Solomon", "Scotty", "Norbert", "Elvin",
"Williams", "Nolan", "Carey", "Rod", "Quinton", "Hal", "Brain", "Rob",
"Elwood", "Kendrick", "Darius", "Moises", "Son", "Marlin", "Fidel",
"Thaddeus", "Cliff", "Marcel", "Ali", "Jackson", "Raphael", "Bryon", "Armand",
"Alvaro", "Jeffry", "Dane", "Joesph", "Thurman", "Ned", "Sammie", "Rusty",
"Michel", "Monty", "Rory", "Fabian", "Reggie", "Mason", "Graham", "Kris",
"Isaiah", "Vaughn", "Gus", "Avery", "Loyd", "Diego", "Alexis", "Adolph",
"Norris", "Millard", "Rocco", "Gonzalo", "Derick", "Rodrigo", "Gerry",
"Stacey", "Carmen", "Wiley", "Rigoberto", "Alphonso", "Ty", "Shelby",
"Rickie", "Noe", "Vern", "Bobbie", "Reed", "Jefferson", "Elvis", "Bernardo",
"Mauricio", "Hiram", "Donovan", "Basil", "Riley", "Ollie", "Nickolas",
"Maynard", "Scot", "Vince", "Quincy", "Eddy", "Sebastian", "Federico",
"Ulysses", "Heriberto", "Donnell", "Cole", "Denny", "Davis", "Gavin", "Emery",
"Ward", "Romeo", "Jayson", "Dion", "Dante", "Clement", "Coy", "Odell",
"Maxwell", "Jarvis", "Bruno", "Issac", "Mary", "Dudley", "Brock", "Sanford",
"Colby", "Carmelo", "Barney", "Nestor", "Hollis", "Stefan", "Donny", "Art",
"Linwood", "Beau", "Weldon", "Galen", "Isidro", "Truman", "Delmar",
"Johnathon", "Silas", "Frederic", "Dick", "Kirby", "Irwin", "Cruz", "Merlin",
"Merrill", "Charley", "Marcelino", "Lane", "Harris", "Cleo", "Carlo",
"Trenton", "Kurtis", "Hunter", "Aurelio", "Winfred", "Vito", "Collin",
"Denver", "Carter", "Leonel", "Emory", "Pasquale", "Mohammad", "Mariano",
"Danial", "Blair", "Landon", "Dirk", "Branden", "Adan", "Numbers", "Clair",
"Buford", "German", "Bernie", "Wilmer", "Joan", "Emerson", "Zachery",
"Fletcher", "Jacques", "Errol", "Dalton", "Monroe", "Josue", "Dominique",
"Edwardo", "Booker", "Wilford", "Sonny", "Shelton", "Carson", "Theron",
"Raymundo", "Daren", "Tristan", "Houston", "Robby", "Lincoln", "Jame",
"Genaro", "Gale", "Bennett", "Octavio", "Cornell", "Laverne", "Hung", "Arron",
"Antony", "Herschel", "Alva", "Giovanni", "Garth", "Cyrus", "Cyril", "Ronny",
"Stevie", "Lon", "Freeman", "Erin", "Duncan", "Kennith", "Carmine",
"Augustine", "Young", "Erich", "Chadwick", "Wilburn", "Russ", "Reid", "Myles",
"Anderson", "Morton", "Jonas", "Forest", "Mitchel", "Mervin", "Zane", "Rich",
"Jamel", "Lazaro", "Alphonse", "Randell", "Major", "Johnie", "Jarrett",
"Brooks", "Ariel", "Abdul", "Dusty", "Luciano", "Lindsey", "Tracey", "Seymour",
"Scottie", "Eugenio", "Mohammed", "Sandy", "Valentin", "Chance", "Arnulfo",
"Lucien", "Ferdinand", "Thad", "Ezra", "Sydney", "Aldo", "Rubin", "Royal",
"Mitch", "Earle", "Abe", "Wyatt", "Marquis", "Lanny", "Kareem", "Jamar",
"Boris", "Isiah", "Emile", "Elmo", "Aron", "Leopoldo", "Everette", "Josef",
"Gail", "Eloy", "Dorian", "Rodrick", "Reinaldo", "Lucio", "Jerrod", "Weston",
"Hershel", "Barton", "Parker", "Lemuel", "Lavern", "Burt", "Jules", "Gil",
"Eliseo", "Ahmad", "Nigel", "Efren", "Antwan", "Alden", "Margarito", "Coleman",
"Refugio", "Dino", "Osvaldo", "Les", "Deandre", "Normand", "Kieth", "Ivory",
"Andrea", "Trey", "Norberto", "Napoleon", "Jerold", "Fritz", "Rosendo",
"Milford", "Sang", "Deon", "Christope", "Alfonzo", "Lyman", "Josiah", "Brant",
"Wilton", "Rico", "Jamaal", "Dewitt", "Carol", "Brenton", "Yong", "Olin",
"Foster", "Faustino", "Claudio", "Judson", "Gino", "Edgardo", "Berry", "Alec",
"Tanner", "Jarred", "Donn", "Trinidad", "Tad", "Shirley", "Prince", "Porfirio",
"Odis", "Maria", "Lenard", "Chauncey", "Chang", "Tod", "Mel", "Marcelo", "Kory",
"Augustus", "Keven", "Hilario", "Bud", "Sal", "Rosario", "Orval", "Mauro",
"Dannie", "Zachariah", "Olen", "Anibal", "Milo", "Jed", "Frances", "Thanh",
"Dillon", "Amado", "Newton", "Connie", "Lenny", "Tory", "Richie", "Lupe",
"Horacio", "Brice", "Mohamed", "Delmer", "Dario", "Reyes", "Dee", "Mac",
"Jonah", "Jerrold", "Robt", "Hank", "Sung", "Rupert", "Rolland", "Kenton",
"Damion", "Chi", "Antone", "Waldo", "Fredric", "Bradly", "Quinn", "Kip", "Burl",
"Walker", "Tyree", "Jefferey", "Ahmed", "Willy", "Stanford", "Oren", "Noble",
"Moshe", "Mikel", "Enoch", "Brendon", "Quintin", "Jamison", "Florencio",
"Darrick", "Tobias", "Minh", "Hassan", "Giuseppe", "Demarcus", "Cletus",
"Tyrell", "Lyndon", "Keenan", "Werner", "Theo", "Geraldo", "Lou", "Columbus",
"Chet", "Bertram", "Markus", "Huey", "Hilton", "Dwain", "Donte", "Tyron",
"Omer", "Isaias", "Hipolito", "Fermin", "Chung", "Adalberto", "Valentine",
"Jamey", "Bo", "Barrett", "Whitney", "Teodoro", "Mckinley", "Maximo",
"Garfield", "Sol", "Raleigh", "Lawerence", "Abram", "Rashad", "King", "Emmitt",
"Daron", "Chong", "Samual", "Paris", "Otha", "Miquel", "Lacy", "Eusebio",
"Dong", "Domenic", "Darron", "Buster", "Antonia", "Wilber", "Renato", "Jc",
"Hoyt", "Haywood", "Ezekiel", "Chas", "Florentin", "Elroy", "Clemente",
"Arden", "Neville", "Kelley", "Edison", "Deshawn", "Carrol", "Shayne",
"Nathanial", "Jordon", "Danilo", "Claud", "Val", "Sherwood", "Raymon",
"Rayford", "Cristobal", "Ambrose", "Titus", "Hyman", "Felton", "Ezequiel",
"Erasmo", "Stanton", "Lonny", "Len", "Ike", "Milan", "Lino", "Jarod", "Herb",
"Andreas", "Walton", "Rhett", "Palmer", "Jude", "Douglass", "Cordell",
"Oswaldo", "Ellsworth", "Virgilio", "Toney", "Nathanael", "Del", "Britt",
"Benedict", "Mose", "Hong", "Leigh", "Johnson", "Isreal", "Gayle", "Garret",
"Fausto", "Asa", "Arlen", "Zack", "Warner", "Modesto", "Francesco", "Manual",
"Jae", "Gaylord", "Gaston", "Filiberto", "Deangelo", "Michale", "Granville",
"Wes", "Malik", "Zackary", "Tuan", "Nicky", "Eldridge", "Cristophe", "Cortez",
"Antione", "Malcom", "Long", "Korey", "Jospeh", "Colton", "Waylon", "Von",
"Hosea", "Shad", "Santo", "Rudolf", "Rolf", "Rey", "Renaldo", "Marcellus",
"Lucius", "Lesley", "Kristofer", "Boyce", "Benton", "Man", "Kasey", "Jewell",
"Hayden", "Harland", "Arnoldo", "Rueben", "Leandro", "Kraig", "Jerrell",
"Jeromy", "Hobert", "Cedrick", "Arlie", "Winford", "Wally", "Patricia",
"Luigi", "Keneth", "Jacinto", "Graig", "Franklyn", "Edmundo", "Sid", "Porter",
"Leif", "Lauren", "Jeramy", "Elisha", "Buck", "Willian", "Vincenzo", "Shon",
"Michal", "Lynwood", "Lindsay", "Jewel", "Jere", "Hai", "Elden", "Dorsey",
"Darell", "Broderick", "Alonso"]

RANDOM_PLANT_NAME_SOURCE = ["Abelia", "Acacia", "Acer", "Acevedo", "Afra", "Akina",
"Alaleh", "Alani", "Alder", "Almond", "Althea ", "Alyssum", "Amaranta", "Amaryllis",
"Anita", "Apricot", "Arousa", "Arusa", "Ash", "Aspen ", "Aster", "Astera", "Avishan",
"Ayame", "Ayla", "Azalea", "Azargol", "Azargoon", "Azarin", "Azhand", "Babuk",
"Bahar", "Baharak", "Banafsheh", "Barnacle", "Basil", "Bay", "Beech", "Begonia",
"Belladonna", "Birch", "Blackberry", "Blossom", "Bluebell ", "Booker", "Botan",
"Bramble", "Bryony", "Bud", "Burke ", "Buttercup", "Cactus", "Caltha", "Camelai",
"Camellia", "Carnation", "Cedar", "Cherise", "Cherry", "Cinnamon", "Cliantha",
"Clover", "Cosmos", "Cyclamen", "Cypress", "Daffodil", "Dahlia", "Daisy", "Dandelion",
"Daphne", "Dianthe", "Dianthus", "Enola ", "Eranthe", "Fern", "Fiorenza", "Fleur",
"Fern", "Fiorenza", "Fleur", "Flora", "Freesia", "Fuchsia", "Gardenia", "Garland",
"Gazania", "Geranium", "Ginger", "Gooseberry", "Gul", "Hawthorne", "Hazel", "Holly",
"Hollyhock", "Honeysuckle", "Hyacinth", "Iris ", "Ivy", "Jacaranda", "Jasmine",
"Jessamine", "Juniper", "Kalei", "Lantana", "Laurel", "Leilani", "Licorice ",
"Lilac", "Lily ", "Lobelia", "Lotus", "Magnolia", "Mallow ", "Mandrake", "Maple",
"Marguerite", "Marigold", "Mayflower", "Miki", "Mimosa", "Mulberry", "Myrtle ",
"Nihal", "Olive", "Pansy ", "Patience", "Peach", "Peony", "Peppermint", "Periwinkle",
"Persimmon", "Petunia", "Pimpernel", "Poppy", "Posey", "Primrose", "Pumpkin",
"Quince", "Rose", "Rosemary", "Saffron", "Sage", "Shamrock", "Snapdragon",
"Snowdrop", "Sorrel", "Sunflower", "Sweet Pea", "Tansy ", "Thistle", "Tiger-lily",
"Truffle", "Tulip", "Verbena ", "Violet", "Willow", "Yasaman", "Yasmin", "Yasminah",
"Yew", "Zara"]

RANDOM_STREET_SUFFIX_SOURCE = ["St.", "Ave.", "Blvd.", "Ln.", "Ct.", "Pl.", "Way"]

RANDOM_EMAIL_DOMAINS = ["example.com", "example.net", "example.org"]
                        # "gmail.com", "yahoo.com", "hotmail.com", "live.com",
                        # "comcast.net", "qwest.com",

RANDOM_CITY_SUFFIX_SOURCE = ["ville", "berg", "ton", "y", "", "land"]

class Factory(DjangoFunctionalFactory):

    @classmethod
    def user(cls, with_site_and_portfolios=False):
        email = cls.rand_email()
        u = User.objects.create(username=email, email=email)
        p = cls.rand_str(length=20)
        u.is_active = True
        u.set_password(p)
        u.save()

        if with_site_and_portfolios:
            num_portfolios = cls.rand_int(end=3)

            for p_count in range(0, num_portfolios):
                num_sites = cls.rand_int(end=5)
                p = cls.portfolio()
                cls.principalportfoliorelation(user=u, portfolio=p)
                for s_count in range(0, num_sites):
                    cls.site(portfolio=p)
        return u, p

    @classmethod
    def userprofile(cls):
        u, p = cls.user()
        up = u.get_profile()
        up.preferred_phone_number = cls.rand_phone()
        up.save()
        return up, p

    @classmethod
    def site(cls, portfolio=None, floor_area="undefined", state="undefined", **kwargs):
        if not portfolio:
            portfolio = cls.portfolio()

        # Needed instead of None so that we can explicitly pass null values into the constructor.
        if floor_area == "undefined":
            floor_area = cls.rand_int(start=5, end=10000)

        if state == "undefined":
            state = cls.rand_us_state()

        site_kwargs = {
            "name": cls.rand_str(length=50),
            "address1": cls.rand_street_address(),
            "address2": "Suite %s" % cls.rand_int(),
            "city": cls.rand_city(),
            "state": state,
            "zipcode": cls.rand_int(start=10000, end=99999),
            # "latitude": cls.rand_float(-90, 90),
            # "longitude": cls.rand_float(0, 180),
            "phone_number": cls.rand_phone(),
            "meta_data": {"area": floor_area}
        }
        site_kwargs.update(kwargs)

        s = Site.objects.create(**site_kwargs)

        s.portfolios.add(portfolio)
        s.save()
        return s

    @classmethod
    @classmethod
    def principalsiterelation(cls, user=None, portfolio=None, site=None):
        if not user:
            user = cls.user()

        if not portfolio:
            portfolio = cls.portfolio()
        if not site:
            site = cls.site()

        return PrincipalSiteRelation.objects.create(
                user=user,
                portfolio=portfolio,
                site=site,
            )

    @classmethod
    def csv_to_import(cls, name=None, num_rows=None, column_headers=None, data_rows=None):
        if not num_rows:
            if data_rows:
                num_rows = len(data_rows)
            else:
                num_rows = cls.rand_int(start=2, end=10)
        if not column_headers:
            column_headers = [cls.rand_str() for i in range(0, cls.rand_int(end=20))]

        if not name:
            f = NamedTemporaryFile(delete=False)
        else:
            tempdir = mkdtemp()
            f = open(os.path.join(tempdir, name), 'w')

        with open(f.name, 'wb') as csvfile:
            csv_writer = csv.writer(csvfile)
            csv_writer.writerow(column_headers)
            if data_rows:
                for r in data_rows:
                    csv_writer.writerow(r)
            else:
                for i in range(0, num_rows):
                    row = ["%s-%s" % (i, cls.rand_int()) for h in range(0, len(column_headers))]
                    csv_writer.writerow(row)

        return f

    @classmethod
    def import_file(cls, csv_to_import=None, import_record=None, num_rows=None, column_headers=None, **kwargs):
        if not import_record:
            import_record = cls.import_record(num_files=0)

        if not csv_to_import:
            csv_to_import = cls.csv_to_import(num_rows=num_rows, column_headers=column_headers)

        import_file = ImportFile.objects.create(
                import_record=import_record,
            )
        import_file.file = File(open(csv_to_import.name))
        import_file.save()

        return import_file

    @classmethod
    def import_record(cls, userprofile=None, num_files=None, **kwargs):
        if not userprofile:
            userprofile, password = cls.userprofile()
        if not num_files:
            num_files = cls.rand_int(start=2, end=10)
        record = ImportRecord.objects.create(
                name=cls.rand_str(),
                created_at=datetime.datetime.now(),
                last_modified_by=userprofile,
            )

        for i in range(0, num_files):
            cls.import_file(import_record=record)
        return record