Merge pull request #24 from bchew/add-basic-tests

Add basic tests using dynalite

Author: bchew

.travis.yml

@@ -4,8 +4,15 @@ python:
 install:
   - pip install -r requirements.txt
   - pip install flake8
+  - nvm install 4
+  - nvm use 4
+  - npm install dynalite
 script:
-  - flake8 --ignore=E501 .
-
+  - flake8 --ignore=E501 --exclude=node_modules
+  - dynalite --port 4567 &
+  - mkdir dump && cp -a test/testTable dump
+  - python dynamodump.py -m restore -r local -s testTable -d testRestoredTable --host localhost --port 4567 --accessKey a --secretKey a
+  - python dynamodump.py -m backup -r local -s testRestoredTable --host localhost --port 4567 --accessKey a --secretKey a
+  - python test/test.py
 matrix:
     fast_finish: true

test/test.py

@@ -0,0 +1,41 @@
+#!/usr/bin/env python
+import json
+import unittest
+
+TEST_DATA_PATH = "test/testTable"
+DUMP_DATA_PATH = "dump/testRestoredTable"
+SCHEMA_FILE = "schema.json"
+DATA_FILE = "0001.json"
+
+
+class TestDynamoDump(unittest.TestCase):
+
+    def setUp(self):
+        self.test_table_schema = json.load(open(TEST_DATA_PATH + "/" + SCHEMA_FILE))
+        self.restored_test_table_schema = json.load(open(DUMP_DATA_PATH + "/" + SCHEMA_FILE))
+        self.test_table_data = json.load(open(TEST_DATA_PATH + "/data/" + DATA_FILE))
+        self.restored_test_table_data = json.load(open(DUMP_DATA_PATH + "/data/" + DATA_FILE))
+
+    def test_schema(self):
+        self.assertEqual(self.test_table_schema["Table"]["AttributeDefinitions"],
+                         self.restored_test_table_schema["Table"]["AttributeDefinitions"])
+        self.assertEqual(self.test_table_schema["Table"]["ProvisionedThroughput"]["WriteCapacityUnits"],
+                         self.restored_test_table_schema["Table"]["ProvisionedThroughput"]["WriteCapacityUnits"])
+        self.assertEqual(self.test_table_schema["Table"]["ProvisionedThroughput"]["ReadCapacityUnits"],
+                         self.restored_test_table_schema["Table"]["ProvisionedThroughput"]["ReadCapacityUnits"])
+        self.assertEqual(self.test_table_schema["Table"]["KeySchema"],
+                         self.restored_test_table_schema["Table"]["KeySchema"])
+        self.assertEqual(self.test_table_schema["Table"]["TableSizeBytes"],
+                         self.restored_test_table_schema["Table"]["TableSizeBytes"])
+        self.assertEqual("testRestoredTable",
+                         self.restored_test_table_schema["Table"]["TableName"])
+        self.assertEqual(self.test_table_schema["Table"]["TableStatus"],
+                         self.restored_test_table_schema["Table"]["TableStatus"])
+        self.assertEqual(self.test_table_schema["Table"]["ItemCount"],
+                         self.restored_test_table_schema["Table"]["ItemCount"])
+
+    def test_data(self):
+        self.assertEqual(self.test_table_data, self.restored_test_table_data)
+
+if __name__ == '__main__':
+    unittest.main()

test/testTable/data/0001.json

@@ -0,0 +1,17 @@
+{
+  "Count": 1, 
+  "Items": [
+    {
+      "lastName": {
+        "S": "Doe"
+      }, 
+      "id": {
+        "N": "1"
+      }, 
+      "firstName": {
+        "S": "John"
+      }
+    }
+  ], 
+  "ScannedCount": 1
+}

test/testTable/schema.json

@@ -0,0 +1,28 @@
+{
+  "Table": {
+    "TableArn": "arn:aws:dynamodb:us-east-1:000000000000:table/testTable",
+    "AttributeDefinitions": [
+      {
+        "AttributeName": "id",
+        "AttributeType": "N"
+      }
+    ],
+    "ProvisionedThroughput": {
+      "WriteCapacityUnits": 25,
+      "NumberOfDecreasesToday": 1,
+      "ReadCapacityUnits": 1,
+      "LastDecreaseDateTime": 1471785062.495
+    },
+    "TableSizeBytes": 0,
+    "TableName": "testTable",
+    "TableStatus": "ACTIVE",
+    "KeySchema": [
+      {
+        "KeyType": "HASH",
+        "AttributeName": "id"
+      }
+    ],
+    "ItemCount": 0,
+    "CreationDateTime": 1471785060.962
+  }
+}