def import_organizations(self):
self.sign_in()
self.crawler.go_to_all_companies()
companies = get_companies(self.crawler.content)
ImportedEntity.import_companies_as_organizations(companies)
self.already_done.append(ORGANIZATIONS)
self.logfile.write('Organizations imported')
def test_get_last_company(self):
companies_page = self.get_sample_content('companies.html')
companies = parser.get_companies(companies_page)
self.assertEquals(len(companies), 42)
company = companies[-1]
self.assertEquals(company['name'], "Zilics")
self.assertEquals(company['original_id'], 26)
self.assertEquals(company['description'], "")
def test_get_first_companies(self):
companies_page = self.get_sample_content('companies.html')
companies = parser.get_companies(companies_page)
self.assertEquals(len(companies), 42)
company = companies[0]
self.assertEquals(company['name'], "Ambar Tec")
self.assertEquals(company['original_id'], 4)
self.assertEquals(company['description'], "")
company = companies[1]
self.assertEquals(company['name'], "Anprotec")
self.assertEquals(company['original_id'], 16)
self.assertEquals(
company['description'], """Katia Sitta Fortini
Coordenadora de Atendimento e Relacionamento ANPROTEC
telefone 8427.1420""")
def test_select_companies(self):
crawler = NetunoCrawler(ROOT_URL)
crawler.login(username='******', password='******')
self.assertTrue(crawler.logged_in)
crawler.go_to_all_companies()
companies = get_companies(crawler.content)
self.assertEquals(len(companies), 43)
company = companies[0]
self.assertEquals(company['name'], "Ambar Tec")
self.assertEquals(company['original_id'], 4)
self.assertEquals(company['description'], "")
company = companies[1]
self.assertEquals(company['name'], "Anprotec")
self.assertEquals(company['original_id'], 16)
company = companies[-1]
self.assertEquals(company['name'], "Zilics")
self.assertEquals(company['original_id'], 26)
def test_select_user_from_companies(self):
crawler = NetunoCrawler(ROOT_URL)
crawler.login(username='******', password='******')
self.assertTrue(crawler.logged_in)
crawler.go_to_all_companies()
companies = get_companies(crawler.content)
self.assertEquals(len(companies), 43)
company = companies[0]
crawler.go_to_users_from_company(company['original_id'])
users = get_users(crawler.content)
self.assertEquals(len(users), 1)
user = users[0]
self.assertEquals(user['original_id'], 20)
self.assertEquals(user['username'], 'lamatuzzi')
self.assertEquals(user['first_name'], 'Luciano')
self.assertEquals(user['last_name'], 'Teixeira')
self.assertEquals(user['middle_name'], 'Amatuzzi')
# SEA
company = companies[26]
crawler.go_to_users_from_company(company['original_id'])
users = get_users(crawler.content)
self.assertEquals(len(users), 60)
user = users[0]
self.assertEquals(user['original_id'], 13)
self.assertEquals(user['username'], 'mabreu')
self.assertEquals(user['first_name'], 'Marianne')
self.assertEquals(user['last_name'], 'Abreu')
self.assertEquals(user['middle_name'], '')
user = users[59]
self.assertEquals(user['original_id'], 6)
self.assertEquals(user['username'], 'msousa')
self.assertEquals(user['first_name'], 'Marcelo')
self.assertEquals(user['last_name'], 'Zouza')
self.assertEquals(user['middle_name'], '')
def test_get_something(self):
companies_page = self.get_sample_content('companies.html')
companies = parser.get_companies(companies_page)
self.assertGreater(len(companies), 0)