Elgato_dark/zeppelin
1
0
Fork 0
mirror of https://github.com/apache/zeppelin synced 2026-05-24 09:38:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

improve tests

Browse source
This commit is contained in:
Khalid Huseynov 2016-09-20 13:14:13 +09:00
parent 139940743f
commit 6614e2bb52
1 changed files with 34 additions and 38 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

72
zeppelin-zengine/src/test/java/org/apache/zeppelin/notebook/NotebookTest.java
View file

@ -192,43 +192,6 @@ public class NotebookTest implements JobListenerFactory{
assertEquals(notes.size(), 0);
}
@Test
public void testReloadAuthorizedNotes() throws IOException {
AuthenticationInfo user1 = new AuthenticationInfo("user1");
AuthenticationInfo user2 = new AuthenticationInfo("user2");
notebook.reloadAllNotes(user1);
List<Note> notes1 = notebook.getAllNotes(user1);
notebook.reloadAllNotes(user2);
List<Note> notes2 = notebook.getAllNotes(user2);
assertEquals(notes1.size(), 0);
assertEquals(notes2.size(), 0);
logger.info("Loaded {} notes", notes1.size());
Note note = notebook.createNote(user1);
setNotePermissions(note.getId(), user1.getUser(), true, true, true);
notebook.reloadAllNotes(user1);
notes1 = notebook.getAllNotes(user1);
notebook.reloadAllNotes(user2);
notes2 = notebook.getAllNotes(user2);
assertEquals(notes1.size(), 1);
assertEquals(notes2.size(), 0);
}
//fails if all failed, otherwise true
private boolean setNotePermissions(String noteId, String user, boolean isOwner, boolean isReader, boolean isWriter) {
NotebookAuthorization notebookAuthorization = notebook.getNotebookAuthorization();
Set<String> owners = notebookAuthorization.getOwners(noteId);
Set<String> readers = notebookAuthorization.getReaders(noteId);
Set<String> writers = notebookAuthorization.getWriters(noteId);
boolean ownerSet = isOwner ? owners.add(user) : owners.remove(user);
boolean readerSet = isReader ? readers.add(user) : readers.remove(user);
boolean writerSet = isWriter ? writers.add(user) : writers.remove(user);
notebookAuthorization.setOwners(noteId, owners);
notebookAuthorization.setReaders(noteId, readers);
notebookAuthorization.setWriters(noteId, writers);
return ownerSet || readerSet || writerSet;
}
@Test
public void testPersist() throws IOException, SchedulerException, RepositoryException {
Note note = notebook.createNote(null);
@ -626,7 +589,7 @@ public class NotebookTest implements JobListenerFactory{
// create a note and a paragraph
Note note = notebook.createNote(null);
NotebookAuthorization notebookAuthorization = notebook.getNotebookAuthorization();
// empty owners, readers and writers means note is public
// empty owners, readers or writers means note is public
assertEquals(notebookAuthorization.isOwner(note.getId(),
new HashSet<String>(Arrays.asList("user2"))), true);
assertEquals(notebookAuthorization.isReader(note.getId(),
@ -911,6 +874,39 @@ public class NotebookTest implements JobListenerFactory{
assertEquals(1, notebook.getAllNotes(new AuthenticationInfo("user2")).size());
}
@Test
public void testGetAllNotesWithDifferentPermissions() throws IOException {
AuthenticationInfo user1 = new AuthenticationInfo("user1");
AuthenticationInfo user2 = new AuthenticationInfo("user2");
List<Note> notes1 = notebook.getAllNotes(user1);
List<Note> notes2 = notebook.getAllNotes(user2);
assertEquals(notes1.size(), 0);
assertEquals(notes2.size(), 0);
//creates note and sets user1 owner
Note note = notebook.createNote(user1);
// note is public since readers and writers empty
notes1 = notebook.getAllNotes(user1);
notes2 = notebook.getAllNotes(user2);
assertEquals(notes1.size(), 1);
assertEquals(notes2.size(), 1);
notebook.getNotebookAuthorization().setReaders(note.getId(), Sets.newHashSet("user1"));
//note is public since writers empty
notes1 = notebook.getAllNotes(user1);
notes2 = notebook.getAllNotes(user2);
assertEquals(notes1.size(), 1);
assertEquals(notes2.size(), 1);
notebook.getNotebookAuthorization().setWriters(note.getId(), Sets.newHashSet("user1"));
notes1 = notebook.getAllNotes(user1);
notes2 = notebook.getAllNotes(user2);
assertEquals(notes1.size(), 1);
assertEquals(notes2.size(), 0);
}
private void delete(File file){
if(file.isFile()) file.delete();
else if(file.isDirectory()){