def test_no_redirect_html_response(self, mock_make_grammar_info):
dset = Dataset(name='blank', user='******')
dset.global_stats_calculated = True
dset.classical = False
dset.grammar_navbar['lengths'] = [1, 3, 6]
dset.global_stats = {
'grams': "[frozenset([(1, 2)]), frozenset([2, 3])]",
'num_poots': 5,
'num_total_poots': 19,
'percent_poots': 5.0/19,
'num_cots': 2,
'num_total_cots': 6,
'percent_cots': 2.0/6
}
dset.save()
response = self.client.get(url_for(
'grammars.global_stats_calculated', dset_name='blank',
sort_value=6, classical=False, page=0, sort_by='size'
))
self.assert_200(response)
print response.data
data = json.loads(response.data)
assert not data['need_redirect']
assert data['finished']
assert not data['retry']
assert data['html_str']
assert data['grammar_stat_url']
mock_make_grammar_info.assert_called_with('blank')
def test_no_grammars_exist(self, mock_make_grammar_info):
dset = Dataset(name='blank', user='******')
dset.global_stats_calculated = True
dset.classical = False
dset.grammar_navbar['lengths'] = []
dset.global_stats = {
'grams': "[]",
'num_poots': 0,
'num_total_poots': 19,
'percent_poots': 0.0,
'num_cots': 0,
'num_total_cots': 6,
'percent_cots': 0.0
}
dset.save()
response = self.client.get(url_for(
'grammars.global_stats_calculated', dset_name='blank',
sort_value=0, classical=False, page=0, sort_by='size'
))
print(response.data)
self.assert_200(response)
data = json.loads(response.data)
assert not data['need_redirect']
assert data['finished']
assert not data['retry']
assert not data['grammars_exist']
assert not mock_make_grammar_info.called
def test_retry_and_not_finished(self):
dset = Dataset(name='blank', user='******')
dset.global_stats_calculated = False
dset.save()
response = self.client.get(url_for('grammars.global_stats_calculated',
dset_name='blank', sort_value=0,
sort_by='size', page=0,
classical=False))
data = json.loads(response.data)
self.assert_200(response)
assert data['retry']
assert not data['finished']
def test_redirect_because_no_grams(self):
dset = Dataset(name='blank', user='******')
dset.global_stats_calculated = True
dset.classical = False
dset.grammar_navbar['lengths'] = [1, 3, 6]
dset.global_stats['grams'] = "[]"
dset.save()
response = self.client.get(url_for('grammars.global_stats_calculated',
dset_name='blank', sort_value=0,
classical=False, page=0,
sort_by='size'))
self.assert_200(response)
data = json.loads(response.data)
assert not data['finished']
assert not data['retry']
assert data['need_redirect']
assert "/6?" in data['redirect_url']
