def test_upload_to():
fake_instance = FakeModel()
tools.assert_equal(model_helpers.upload_to(fake_instance, "/tmp/filezx/myfile.png"), "FakeModel/%Y/myfile.png")
tools.assert_equal(
model_helpers.upload_to(fake_instance, "/tmp/filezx/1234567890123456.png"), "FakeModel/%Y/1234567890.png"
)
tools.assert_raises(ValueError, model_helpers.upload_to, fake_instance, "/tmp/filezx/1234567890123456.php")
tools.assert_raises(ValueError, model_helpers.upload_to, fake_instance, "/tmp/filezx/1234567890123456.pHp")
tools.assert_raises(ValueError, model_helpers.upload_to, fake_instance, "/tmp/filezx/.pHp")
def test_upload_to():
fake_instance = FakeModel()
upload_to = model_helpers.UploadTo(max_filename_length=10) # get upload_to function with short filename
year = date.today().year
tools.assert_equal(
upload_to(fake_instance, "/tmp/filezx/myfile.png"),
"FakeModel/%d/myfile-1.png" % year)
tools.assert_equal(
upload_to(fake_instance, "/tmp/filezx/1234567890123456.png"),
"FakeModel/%d/1234567890-1.png" % year)
tools.assert_raises(ValueError, upload_to, fake_instance, "/tmp/filezx/1234567890123456.php")
tools.assert_raises(ValueError, upload_to, fake_instance, "/tmp/filezx/1234567890123456.pHp")
tools.assert_raises(ValueError, upload_to, fake_instance, "/tmp/filezx/.pHp")
# Validate model_helper's upload_to function (Shortcut for using UploadTo class)
tools.assert_equal(
model_helpers.upload_to(fake_instance, "/tmp/filezx/myfile.png"),
"FakeModel/%d/myfile-1.png" % year)
