mostly finished port to ruby3

This commit is contained in:
Kyle Drake 2022-08-10 14:31:36 -05:00
parent a9dd102383
commit 0ca1473d22
30 changed files with 765 additions and 731 deletions

View file

@ -34,11 +34,11 @@ describe 'site_files' do
upload 'files[]' => uploaded_file
testfile = @site.site_files_dataset.where(path: 'test.jpg').first
testfile.wont_equal nil
_(testfile).wont_equal nil
testfile.rename 'derp.jpg'
@site.site_files_dataset.where(path: 'derp.jpg').first.wont_equal nil
PurgeCacheWorker.jobs.first['args'].last.must_equal '/test.jpg'
File.exist?(@site.files_path('derp.jpg')).must_equal true
_(@site.site_files_dataset.where(path: 'derp.jpg').first).wont_equal nil
_(PurgeCacheWorker.jobs.first['args'].last).must_equal '/test.jpg'
_(File.exist?(@site.files_path('derp.jpg'))).must_equal true
end
it 'fails for bad extension change' do
@ -47,8 +47,8 @@ describe 'site_files' do
testfile = @site.site_files_dataset.where(path: 'test.jpg').first
res = testfile.rename('dasharezone.exe')
res.must_equal [false, 'unsupported file type']
@site.site_files_dataset.where(path: 'test.jpg').first.wont_equal nil
_(res).must_equal [false, 'unsupported file type']
_(@site.site_files_dataset.where(path: 'test.jpg').first).wont_equal nil
end
it 'renames nonstandard file type for supporters' do
@ -58,27 +58,27 @@ describe 'site_files' do
upload 'files[]' => Rack::Test::UploadedFile.new('./tests/files/flowercrime.wav', 'audio/x-wav')
testfile = @site.site_files_dataset.where(path: 'flowercrime.wav').first
res = testfile.rename('flowercrime.exe')
res.first.must_equal true
File.exists?(@site.files_path('flowercrime.exe')).must_equal true
@site.site_files_dataset.where(path: 'flowercrime.exe').first.wont_equal nil
_(res.first).must_equal true
_(File.exists?(@site.files_path('flowercrime.exe'))).must_equal true
_(@site.site_files_dataset.where(path: 'flowercrime.exe').first).wont_equal nil
end
end
it 'works for directory' do
@site.create_directory 'dirone'
@site.site_files.select {|sf| sf.path == 'dirone'}.length.must_equal 1
_(@site.site_files.select {|sf| sf.path == 'dirone'}.length).must_equal 1
dirone = @site.site_files_dataset.where(path: 'dirone').first
dirone.wont_equal nil
dirone.is_directory.must_equal true
_(dirone).wont_equal nil
_(dirone.is_directory).must_equal true
res = dirone.rename('dasharezone')
res.must_equal [true, nil]
_(res).must_equal [true, nil]
dasharezone = @site.site_files_dataset.where(path: 'dasharezone').first
dasharezone.wont_equal nil
dasharezone.is_directory.must_equal true
_(dasharezone).wont_equal nil
_(dasharezone.is_directory).must_equal true
PurgeCacheWorker.jobs.first['args'].last.must_equal 'dirone'
PurgeCacheWorker.jobs.last['args'].last.must_equal 'dasharezone'
_(PurgeCacheWorker.jobs.first['args'].last).must_equal 'dirone'
_(PurgeCacheWorker.jobs.last['args'].last).must_equal 'dasharezone'
end
it 'changes path of files and dirs within directory when changed' do
@ -92,11 +92,11 @@ describe 'site_files' do
)
@site.site_files.select {|s| s.path == 'test'}.first.rename('test2')
@site.site_files.select {|sf| sf.path =~ /test2\/index.html/}.length.must_equal 1
@site.site_files.select {|sf| sf.path =~ /test2\/test.jpg/}.length.must_equal 1
@site.site_files.select {|sf| sf.path =~ /test\/test.jpg/}.length.must_equal 0
_(@site.site_files.select {|sf| sf.path =~ /test2\/index.html/}.length).must_equal 1
_(@site.site_files.select {|sf| sf.path =~ /test2\/test.jpg/}.length).must_equal 1
_(@site.site_files.select {|sf| sf.path =~ /test\/test.jpg/}.length).must_equal 0
PurgeCacheWorker.jobs.collect {|p| p['args'].last}.sort.must_equal ["/test/test.jpg", "/test/index.html", "/test/", "test", "test2", "test/test.jpg", "test2/test.jpg", "test/index.html", "test/", "test2/index.html", "test2/"].sort
_(PurgeCacheWorker.jobs.collect {|p| p['args'].last}.sort).must_equal ["/test/test.jpg", "/test/index.html", "/test/", "test", "test2", "test/test.jpg", "test2/test.jpg", "test/index.html", "test/", "test2/index.html", "test2/"].sort
end
it 'doesnt wipe out existing file' do
@ -110,35 +110,35 @@ describe 'site_files' do
)
res = @site.site_files_dataset.where(path: 'test/index.html').first.rename('test/test.jpg')
res.must_equal [false, 'file already exists']
_(res).must_equal [false, 'file already exists']
end
it 'doesnt wipe out existing dir' do
@site.create_directory 'dirone'
@site.create_directory 'dirtwo'
res = @site.site_files.select{|sf| sf.path == 'dirtwo'}.first.rename 'dirone'
res.must_equal [false, 'directory already exists']
_(res).must_equal [false, 'directory already exists']
end
it 'refuses to move index.html' do
res = @site.site_files.select {|sf| sf.path == 'index.html'}.first.rename('notindex.html')
res.must_equal [false, 'cannot rename or move root index.html']
_(res).must_equal [false, 'cannot rename or move root index.html']
end
it 'works with unicode characters' do
uploaded_file = Rack::Test::UploadedFile.new('./tests/files/test.jpg', 'image/jpeg')
upload 'files[]' => uploaded_file
@site.site_files_dataset.where(path: 'test.jpg').first.rename("HELL💩؋.jpg")
@site.site_files_dataset.where(path: "HELL💩؋.jpg").first.wont_equal nil
_(@site.site_files_dataset.where(path: "HELL💩؋.jpg").first).wont_equal nil
end
it 'scrubs weird carriage return shit characters' do
uploaded_file = Rack::Test::UploadedFile.new('./tests/files/test.jpg', 'image/jpeg')
upload 'files[]' => uploaded_file
proc {
_(proc {
@site.site_files_dataset.where(path: 'test.jpg').first.rename("\r\n\t.jpg")
}.must_raise ArgumentError
@site.site_files_dataset.where(path: 'test.jpg').first.wont_equal nil
}).must_raise ArgumentError
_(_(@site.site_files_dataset.where(path: 'test.jpg').first)).wont_equal nil
end
end
@ -154,26 +154,26 @@ describe 'site_files' do
PurgeCacheWorker.jobs.clear
@site.reload.space_used.must_equal initial_space_used + uploaded_file.size
@site.actual_space_used.must_equal @site.space_used
_(@site.reload.space_used).must_equal initial_space_used + uploaded_file.size
_(@site.actual_space_used).must_equal @site.space_used
file_path = @site.files_path 'test.jpg'
File.exists?(file_path).must_equal true
_(File.exists?(file_path)).must_equal true
delete_file filename: 'test.jpg'
File.exists?(file_path).must_equal false
SiteFile[site_id: @site.id, path: 'test.jpg'].must_be_nil
@site.reload.space_used.must_equal initial_space_used
@site.actual_space_used.must_equal @site.space_used
_(File.exists?(file_path)).must_equal false
_(SiteFile[site_id: @site.id, path: 'test.jpg']).must_be_nil
_(@site.reload.space_used).must_equal initial_space_used
_(@site.actual_space_used).must_equal @site.space_used
args = PurgeCacheWorker.jobs.first['args']
args[0].must_equal @site.username
args[1].must_equal '/test.jpg'
_(args[0]).must_equal @site.username
_(args[1]).must_equal '/test.jpg'
end
it 'property deletes directories with regexp special chars in them' do
upload 'dir' => '8)', 'files[]' => Rack::Test::UploadedFile.new('./tests/files/test.jpg', 'image/jpeg')
delete_file filename: '8)'
@site.reload.site_files.select {|f| f.path =~ /#{Regexp.quote '8)'}/}.length.must_equal 0
_(@site.reload.site_files.select {|f| f.path =~ /#{Regexp.quote '8)'}/}.length).must_equal 0
end
it 'deletes with escaped apostrophe' do
@ -181,9 +181,9 @@ describe 'site_files' do
'dir' => "test'ing",
'files[]' => Rack::Test::UploadedFile.new('./tests/files/test.jpg', 'image/jpeg')
)
@site.reload.site_files.select {|s| s.path == "test'ing"}.length.must_equal 1
_(@site.reload.site_files.select {|s| s.path == "test'ing"}.length).must_equal 1
delete_file filename: "test'ing"
@site.reload.site_files.select {|s| s.path == "test'ing"}.length.must_equal 0
_(@site.reload.site_files.select {|s| s.path == "test'ing"}.length).must_equal 0
end
it 'deletes a directory and all files in it' do
@ -199,11 +199,11 @@ describe 'site_files' do
space_used = @site.reload.space_used
delete_file filename: 'test'
@site.reload.space_used.must_equal(space_used - File.size('./tests/files/test.jpg'))
_(@site.reload.space_used).must_equal(space_used - File.size('./tests/files/test.jpg'))
@site.site_files.select {|f| f.path == 'test'}.length.must_equal 0
@site.site_files.select {|f| f.path =~ /^test\//}.length.must_equal 0
@site.site_files.select {|f| f.path =~ /^test.jpg/}.length.must_equal 1
_(@site.site_files.select {|f| f.path == 'test'}.length).must_equal 0
_(@site.site_files.select {|f| f.path =~ /^test\//}.length).must_equal 0
_(@site.site_files.select {|f| f.path =~ /^test.jpg/}.length).must_equal 1
end
it 'deletes records for nested directories' do
@ -215,7 +215,7 @@ describe 'site_files' do
expected_site_file_paths = ['derp', 'derp/ing', 'derp/ing/tons', 'derp/ing/tons/test.jpg']
expected_site_file_paths.each do |path|
@site.site_files.select {|f| f.path == path}.length.must_equal 1
_(@site.site_files.select {|f| f.path == path}.length).must_equal 1
end
delete_file filename: 'derp'
@ -223,7 +223,7 @@ describe 'site_files' do
@site.reload
expected_site_file_paths.each do |path|
@site.site_files.select {|f| f.path == path}.length.must_equal 0
_(@site.site_files.select {|f| f.path == path}.length).must_equal 0
end
end
@ -233,71 +233,71 @@ describe 'site_files' do
'files[]' => Rack::Test::UploadedFile.new('./tests/files/test.jpg', 'image/jpeg')
)
delete_file filename: 'test/test.jpg'
last_response.headers['Location'].must_equal "http://example.org/dashboard?dir=test"
_(last_response.headers['Location']).must_equal "http://example.org/dashboard?dir=test"
upload(
'files[]' => Rack::Test::UploadedFile.new('./tests/files/test.jpg', 'image/jpeg')
)
delete_file filename: 'test.jpg'
last_response.headers['Location'].must_equal "http://example.org/dashboard"
_(last_response.headers['Location']).must_equal "http://example.org/dashboard"
end
end
describe 'upload' do
it 'works with empty files' do
upload 'files[]' => Rack::Test::UploadedFile.new('./tests/files/empty.js', 'text/javascript')
File.exists?(@site.files_path('empty.js')).must_equal true
_(File.exists?(@site.files_path('empty.js'))).must_equal true
end
it 'manages files with invalid UTF8' do
upload 'files[]' => Rack::Test::UploadedFile.new('./tests/files/invalidutf8.html', 'text/html')
File.exists?(@site.files_path('invalidutf8.html')).must_equal true
_(File.exists?(@site.files_path('invalidutf8.html'))).must_equal true
end
it 'works with manifest files' do
upload 'files[]' => Rack::Test::UploadedFile.new('./tests/files/cache.manifest', 'text/cache-manifest')
File.exists?(@site.files_path('cache.manifest')).must_equal true
_(File.exists?(@site.files_path('cache.manifest'))).must_equal true
end
it 'works with otf fonts' do
upload 'files[]' => Rack::Test::UploadedFile.new('./tests/files/chunkfive.otf', 'application/vnd.ms-opentype')
File.exists?(@site.files_path('chunkfive.otf')).must_equal true
_(File.exists?(@site.files_path('chunkfive.otf'))).must_equal true
end
it 'succeeds with index.html file' do
@site.site_changed.must_equal false
_(@site.site_changed).must_equal false
upload 'files[]' => Rack::Test::UploadedFile.new('./tests/files/index.html', 'text/html')
last_response.body.must_match /successfully uploaded/i
File.exists?(@site.files_path('index.html')).must_equal true
_(last_response.body).must_match /successfully uploaded/i
_(File.exists?(@site.files_path('index.html'))).must_equal true
args = ScreenshotWorker.jobs.first['args']
args.first.must_equal @site.username
args.last.must_equal 'index.html'
@site.title.must_equal "The web site of #{@site.username}"
_(args.first).must_equal @site.username
_(args.last).must_equal 'index.html'
_(@site.title).must_equal "The web site of #{@site.username}"
@site.reload
@site.site_changed.must_equal true
@site.title.must_equal 'Hello?'
_(@site.site_changed).must_equal true
_(@site.title).must_equal 'Hello?'
# Purge cache needs to flush / and index.html for either scenario.
PurgeCacheWorker.jobs.length.must_equal 3
_(PurgeCacheWorker.jobs.length).must_equal 3
first_purge = PurgeCacheWorker.jobs.first
surf_purge = PurgeCacheWorker.jobs[1]
dirname_purge = PurgeCacheWorker.jobs.last
username, pathname = first_purge['args']
username.must_equal @site.username
pathname.must_equal '/index.html'
_(username).must_equal @site.username
_(pathname).must_equal '/index.html'
surf_purge['args'].last.must_equal '/?surf=1'
_(surf_purge['args'].last).must_equal '/?surf=1'
username, pathame = nil
username, pathname = dirname_purge['args']
username.must_equal @site.username
pathname.must_equal '/'
_(username).must_equal @site.username
_(pathname).must_equal '/'
@site.space_used.must_equal @site.actual_space_used
_(@site.space_used).must_equal @site.actual_space_used
(@site.space_used > 0).must_equal true
_((@site.space_used > 0)).must_equal true
end
it 'provides the correct space used after overwriting an existing file' do
@ -306,8 +306,8 @@ describe 'site_files' do
upload 'files[]' => uploaded_file
second_uploaded_file = Rack::Test::UploadedFile.new('./tests/files/img/test.jpg', 'image/jpeg')
upload 'files[]' => second_uploaded_file
@site.reload.space_used.must_equal initial_space_used + second_uploaded_file.size
@site.space_used.must_equal @site.actual_space_used
_(@site.reload.space_used).must_equal initial_space_used + second_uploaded_file.size
_(@site.space_used).must_equal @site.actual_space_used
end
it 'does not change title for subdir index.html' do
@ -316,7 +316,7 @@ describe 'site_files' do
'dir' => 'derpie',
'files[]' => Rack::Test::UploadedFile.new('./tests/files/index.html', 'text/html')
)
@site.reload.title.must_equal title
_(@site.reload.title).must_equal title
end
it 'purges cache for /subdir/' do # (not /subdir which is just a redirect to /subdir/)
@ -324,7 +324,7 @@ describe 'site_files' do
'dir' => 'subdir',
'files[]' => Rack::Test::UploadedFile.new('./tests/files/index.html', 'text/html')
)
PurgeCacheWorker.jobs.select {|j| j['args'].last == '/subdir/'}.length.must_equal 1
_(PurgeCacheWorker.jobs.select {|j| j['args'].last == '/subdir/'}.length).must_equal 1
end
it 'succeeds with multiple files' do
@ -336,51 +336,51 @@ describe 'site_files' do
]
)
@site.site_files.select {|s| s.path == 'one'}.length.must_equal 1
@site.site_files.select {|s| s.path == 'one/test.jpg'}.length.must_equal 1
@site.site_files.select {|s| s.path == 'two'}.length.must_equal 1
@site.site_files.select {|s| s.path == 'two/test.jpg'}.length.must_equal 1
_(@site.site_files.select {|s| s.path == 'one'}.length).must_equal 1
_(@site.site_files.select {|s| s.path == 'one/test.jpg'}.length).must_equal 1
_(@site.site_files.select {|s| s.path == 'two'}.length).must_equal 1
_(@site.site_files.select {|s| s.path == 'two/test.jpg'}.length).must_equal 1
end
it 'succeeds with valid file' do
initial_space_used = @site.space_used
uploaded_file = Rack::Test::UploadedFile.new('./tests/files/test.jpg', 'image/jpeg')
upload 'files[]' => uploaded_file
last_response.body.must_match /successfully uploaded/i
File.exists?(@site.files_path('test.jpg')).must_equal true
_(last_response.body).must_match /successfully uploaded/i
_(File.exists?(@site.files_path('test.jpg'))).must_equal true
username, path = PurgeCacheWorker.jobs.first['args']
username.must_equal @site.username
path.must_equal '/test.jpg'
_(username).must_equal @site.username
_(path).must_equal '/test.jpg'
@site.reload
@site.space_used.wont_equal 0
@site.space_used.must_equal initial_space_used + uploaded_file.size
@site.space_used.must_equal @site.actual_space_used
_(@site.space_used).wont_equal 0
_(@site.space_used).must_equal initial_space_used + uploaded_file.size
_(@site.space_used).must_equal @site.actual_space_used
ThumbnailWorker.jobs.length.must_equal 1
_(ThumbnailWorker.jobs.length).must_equal 1
ThumbnailWorker.drain
Site::THUMBNAIL_RESOLUTIONS.each do |resolution|
File.exists?(@site.thumbnail_path('test.jpg', resolution)).must_equal true
_(File.exists?(@site.thumbnail_path('test.jpg', resolution))).must_equal true
end
@site.site_changed.must_equal false
_(@site.site_changed).must_equal false
end
it 'sets site changed to false if index is empty' do
uploaded_file = Rack::Test::UploadedFile.new('./tests/files/blankindex/index.html', 'text/html')
upload 'files[]' => uploaded_file
last_response.body.must_match /successfully uploaded/i
@site.empty_index?.must_equal true
@site.site_changed.must_equal false
_(last_response.body).must_match /successfully uploaded/i
_(@site.empty_index?).must_equal true
_(@site.site_changed).must_equal false
end
it 'fails with unsupported file' do
upload 'files[]' => Rack::Test::UploadedFile.new('./tests/files/flowercrime.wav', 'audio/x-wav')
last_response.body.must_match /only supported by.+supporter account/i
File.exists?(@site.files_path('flowercrime.wav')).must_equal false
@site.site_changed.must_equal false
_(last_response.body).must_match /only supported by.+supporter account/i
_(File.exists?(@site.files_path('flowercrime.wav'))).must_equal false
_(@site.site_changed).must_equal false
end
it 'succeeds for unwhitelisted file on supporter plans' do
@ -388,20 +388,20 @@ describe 'site_files' do
no_file_restriction_plans.each do |plan_type,hash|
@site = Fabricate :site, plan_type: plan_type
upload 'files[]' => Rack::Test::UploadedFile.new('./tests/files/flowercrime.wav', 'audio/x-wav')
last_response.body.must_match /successfully uploaded/i
File.exists?(@site.files_path('flowercrime.wav')).must_equal true
_(last_response.body).must_match /successfully uploaded/i
_(File.exists?(@site.files_path('flowercrime.wav'))).must_equal true
end
end
it 'overwrites existing file with new file' do
upload 'files[]' => Rack::Test::UploadedFile.new('./tests/files/test.jpg', 'image/jpeg')
last_response.body.must_match /successfully uploaded/i
_(last_response.body).must_match /successfully uploaded/i
digest = @site.reload.site_files.first.sha1_hash
upload 'files[]' => Rack::Test::UploadedFile.new('./tests/files/img/test.jpg', 'image/jpeg')
last_response.body.must_match /successfully uploaded/i
@site.reload.changed_count.must_equal 2
@site.site_files.select {|f| f.path == 'test.jpg'}.length.must_equal 1
digest.wont_equal @site.site_files_dataset.where(path: 'test.jpg').first.sha1_hash
_(last_response.body).must_match /successfully uploaded/i
_(@site.reload.changed_count).must_equal 2
_(@site.site_files.select {|f| f.path == 'test.jpg'}.length).must_equal 1
_(digest).wont_equal @site.site_files_dataset.where(path: 'test.jpg').first.sha1_hash
end
it 'works with directory path' do
@ -409,24 +409,24 @@ describe 'site_files' do
'dir' => 'derpie/derptest',
'files[]' => Rack::Test::UploadedFile.new('./tests/files/test.jpg', 'image/jpeg')
)
last_response.body.must_match /successfully uploaded/i
File.exists?(@site.files_path('derpie/derptest/test.jpg')).must_equal true
_(last_response.body).must_match /successfully uploaded/i
_(File.exists?(@site.files_path('derpie/derptest/test.jpg'))).must_equal true
PurgeCacheWorker.jobs.length.must_equal 1
_(PurgeCacheWorker.jobs.length).must_equal 1
username, path = PurgeCacheWorker.jobs.first['args']
username.must_equal @site.username
path.must_equal '/derpie/derptest/test.jpg'
_(username).must_equal @site.username
_(path).must_equal '/derpie/derptest/test.jpg'
ThumbnailWorker.jobs.length.must_equal 1
_(ThumbnailWorker.jobs.length).must_equal 1
ThumbnailWorker.drain
@site.site_files_dataset.where(path: 'derpie').count.must_equal 1
@site.site_files_dataset.where(path: 'derpie/derptest').count.must_equal 1
@site.site_files_dataset.where(path: 'derpie/derptest/test.jpg').count.must_equal 1
_(@site.site_files_dataset.where(path: 'derpie').count).must_equal 1
_(@site.site_files_dataset.where(path: 'derpie/derptest').count).must_equal 1
_(@site.site_files_dataset.where(path: 'derpie/derptest/test.jpg').count).must_equal 1
Site::THUMBNAIL_RESOLUTIONS.each do |resolution|
File.exists?(@site.thumbnail_path('derpie/derptest/test.jpg', resolution)).must_equal true
@site.thumbnail_url('derpie/derptest/test.jpg', resolution).must_equal(
_(File.exists?(@site.thumbnail_path('derpie/derptest/test.jpg', resolution))).must_equal true
_(@site.thumbnail_url('derpie/derptest/test.jpg', resolution)).must_equal(
File.join "#{Site::THUMBNAILS_URL_ROOT}", Site.sharding_dir(@site.username), @site.username, "/derpie/derptest/test.jpg.#{resolution}.jpg"
)
end
@ -437,13 +437,13 @@ describe 'site_files' do
'dir' => 'derpie/derptest',
'files[]' => Rack::Test::UploadedFile.new('./tests/files/test.jpg', 'image/jpeg')
)
@site.reload.site_changed.must_equal false
_(@site.reload.site_changed).must_equal false
upload 'files[]' => Rack::Test::UploadedFile.new('./tests/files/index.html', 'text/html')
@site.reload.site_changed.must_equal true
_(@site.reload.site_changed).must_equal true
upload 'files[]' => Rack::Test::UploadedFile.new('./tests/files/chunkfive.otf', 'application/vnd.ms-opentype')
@site.reload.site_changed.must_equal true
_(@site.reload.site_changed).must_equal true
end
it 'does not store new file if hash matches' do
@ -451,22 +451,22 @@ describe 'site_files' do
'dir' => 'derpie/derptest',
'files[]' => Rack::Test::UploadedFile.new('./tests/files/test.jpg', 'image/jpeg')
)
@site.reload.changed_count.must_equal 1
_(@site.reload.changed_count).must_equal 1
upload(
'dir' => 'derpie/derptest',
'files[]' => Rack::Test::UploadedFile.new('./tests/files/test.jpg', 'image/jpeg')
)
@site.reload.changed_count.must_equal 1
_(@site.reload.changed_count).must_equal 1
upload 'files[]' => Rack::Test::UploadedFile.new('./tests/files/index.html', 'text/html')
@site.reload.changed_count.must_equal 2
_(@site.reload.changed_count).must_equal 2
end
describe 'directory create' do
it 'scrubs ../ from directory' do
@site.create_directory '../../test'
@site.site_files.select {|site_file| site_file.path =~ /\.\./}.length.must_equal 0
_(@site.site_files.select {|site_file| site_file.path =~ /\.\./}.length).must_equal 0
end
end
@ -485,9 +485,9 @@ describe 'site_files' do
@site.train 'spam.html', 'spam'
@site.train 'phishing.html', 'phishing'
@site.classify('ham.html').must_equal 'ham'
@site.classify('spam.html').must_equal 'spam'
@site.classify('phishing.html').must_equal 'phishing'
_(@site.classify('ham.html')).must_equal 'ham'
_(@site.classify('spam.html')).must_equal 'spam'
_(@site.classify('phishing.html')).must_equal 'phishing'
end
=end
end