def setUp(self): # Set up a socket client_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) sec_clisock = ssl.wrap_socket(client_sock) sec_clisock.settimeout(10) sec_clisock.connect((socket.gethostname(), 14338)) self.sock = sec_clisock # Authenticate sc.auth(self.sock, 'lev', 'lev')
def test_remove_shared(self): # Get a list of datasets dsets = sc.lst(self.sock)['list'] test_elt = dsets[0] # Share test_elt with bob sc.share(self.sock, test_elt, 'bob', 0) # auth as bob sc.auth(self.sock, 'bob', 'bob') # remove shared dataset sc.remove_shared(self.sock, test_elt, 'lev') # verify dataset was removed dsets = sc.lst(self.sock)["requests"] self.assertTrue(not (["lev", "bob", test_elt, 0, 1] in dsets), "Shared dataset removed successfully")
def test_rename_admin(self): # Get a dataset owned by bob sc.auth(self.sock, 'bob', 'bob') dsets = sc.lst(self.sock)['list'] test_elt = dsets[0] ren_elt = test_elt+'renamed' # Rename the dataset sc.auth(self.sock, 'lev', 'lev') sc.rename_admin(self.sock, test_elt, ren_elt, 'bob') # Verify the rename sc.auth(self.sock, 'bob', 'bob') dsets = sc.lst(self.sock)['list'] self.assertIn(ren_elt, dsets, "Other user's dataset not renamed!") # Put things back the way they were sc.rename(self.sock, ren_elt, test_elt) sc.auth(self.sock, 'lev', 'lev')
def test_remove_admin(self): # Get a dataset owned by bob sc.auth(self.sock, 'bob', 'bob') dsets = sc.lst(self.sock)['list'] test_elt = dsets[0] cop_elt = test_elt+'copied' sc.copy(self.sock, test_elt, cop_elt) # Verify the copied element is there dsets = sc.lst(self.sock)['list'] self.assertIn(cop_elt, dsets, "Other user's dataset not copied!") # Rename the dataset sc.auth(self.sock, 'lev', 'lev') sc.remove_admin(self.sock, cop_elt, 'bob') # Verify the rename sc.auth(self.sock, 'bob', 'bob') dsets = sc.lst(self.sock)['list'] self.assertNotIn(cop_elt, dsets, "Other user's dataset not removed!") # Put things back the way they were sc.auth(self.sock, 'lev', 'lev')