def test_save_data_saves_model_to_file_as_json(consumestore_fixture):
"""Test consumestore.save_data saves the model to the passed file.
Model should be saved as JSON.
"""
test_consumestore = consumestore_fixture
test_consumestore.save_data()
assert os.path.exists(TEST_FILE)
assert os.path.isfile(TEST_FILE)
line = fileops.get_line_from_file(TEST_FILE).next()
for key, value in EXPECTED_DATA_DICT.iteritems():
assert value in line
os.remove(TEST_FILE)
def test_consumer_writes_data_to_file(client_fixture):
"""Test workers.consumer writes correct data to file system."""
consumer(TEST_SITE, TEST_URI)
assert os.path.isfile(EXPECTED_MEDIA_FILE)
assert os.path.exists(EXPECTED_FILE)
assert os.path.isfile(EXPECTED_FILE)
line = fileops.get_line_from_file(EXPECTED_FILE).next()
for key, value in EXPECTED_DATA_DICT.iteritems():
assert value in line
os.remove(EXPECTED_FILE)
shutil.rmtree(os.path.join(
fileops.IMG_DIR,
TEST_CLASSIFICATION,
TEST_SITE
))
