コード例 #1
0
ファイル: test_database.py プロジェクト: goldenboy/shared
    def test__export_as_csv(self):
        t = DbObject(DBH.test)
        t.tbl_.truncate()
        t.db_.commit()

        t.company_id = COMPANY_ID
        t.name = NAME
        t.number = NUMBER
        t.amount = AMOUNT
        t.start_date = START_DATE
        t.status = STATUS
        self.assertTrue(t.add())  # Add succeeds

        output = cStringIO.StringIO()
        t.export_as_csv(out=output)

        # String fields are double quoted.

        self.assertEqual(
            """%s,%s,%s,"%s",%s,"%s","%s","%s","%s"\r\n"""
            % (t.id, COMPANY_ID, NUMBER, NAME, AMOUNT, START_DATE, STATUS, t.created_on, t.updated_on),
            output.getvalue(),
        )

        t.tbl_.truncate()
        t.db_.commit()
        return
コード例 #2
0
ファイル: test_database.py プロジェクト: goldenboy/shared
    def test__remove(self):
        def names():
            n = []
            rows = t.db_.executesql("""SELECT name FROM test;""")
            if len(rows) > 0:
                for r in rows:
                    if len(r) > 0:
                        n.append(r[0])
            return n

        t = DbObject(DBH.test)
        t.tbl_.truncate()
        t.db_.commit()

        t.company_id = COMPANY_ID
        t.name = NAME
        t.number = NUMBER
        t.start_date = START_DATE
        t.status = STATUS
        self.assertFalse(t.remove())  # Remove fails if record not added

        self.assertTrue(t.add())  # Add succeeds

        t2 = DbObject(DBH.test)
        t2.company_id = COMPANY_ID
        t2.name = NAME_2
        t2.number = NUMBER_2
        t2.start_date = START_DATE_2
        t2.status = STATUS_2
        self.assertTrue(t2.add())  # Add (2) succeeds

        # Table populated as expected
        self.assertEqual(names(), [NAME, NAME_2])

        self.assertTrue(t.remove())  # Remove returns success

        self.assertEqual(names(), [NAME_2])  # Table populated as expected

        self.assertTrue(t.add())  # Object can be re-added.

        self.assertTrue(t2.remove())  # Remove (2) returns success
        self.assertEqual(names(), [NAME])  # Table populated as expected
        self.assertTrue(t.remove())  # Remove returns success
        return
コード例 #3
0
ファイル: test_database.py プロジェクト: goldenboy/shared
 def test____deepcopy__(self):
     t = DbObject(DBH.test)
     t.company_id = COMPANY_ID
     t.number = NUMBER
     t.name = NAME
     t.amount = AMOUNT
     t.start_date = START_DATE
     t.status = STATUS
     t2 = copy.deepcopy(t)
     self.assertEqual(t, t2)
     return
コード例 #4
0
ファイル: test_database.py プロジェクト: goldenboy/shared
 def _obj(self, number=0):
     if number == 1:
         t = DbObject(DBH.test)
         t.company_id = COMPANY_ID
         t.number = NUMBER
         t.name = NAME
         t.amount = AMOUNT
         t.start_date = START_DATE
         t.status = STATUS
         self.assertTrue(t.add())  # Add test object
         return t
     if number == 2:
         t2 = DbObject(DBH.test)
         t2.company_id = COMPANY_ID
         t2.number = NUMBER_2
         t2.name = NAME_2
         t2.amount = AMOUNT_2
         t2.start_date = START_DATE_2
         t2.status = STATUS_2
         self.assertTrue(t2.add())  # Add test object 2
         return t2
コード例 #5
0
ファイル: test_database.py プロジェクト: goldenboy/shared
        def initialize():
            t = DbObject(DBH.test)
            t.tbl_.truncate()
            t.db_.commit()

            t.company_id = COMPANY_ID
            t.number = NUMBER
            t.name = NAME
            t.amount = AMOUNT
            t.start_date = START_DATE
            t.status = STATUS
            self.assertTrue(t.add())
            return t
