1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
# -*- coding: utf-8 -*-
from lettuce import step, world
from nose.tools import assert_true, assert_equals
from polls.models import Poll
from poll_result_details import (create_poll_result_file,
get_pollster_by_username)
@step(u'And "([^"]*)" is authored by "([^"]*)"')
def result_file_is_authored_by_pollster(step, file_name, pollster_username):
pollster = get_pollster_by_username(pollster_username)
data = {}
data['poll_id'] = world.poll_id.__str__()
data['pollster_username'] = pollster.username
data['pollster_id'] = pollster.id.__str__()
data['result'] = {}
world.poll_result_path = create_poll_result_file(data, file_name)
@step(u'And I upload "([^"]*)" on "([^"]*)"')
def upload_file_name_on_time_string(step, file_name, time_string):
b = world.browser
b.find_by_xpath('/html/body/div/div[2]/form/fieldset/div/button').click()
b.attach_file('result', world.poll_result_path)
b.find_by_xpath('/html/body/div/div[2]/form/div[1]/div/button').click()
@step(u'Then I should see a message that says "([^"]*)" and "([^"]*)"')
def i_should_see_a_message_that_says_text1_and_text2(step, text1, text2):
b = world.browser
assert_true(b.is_text_present(text1))
assert_true(b.is_text_present(text2))
@step(u'And "([^"]*)" has the time string "([^"]*)"')
def file_name_has_the_time_string_time_string(step, file_name, time_string):
poll = Poll.get(world.poll_id)
result_files = poll.get_result_files()
result_file = filter(
lambda f: file_name == f.get_file_name(), result_files)[0]
# Using (mock) Clock class from django settings
assert_equals(time_string, result_file.get_upload_timestamp())
|
