def test_get_data_from_delayedTestData_set_done_later(self): """Test that data can be extracted from delayedTestData without making it done until later""" generated_uuid = str(uuid.uuid4()) timestamp = datetime.datetime.fromtimestamp(time.time()).isoformat() expected_result = "test_of_delayed_test+%s" % timestamp test_id = "DelayedUnitTestSetDoneLater" # First clear any leftover rows from previous unittests of this DelayedTestAssistant.get_delayed_results(test_id=test_id, seconds=1, set_done=True) # Now begin putting in new data DelayedTestStorage.insert_delayed_test_data(generated_uuid, test_id, expected_result, 0, 1000000000000) time.sleep(1.2) for item in DelayedTestAssistant.get_delayed_results(test_id=test_id, seconds=1, set_done=False): pass # making sure data was not set to done (if not working, no data would come later) found = False for item in DelayedTestAssistant.get_delayed_results(test_id=test_id, seconds=1, set_done=False): found = True extracted_guid = item[0] self.assertEqual(extracted_guid, generated_uuid) DelayedTestAssistant.set_test_done(extracted_guid) self.assertTrue(found) found = False for item in DelayedTestAssistant.get_delayed_results(test_id=test_id, seconds=1, set_done=False): found = True self.assertFalse(found) # Should not find anything this time DelayedTestAssistant.get_delayed_results(test_id=test_id, seconds=1, set_done=True) # Clean Up
def test_add_row_delayedTestData(self): """Test that data can be inserted into and extracted from delayedTestData""" generated_uuid = str(uuid.uuid4()) timestamp = datetime.datetime.fromtimestamp(time.time()).isoformat() expected_result = "test_of_delayed_test+%s" % timestamp test_id = "DelayedUnitTestAddRow" DelayedTestStorage.insert_delayed_test_data(generated_uuid, test_id, expected_result, 0, 1000000000000) # Now check that we can pull the data out data = DelayedTestStorage.get_delayed_test_data(testcase_address=test_id) extracted_uuid = None self.assertTrue(data is not None) for item in data: if ((item[3] == expected_result)): extracted_uuid = item[0] break self.assertEqual(generated_uuid, extracted_uuid) # Check that we can change the data DelayedTestStorage.set_delayed_test_to_done(generated_uuid) data = DelayedTestStorage.get_delayed_test_data(testcase_address=test_id, done=1) self.assertTrue(data is not None) for item in data: if (item[0] == generated_uuid): self.assertEqual(item[4], 1) break DelayedTestAssistant.get_delayed_results(test_id=test_id, seconds=1, set_done=True) # Clean Up