コード例 #6
0
ファイル: test_database.py プロジェクト: goldenboy/shared
    def test__modify(self):

        t = DbObject(DBH.test)
        t.tbl_.truncate()
        t.db_.commit()

        t.company_id = COMPANY_ID
        t.name = NAME
        t.number = NUMBER
        t.start_date = START_DATE
        t.status = STATUS
        self.assertTrue(t.add())  # Add succeeds

        original_updated_on = t.updated_on

        t.name = NAME_2
        time.sleep(1)  # Pause one sec so the modified date will change
        self.assertTrue(t.modify())  # Modify succeeds

        rows = t.db_.executesql(
            """
            SELECT
                id,
                company_id,
                name,
                number,
                start_date,
                status,
                created_on,
                updated_on
            FROM test;
            """
        )
        self.assertEqual(rows[0][0], t.id)  # Id matches
        self.assertEqual(rows[0][1], COMPANY_ID)  # company_id matches
        self.assertEqual(rows[0][2], NAME_2)  # name matches
        self.assertEqual(rows[0][3], NUMBER)  # number matches
        self.assertEqual(rows[0][4], START_DATE)  # start_date matches
        self.assertEqual(rows[0][5], STATUS)  # status matches
        # Creation and modified dates should be set.
        self.assertTrue(isinstance(rows[0][6], datetime.datetime))
        self.assertTrue(isinstance(rows[0][7], datetime.datetime))

        # modified date is updated
        self.assertNotEqual(t.updated_on, original_updated_on)

        t.tbl_.truncate()
        t.db_.commit()
        return
コード例 #7
0
ファイル: test_database.py プロジェクト: goldenboy/shared
    def test__as_list(self):
        t = DbObject(DBH.test)

        # Uninitialized returns list of default values
        self.assertEqual(t.as_list()[:7], [None, None, None, None, D("0.00"), None, None])

        t.company_id = COMPANY_ID
        t.number = NUMBER
        t.name = NAME
        t.amount = AMOUNT
        t.start_date = START_DATE
        t.status = STATUS
        # Initialized returns expected list
        self.assertEqual(t.as_list()[:7], [None, COMPANY_ID, NUMBER, NAME, AMOUNT, START_DATE, STATUS])
        return
コード例 #8
0
ファイル: test_database.py プロジェクト: goldenboy/shared
 def test__as_dict(self):
     t = DbObject(DBH.test)
     as_d = t.as_dict()
     del as_d["created_on"]
     del as_d["updated_on"]
     self.assertEqual(
         as_d,
         {
             "id": None,
             "company_id": None,
             "number": None,
             "name": None,
             "amount": D("0.00"),
             "start_date": None,
             "status": None,
         },
     )
     t.company_id = COMPANY_ID
     t.number = NUMBER
     t.name = NAME
     t.amount = AMOUNT
     t.start_date = START_DATE
     t.status = STATUS
     as_d = t.as_dict()
     del as_d["created_on"]
     del as_d["updated_on"]
     self.assertEqual(
         as_d,
         {
             "id": None,
             "company_id": COMPANY_ID,
             "number": NUMBER,
             "name": NAME,
             "amount": AMOUNT,
             "start_date": START_DATE,
             "status": STATUS,
         },
     )
     return
