integrate with new ui

test/backend/storage-tests.js

@@ -20,14 +20,17 @@ class MockStorage {
   }
 }
 
-const expire_seconds = 10;
+const config = {
+  default_expire_seconds: 10,
+  num_of_buckets: 3,
+  expire_times_seconds: [86400, 604800, 1209600],
+  s3_buckets: ['foo', 'bar', 'baz'],
+  env: 'development',
+  redis_host: 'localhost'
+};
+
 const storage = proxyquire('../../server/storage', {
-  '../config': {
-    expire_seconds,
-    s3_bucket: 'foo',
-    env: 'development',
-    redis_host: 'localhost'
-  },
+  '../config': config,
   '../log': () => {},
   './s3': MockStorage
 });
@@ -35,39 +38,58 @@ const storage = proxyquire('../../server/storage', {
 describe('Storage', function() {
   describe('ttl', function() {
     it('returns milliseconds remaining', async function() {
-      await storage.set('x', null, { foo: 'bar' });
+      const time = 40;
+      await storage.set('x', null, { foo: 'bar' }, time);
       const ms = await storage.ttl('x');
       await storage.del('x');
-      assert.equal(ms, expire_seconds * 1000);
+      assert.equal(ms, time * 1000);
     });
   });
 
   describe('length', function() {
     it('returns the file size', async function() {
+      await storage.set('x', null);
       const len = await storage.length('x');
       assert.equal(len, 12);
     });
   });
 
   describe('get', function() {
-    it('returns a stream', function() {
-      const s = storage.get('x');
+    it('returns a stream', async function() {
+      await storage.set('x', null);
+      const s = await storage.get('x');
       assert.equal(s, stream);
     });
   });
 
   describe('set', function() {
-    it('sets expiration to config.expire_seconds', async function() {
-      await storage.set('x', null, { foo: 'bar' });
+    it('sets expiration to expire time', async function() {
+      const seconds = 100;
+      await storage.set('x', null, { foo: 'bar' }, seconds);
       const s = await storage.redis.ttlAsync('x');
       await storage.del('x');
-      assert.equal(Math.ceil(s), expire_seconds);
+      assert.equal(Math.ceil(s), seconds);
+    });
+
+    it('puts into right bucket based on expire time', async function() {
+      for (let i = 0; i < config.num_of_buckets; i++) {
+        await storage.set(
+          'x',
+          null,
+          { foo: 'bar' },
+          config.expire_times_seconds[i]
+        );
+        const bucket = await storage.getBucket('x');
+        assert.equal(bucket, i);
+        await storage.del('x');
+      }
     });
 
     it('sets metadata', async function() {
       const m = { foo: 'bar' };
       await storage.set('x', null, m);
       const meta = await storage.redis.hgetallAsync('x');
+      delete meta.bucket;
       await storage.del('x');
       assert.deepEqual(meta, m);
     });
@@ -77,6 +99,7 @@ describe('Storage', function() {
 
   describe('setField', function() {
     it('works', async function() {
+      await storage.set('x', null);
       storage.setField('x', 'y', 'z');
       const z = await storage.redis.hgetAsync('x', 'y');
       assert.equal(z, 'z');