class TestControl(TestCase):
def setUp(self):
mock_opts = [("--user", "test"), ("--pwd", "test")]
self.control = Control(mock_opts, MockObject(MockUser()))
def mock_function(self):
return True
def test_run_no_mode(self):
self.control.debug = True
self.assertIsNone(self.control.run())
self.control.debug = False
self.assertIsNone(self.control.run())
def test_run_help(self):
mock_opts = [("--help", "help")]
self.control = Control(mock_opts, MockObject(MockUser()))
self.assertIsNone(self.control.run())
def test_set_help_and_overwrite_flags(self):
mock_opts = [("--help", "help")]
self.control = Control(mock_opts, MockObject(MockUser()))
self.control.set_help_and_overwrite_flags()
self.assertTrue(self.control.help)
mock_opts = [("--overwrite", "overwrite")]
self.control = Control(mock_opts, MockObject(MockUser()))
self.control.set_help_and_overwrite_flags()
self.assertTrue(self.control.overwrite)
def test_check_entry_exists(self):
self.control = Control([("--mode", "get")], MockObject(MockUser()))
self.assertTrue(self.control.check_entry_exists())
self.control.client.get_one = lambda x, y: MockResponse()
self.assertFalse(self.control.check_entry_exists())
def test_run(self):
self.control = Control([("--mode", "get")], MockObject(MockUser()))
self.control.api_url = "local"
self.control.get = self.mock_function
self.assertTrue(self.control.run())
self.control = Control([("--mode", "post")], MockObject(MockUser()))
self.control.post = self.mock_function
self.assertTrue(self.control.run())
self.control = Control([("--mode", "delete")], MockObject(MockUser()))
self.control.delete = self.mock_function
self.assertTrue(self.control.run())
self.control = Control([("--mode", "foo")], MockObject(MockUser()))
self.assertIsNone(self.control.run())
def test_get(self):
self.control.client = MockObject(MockUser())
self.assertIsNone(self.control.get())
self.control.pdb_id = "foo"
self.assertIsNone(self.control.get())
self.control.resource = "bar"
self.assertIsNotNone(self.control.get())
def test_post(self):
self.control.client = MockObject(MockUser())
self.assertIsNone(self.control.post())
self.control.path = ".json"
self.assertIsNotNone(self.control.post())
os.system("touch foo.json")
self.control.path = "./"
self.assertIsNotNone(self.control.post())
os.system("rm foo.json")
def test_batch_post(self):
self.control.client = MockObject(MockUser())
self.assertIsNone(self.control.batch_post())
def test_delete(self):
self.control.client = MockObject(MockUser())
self.assertIsNotNone(self.control.delete())
def test_single_validate(self):
self.assertFalse(self.control.single_validate(None))
self.assertTrue(self.control.single_validate("path/to/json"))
def test_validate(self):
self.control.path = None
self.assertFalse(self.control.validate())
self.control.path = "path/to/file.json"
self.assertTrue(self.control.validate())
self.control.path = "tests/"
self.assertTrue(self.control.validate())
self.control.single_validate = lambda x: False
self.assertFalse(self.control.validate())
class TestControl(TestCase):
def setUp(self):
mock_opts = [("--user", "test"), ("--pwd", "test")]
self.control = Control(mock_opts, MockObject(MockUser()))
# self.control.user_name = None
# self.control.pwd = None
# self.control.mode = None
# self.control.pdb_id = None
# self.control.resource = None
# self.control.path = None
def test_run_no_mode(self):
self.control.debug = True
self.assertIsNone(self.control.run())
self.control.debug = False
self.assertIsNone(self.control.run())
def test_run_help(self):
mock_opts = [("--help", "help")]
self.control = Control(mock_opts, MockObject(MockUser()))
self.assertIsNone(self.control.run())
def mock_function(self):
return True
def test_run(self):
self.control = Control([("--mode", "get")], MockObject(MockUser()))
self.control.get = self.mock_function
self.assertTrue(self.control.run())
self.control = Control([("--mode", "put")], MockObject(MockUser()))
self.control.put = self.mock_function
self.assertTrue(self.control.run())
self.control = Control([("--mode", "post")], MockObject(MockUser()))
self.control.post = self.mock_function
self.assertTrue(self.control.run())
self.control = Control([("--mode", "delete")], MockObject(MockUser()))
self.control.delete = self.mock_function
self.assertTrue(self.control.run())
self.control = Control([("--mode", "foo")], MockObject(MockUser()))
self.assertIsNone(self.control.run())
def test_get(self):
self.control.client = MockObject(MockUser())
self.assertIsNotNone(self.control.get())
self.control.pdb_id = "foo"
self.assertIsNotNone(self.control.get())
def test_post(self):
self.control.client = MockObject(MockUser())
self.assertIsNone(self.control.post())
self.control.path = ".json"
self.assertIsNotNone(self.control.post())
os.system("touch foo.json")
self.control.path = "./"
self.assertIsNotNone(self.control.post())
os.system("rm foo.json")
def test_put(self):
self.control.client = MockObject(MockUser())
self.assertIsNone(self.control.put())
self.control.path = ".json"
self.assertIsNotNone(self.control.put())
def test_delete(self):
self.control.client = MockObject(MockUser())
self.assertIsNotNone(self.control.delete())