コード例 #9
0
ファイル: test_database.py プロジェクト: goldenboy/shared
    def test__update(self):
        def initialize():
            t = DbObject(DBH.test)
            t.tbl_.truncate()
            t.db_.commit()

            t.company_id = COMPANY_ID
            t.number = NUMBER
            t.name = NAME
            t.amount = AMOUNT
            t.start_date = START_DATE
            t.status = STATUS
            self.assertTrue(t.add())
            return t

        records = DbObject(DBH.test).set_.get()
        self.assertEqual(len(records), 0)

        t = initialize()
        records = DbObject(DBH.test).set_.get()
        self.assertEqual(len(records), 1)
        self.assertEqual(records[0], t)

        t2 = DbObject(DBH.test)
        t2.company_id = COMPANY_ID_2
        t2.number = NUMBER_2
        t2.name = NAME_2
        t2.amount = AMOUNT_2
        t2.start_date = START_DATE_2
        t2.status = STATUS_2

        # Record is not in database, should add
        self.assertTrue(t2.update())
        records = DbObject(DBH.test).set_.get()
        self.assertEqual(len(records), 2)
        self.assertNotEqual(t.id, t2.id)

        # Record is in database, should modify
        self.assertTrue(t2.update())
        records = DbObject(DBH.test).set_.get()
        self.assertEqual(len(records), 2)
        self.assertNotEqual(t.id, t2.id)

        # Update record with identical key fields, should modify t
        t = initialize()
        t2 = DbObject(DBH.test)
        t2.company_id = COMPANY_ID
        t2.number = NUMBER_2
        t2.name = NAME
        t2.amount = AMOUNT_2
        t2.start_date = START_DATE_2
        t2.status = STATUS_2

        self.assertTrue(t2.update(key_fields=["company_id", "name"]))

        tests = t.set_.get()
        self.assertEqual(len(tests), 1)
        self.assertEqual(t.id, t2.id)

        # Update record with non-identical key fields but matching id, should
        # modify.
        t = initialize()
        t2 = DbObject(DBH.test)
        t2.id = t.id
        t2.company_id = COMPANY_ID_2
        t2.number = NUMBER_2
        t2.name = NAME_2
        t2.amount = AMOUNT_2
        t2.start_date = START_DATE_2
        t2.status = STATUS_2

        self.assertTrue(t2.update(key_fields=["company_id", "name"]))

        tests = t.set_.get()
        self.assertEqual(len(tests), 1)
        self.assertEqual(t.id, t2.id)
        self.assertEqual(tests[0].company_id, t2.company_id)

        # Update record with non-identical key fields, no id, should add
        t = initialize()
        t2 = DbObject(DBH.test)
        t2.id = None
        t2.company_id = COMPANY_ID_2
        t2.number = NUMBER_2
        t2.name = NAME_2
        t2.amount = AMOUNT_2
        t2.start_date = START_DATE_2
        t2.status = STATUS_2

        self.assertTrue(t2.update(key_fields=["company_id", "name"]))

        tests = t.set_.get()
        self.assertEqual(len(tests), 2)
        self.assertNotEqual(t.id, t2.id)

        # Test with a key field not defined.
        t = initialize()
        t2 = DbObject(DBH.test)
        t2.id = None
        t2.company_id = COMPANY_ID_2

        self.assertTrue(t2.update(key_fields=["company_id", "name"]))

        tests = t.set_.get()
        self.assertEqual(len(tests), 2)
        self.assertNotEqual(t.id, t2.id)

        t3 = DbObject(DBH.test)
        t3.id = None
        t3.company_id = COMPANY_ID_2

        self.assertTrue(t3.update(key_fields=["company_id", "name"]))

        tests = t.set_.get()
        self.assertEqual(len(tests), 2)
        self.assertEqual(t2.id, t3.id)

        # Make sure it distinguishes between NULL and empty string.
        t = initialize()
        t2 = DbObject(DBH.test)
        t2.id = None
        t2.company_id = COMPANY_ID_2
        t2.name = ""

        self.assertTrue(t2.update(key_fields=["company_id", "name"]))

        t3 = DbObject(DBH.test)
        t3.id = None
        t3.company_id = COMPANY_ID_2

        self.assertTrue(t3.update(key_fields=["company_id", "name"]))

        tests = t.set_.get()
        self.assertEqual(len(tests), 3)
        self.assertNotEqual(t2.id, t3.id)

        t.tbl_.truncate()
        t.db_.commit()
        return
コード例 #10
0
ファイル: test_database.py プロジェクト: goldenboy/shared
    def test__add(self):

        t = DbObject(DBH.test)
        t.tbl_.truncate()
        t.db_.commit()

        t.company_id = COMPANY_ID
        t.number = NUMBER
        t.name = NAME
        t.amount = AMOUNT
        t.start_date = START_DATE
        t.status = STATUS
        self.assertTrue(t.add())  # Add succeeds
        self.assertTrue(self.i_re.match(str(t.id)))  # Object id field is set
        # Creation and modified dates should be set.
        self.assertTrue(isinstance(t.created_on, datetime.datetime))
        self.assertTrue(isinstance(t.updated_on, datetime.datetime))

        rows = t.db_.executesql(
            """
            SELECT
                id,
                company_id,
                number,
                name,
                amount,
                start_date,
                status,
                created_on,
                updated_on
            FROM test;
            """
        )
        self.assertEqual(rows[0][0], t.id)  # Id matches
        self.assertEqual(rows[0][1], COMPANY_ID)  # company_id matches
        self.assertEqual(rows[0][2], NUMBER)  # number matches
        self.assertEqual(rows[0][3], NAME)  # name matches
        self.assertEqual(rows[0][4], AMOUNT)  # amount matches
        self.assertEqual(rows[0][5], START_DATE)  # start_date matches
        self.assertEqual(rows[0][6], STATUS)  # status matches
        # Creation and modified dates should be set.
        self.assertTrue(isinstance(rows[0][7], datetime.datetime))
        self.assertTrue(isinstance(rows[0][8], datetime.datetime))

        old_id = t.id
        self.assertTrue(t.add())  # Successive add succeeds
        self.assertTrue(t.id != old_id)  # Object has new id
        self.assertEqual(t.db_(t.tbl_.id > 0).count(), 2)  # Now two records

        t.tbl_.truncate()
        t.db_.commit()

        # Test MySQL column default values.
        t2 = DbObject(DBH.test)
        self.assertEqual(t2.status, None)  # Status initially null.
        t2.add()
        rows = t2.db_.executesql(
            """
            SELECT
                id,
                company_id,
                number,
                name,
                amount,
                start_date,
                status,
                created_on,
                updated_on
            FROM test;
            """
        )
        self.assertEqual(rows[0][0], t2.id)  # Id matches
        self.assertEqual(rows[0][6], "a")  # status matches default
        self.assertEqual(t2.status, "a")  # Instance is updated.

        t2.tbl_.truncate()
        t2.db_.commit()
        return
