completed sync and add tests

metobscommon/tests/archive/unit_tests.py

@@ -13,15 +13,15 @@ DATABASE = 'test.db'
 
 class TestCreate(object):
     """Test create functionary."""
-    @pytest.fixture(autouse=True)
-    def setup(self, tmpdir):
-        self.tmpdir = tmpdir.strpath
-        self.database = self.tmpdir = '/' + DATABASE
-        if os.path.isfile(self.database):
-            os.remove(self.database)
+    # @pytest.fixture(autouse=True)
+    def setup(self):
+        # self.tmpdir = tmpdir.strpath
+        # self.database = self.tmpdir = '/' + DATABASE
+        if os.path.isfile(DATABASE):
+            os.remove(DATABASE)
 
     def teardown(self):
-        os.remove(self.database)
+        os.remove(DATABASE)
 
     def setup_class(cls):
         pass
@@ -38,15 +38,15 @@ class TestCreate(object):
     def test_create_file_exists(self):
         """Test create functionality by checking if the database file exists."""
         #simulate 'python db.py test.db create'
-        args = db.parse_args([self.database, 'create'])
+        args = db.parse_args([DATABASE, 'create'])
         args.func(args)
         
-        assert os.path.exists(self.database)
+        assert os.path.exists(DATABASE)
         
     def test_create_query(self):
         """Test create functionality by querying the entire database and comparing results."""
         #simulate 'python db.py test.db create'
-        args = db.parse_args([self.database, 'create'])
+        args = db.parse_args([DATABASE, 'create'])
         args.func(args)
         
         #test input for querying Instrument, Site, Experiment, and FileType
