def test_session_assets(): store = AssetStore() s = SessionAssets(store) s._send_command = lambda x: None assert not s.get_used_asset_names() with open(test_filename, 'wb') as f: f.write(b'bar\n') # Add assets, check mangles name a1 = s.add_asset('foo.css', b'foo\n') a2 = s.add_asset('foo.js', test_filename) assert 'foo' in a1 and s.id in a1 and a1.endswith('.css') assert 'foo' in a2 and s.id in a2 and a2.endswith('.js') assert s.get_used_asset_names() == [a1, a2] # order in which it came # Get the asset raises(IndexError, store.load_asset, 'foo.css') raises(IndexError, store.load_asset, 'foo.js') assert store.load_asset(a1) == b'foo\n' assert store.load_asset(a2) == b'bar\n' # Use asset store.add_asset('spam.js', b'1234\x00') s.use_global_asset('spam.js') assert s.get_used_asset_names()[-1] == 'spam.js' raises(IndexError, s.use_global_asset, 'unknown-asset.js') raises(ValueError, s.add_asset, 3, b'a\n') # Add assets after loading page s.get_page() s.use_global_asset('spam.js') # prints a warning, but it does work # Global assets s.add_global_asset('eggs.js', b'12345\x00') assert s.get_used_asset_names()[-1] == 'eggs.js' assert store.load_asset('eggs.js') == b'12345\x00' raises(ValueError, s.use_global_asset, 3) # Remote assets s.use_remote_asset('http://linked.com/not/verified.js') s.use_remote_asset('http://linked.com/not/verified.css') s.use_remote_asset('http://linked.com/not/verified.css') # twice is ok raises(ValueError, s.use_remote_asset, 3) page = s.get_page() assert 'not/verified.js' in page assert 'not/verified.css' in page