| OLD | NEW |
| 1 # Copyright (C) 2013 Google Inc. All rights reserved. | 1 # Copyright (C) 2013 Google Inc. All rights reserved. |
| 2 # | 2 # |
| 3 # Redistribution and use in source and binary forms, with or without | 3 # Redistribution and use in source and binary forms, with or without |
| 4 # modification, are permitted provided that the following conditions are | 4 # modification, are permitted provided that the following conditions are |
| 5 # met: | 5 # met: |
| 6 # | 6 # |
| 7 # * Redistributions of source code must retain the above copyright | 7 # * Redistributions of source code must retain the above copyright |
| 8 # notice, this list of conditions and the following disclaimer. | 8 # notice, this list of conditions and the following disclaimer. |
| 9 # * Redistributions in binary form must reproduce the above | 9 # * Redistributions in binary form must reproduce the above |
| 10 # copyright notice, this list of conditions and the following disclaimer | 10 # copyright notice, this list of conditions and the following disclaimer |
| (...skipping 39 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 50 | 50 |
| 51 self.assertTrue(self.test_file.save_data(test_data)) | 51 self.assertTrue(self.test_file.save_data(test_data)) |
| 52 self.assertEqual(test_data, self.test_file.data) | 52 self.assertEqual(test_data, self.test_file.data) |
| 53 | 53 |
| 54 self.assertEqual(test_data, self.test_file.load_data()) | 54 self.assertEqual(test_data, self.test_file.load_data()) |
| 55 self.assertEqual(test_data, self.test_file.data) | 55 self.assertEqual(test_data, self.test_file.data) |
| 56 | 56 |
| 57 self.test_file.delete_data() | 57 self.test_file.delete_data() |
| 58 self.assertFalse(self.test_file.load_data()) | 58 self.assertFalse(self.test_file.load_data()) |
| 59 | 59 |
| 60 def testLoadDataInvalidKey(self): |
| 61 test_data = 'x' * datastorefile.MAX_ENTRY_LEN * 3 |
| 62 |
| 63 self.assertTrue(self.test_file.save_data(test_data)) |
| 64 self.assertEqual(test_data, self.test_file.data) |
| 65 |
| 66 self.test_file.delete_data() |
| 67 self.assertEqual('', self.test_file.load_data()) |
| 68 |
| 69 def testLoadDataNoKeys(self): |
| 70 # This should never happen. |
| 71 self.assertEqual(None, self.test_file.load_data()) |
| 72 |
| 60 def testSaveData(self): | 73 def testSaveData(self): |
| 61 self.assertFalse(self.test_file.save_data(None)) | 74 self.assertFalse(self.test_file.save_data(None)) |
| 62 | 75 |
| 63 too_big_data = 'x' * (datastorefile.MAX_DATA_ENTRY_PER_FILE * datastoref
ile.MAX_ENTRY_LEN + 1) | 76 too_big_data = 'x' * (datastorefile.MAX_DATA_ENTRY_PER_FILE * datastoref
ile.MAX_ENTRY_LEN + 1) |
| 64 self.assertFalse(self.test_file.save_data(too_big_data)) | 77 self.assertFalse(self.test_file.save_data(too_big_data)) |
| 65 | 78 |
| 66 test_data = 'x' * datastorefile.MAX_ENTRY_LEN * 5 | 79 test_data = 'x' * datastorefile.MAX_ENTRY_LEN * 5 |
| 67 self.assertTrue(self.test_file.save_data(test_data)) | 80 self.assertTrue(self.test_file.save_data(test_data)) |
| 68 nchunks = datastorefile.DataEntry.all().count() | 81 nchunks = datastorefile.DataEntry.all().count() |
| 69 nkeys = len(self.test_file.data_keys) + len(self.test_file.new_data_keys
) | 82 nkeys = len(self.test_file.data_keys) + len(self.test_file.new_data_keys
) |
| (...skipping 24 matching lines...) Expand all Loading... |
| 94 self.assertEqual(len(chunk_indices), 3) | 107 self.assertEqual(len(chunk_indices), 3) |
| 95 self.assertNotEqual(chunk_indices[0], chunk_indices[-1]) | 108 self.assertNotEqual(chunk_indices[0], chunk_indices[-1]) |
| 96 | 109 |
| 97 data_length += 1 | 110 data_length += 1 |
| 98 chunk_indices = self.test_file._get_chunk_indices(data_length) | 111 chunk_indices = self.test_file._get_chunk_indices(data_length) |
| 99 self.assertEqual(len(chunk_indices), 4) | 112 self.assertEqual(len(chunk_indices), 4) |
| 100 | 113 |
| 101 | 114 |
| 102 if __name__ == '__main__': | 115 if __name__ == '__main__': |
| 103 unittest.main() | 116 unittest.main() |
| OLD | NEW |