コード例 #11
0
ファイル: test_database.py プロジェクト: goldenboy/shared
    def test____init__(self):
        # R0915: *Too many statements (%s/%s)*
        # pylint: disable=R0915

        t = DbObject(DBH.test)
        DBH.executesql("""DELETE from test;""")
        self.assertTrue(t)  # Object created

        # Ensure object has default properties
        self.assertTrue(hasattr(t, "db_"))
        self.assertTrue(hasattr(t, "set_"))
        self.assertTrue(isinstance(t.set_, Collection))

        # Test a few object properties
        self.assertTrue(hasattr(t, "company_id"))
        self.assertTrue(hasattr(t, "status"))

        # Test setting and getting field values
        t.company_id = COMPANY_ID
        t.status = "d"
        self.assertEqual(t.company_id, COMPANY_ID)
        self.assertEqual(t.status, "d")

        self.assertEqual(t.db_(t.tbl_.id > 0).count(), 0)  # Has no records

        # Raw sql

        t.db_.executesql(
            """
            INSERT into test
                (company_id, status)
            VALUES
                (%s, '%s');
            """
            % (COMPANY_ID, STATUS)
        )
        self.assertEqual(t.db_(t.tbl_.id > 0).count(), 1)  # Has one record
        rows = t.db_.executesql(
            """
            SELECT id, company_id, name, status from test;
            """
        )
        self.assertTrue(self.i_re.match(str(rows[0][0])))  # Id is integer
        self.assertEqual(rows[0][1], COMPANY_ID)  # company_id matches
        self.assertEqual(rows[0][2], None)  # unset name is None
        self.assertEqual(rows[0][3], STATUS)  # status matches

        t.tbl_.truncate()
        t.db_.commit()
        self.assertEqual(t.db_(t.tbl_.id > 0).count(), 0)  # Has no records

        # Test passing attributes as parameters.

        t = DbObject(DBH.test)
        # No parameters, name is None
        self.assertEqual(t.name, None)
        t = DbObject(DBH.test, name=NAME)
        # Provide name as parameter, gets set.
        self.assertEqual(t.name, NAME)
        t = DbObject(DBH.test, name=NAME, xxxx="__fake__")
        # Provide name as parameter, gets set.
        # No field attribute is not set.
        self.assertEqual(t.name, NAME)
        self.assertFalse(hasattr(t, "xxxx"))

        # Test set_ property.
        t = DbObject(DBH.test)
        t.tbl_.truncate()
        t.db_.commit()

        t.company_id = COMPANY_ID
        t.name = NAME
        t.number = NUMBER
        t.start_date = START_DATE
        t.status = STATUS
        self.assertTrue(t.add())  # Add succeeds

        t2 = DbObject(DBH.test)
        t2.company_id = COMPANY_ID
        t2.name = NAME_2
        t2.number = NUMBER_2
        t2.start_date = START_DATE_2
        t2.status = STATUS_2
        self.assertTrue(t2.add())  # Add (2) succeeds

        collection = t.set_.get(id=t.id)
        self.assertEqual(collection.objects, [t])
        collection = t.set_.get(id=t2.id)
        self.assertEqual(collection.objects, [t2])
        collection = t.set_.get(query=t.tbl_.name == "fake")
        self.assertEqual(collection.objects, [])
        collection = t.set_.get(query=t.tbl_.status != "")
        self.assertEqual(collection.objects, [t, t2])

        return