@@ -67,18 +67,15 @@ class TestCreate(object):
 
 
         '''
-        
         #remove indentation
         test_input_str = test_input_str.replace('        ', '')
-        
         #put test input into a generator that returns the string line-by-line
         test_input_generator = iter(test_input_str.split('\n'))
         
         #replace default input function with generator reading line-by-line from string
         with mock.patch.object(db, 'input', lambda x: next(test_input_generator)):
-        
             #simulate 'python db.py test.db query -i -s -e -t'
-            args = db.parse_args([self.database, 'query', '-i', '-s', '-e', '-t'])
+            args = db.parse_args([DATABASE, 'query', '-i', '-s', '-e', '-t'])
             queries = args.func(args)
         
         #find how many query results are found total
@@ -93,12 +90,16 @@ class TestCreate(object):
         assert num_queries == expected_num_queries
 
 class TestSync(object):
-
+    """Test sync functionality."""
     def setup(self):
-        pass
+        if os.path.isfile(DATABASE):
+            os.remove(DATABASE)
+        #simulate 'python db.py test.db create'
+        args = db.parse_args([DATABASE, 'create'])
+        args.func(args)
 
     def teardown(self):
-        pass
+        os.remove(DATABASE)
 
     def setup_class(cls):
         pass
@@ -111,6 +112,62 @@ class TestSync(object):
 
     def teardown_method(self, method):
         pass
+        
+    def test_sync_basic(self):
+        """Test sync by running a query before and after to ensure files are properly added."""
+        #test input for query
+        test_input_str = '''
+        
+        
+        
+        
+        
+        '''
+        #remove indentation from test input
+        test_input_str = test_input_str.replace('        ', '')
+        #create generator of the test input that reads it one line at a time
+        test_input_generator = iter(test_input_str.split('\n'))
+        #replace the default input with generator reading line-by-line from a string
+        with mock.patch.object(db, 'input', lambda x: next(test_input_generator)):
+            #simulate 'python db.py test.db query -f'
+            args = db.parse_args([DATABASE, 'query', '-f'])
+            queries = args.func(args)
+            
+        #find the number of results from the query
+        num_results = 0
+        for col in queries.values():
+            for qry in col:
+                for val in qry:
+                    num_results += 1        
+        #expected number of results is 0
+        expected_num_results = 0
+        
+        #ensure the num_results is the same as the expected
+        assert num_results == expected_num_results
+        
+        #simulate 'python db.py test.db sync'
+        args = db.parse_args([DATABASE, 'sync'])
+        args.func(args)
+        
+        #re-run the query and compare with an updated expected value
+        test_input_generator = iter(test_input_str.split('\n'))
+        #replace the default input with generator reading line-by-line from a string
+        with mock.patch.object(db, 'input', lambda x: next(test_input_generator)):
+            #simulate 'python db.py test.db query -f'
+            args = db.parse_args([DATABASE, 'query', '-f'])
+            queries = args.func(args)
+            
+        #find the number of results from the query
+        num_results = 0
+        for col in queries.values():
+            for qry in col:
+                for val in qry:
+                    num_results += 1            
+        #expected number of results is 4
+        expected_num_results = 4
+        
+        #ensure the num_results is the same as the expected
+        assert num_results == expected_num_results
 
 class TestQuery(object):
     """Test query functionality.
@@ -195,7 +252,6 @@ class TestQuery(object):
 
         #replace default input function with generator reading line-by-line from string
         with mock.patch.object(db, 'input', lambda x: next(test_input_generator)):
-
             #simulate 'python db.py test.db query -i'
             args = db.parse_args([DATABASE, 'query', '-i', '-s', '-e', '-t', '-f'])
             results = args.func(args)
@@ -228,12 +284,16 @@ class TestQuery(object):
         assert query_result == query_expected
 
 class TestAdd(object):
-
+    """Test add functionality."""
     def setup(self):
-        pass
+        if os.path.isfile(DATABASE):
+            os.remove(DATABASE)
+        #simulate 'python db.py test.db create'
+        args = db.parse_args([DATABASE, 'create'])
+        args.func(args)
 
     def teardown(self):
-        pass
+        os.remove(DATABASE)
 
     def setup_class(cls):
         pass
@@ -246,3 +306,60 @@ class TestAdd(object):
 
     def teardown_method(self, method):
         pass
+        
+    def test_add_basic(self):
+        """Test add functionality by querying before and after adding two Files."""
+        #test input for query
+        test_input_str = '''
+        
+        
+        
+        
+        
+        '''
+        #remove indentation from test input
+        test_input_str = test_input_str.replace('        ', '')
+        #create generator of the test input that reads it one line at a time
+        test_input_generator = iter(test_input_str.split('\n'))
+        #replace the default input with generator reading line-by-line from a string
+        with mock.patch.object(db, 'input', lambda x: next(test_input_generator)):
+            #simulate 'python db.py test.db query -f'
+            args = db.parse_args([DATABASE, 'query', '-f'])
+            queries = args.func(args)
+            
+        #find the number of results from the query
+        num_results = 0
+        for col in queries.values():
+            for qry in col:
+                for val in qry:
+                    num_results += 1        
+        #expected number of results is 0
+        expected_num_results = 0
+        
+        #ensure the num_results is the same as the expected
+        assert num_results == expected_num_results
+        
+        #simulate 'python db.py test.db sync'
+        args = db.parse_args([DATABASE, 'add', '-i', 'test_inst_1', '-s', 'test_site_1', '-S'
+                            '2000-01-01', '-f', '000101.ascii', '000101.qc'])
+        args.func(args)
+        
+        #re-run the query and compare with an updated expected value
+        test_input_generator = iter(test_input_str.split('\n'))
+        #replace the default input with generator reading line-by-line from a string
+        with mock.patch.object(db, 'input', lambda x: next(test_input_generator)):
+            #simulate 'python db.py test.db query -f'
+            args = db.parse_args([DATABASE, 'query', '-f'])
+            queries = args.func(args)
+            
+        #find the number of results from the query
+        num_results = 0
+        for col in queries.values():
+            for qry in col:
+                for val in qry:
+                    num_results += 1            
+        #expected number of results is 2
+        expected_num_results = 2
+        
+        #ensure the num_results is the same as the expected
+        assert num_results == expected_num_results