Example #1
0
 def testState(self):
     state = dict(val=100)
     fs = FunctionSource(dataFunction, state=state, resetFieldName="reset", sequenceIdFieldName="sequence")
     self.assertIsNotNone(fs)
     r = fs.getNextRecordDict()
     self.assertIsNotNone(r)
     r = fs.getNextRecordDict()
     r = fs.getNextRecordDict()
     self.assertEqual(103, state["val"])
Example #2
0
 def testState(self):
     state = dict(val=100)
     fs = FunctionSource(dataFunction,
                         state=state,
                         resetFieldName="reset",
                         sequenceIdFieldName="sequence")
     self.assertIsNotNone(fs)
     r = fs.getNextRecordDict()
     self.assertIsNotNone(r)
     r = fs.getNextRecordDict()
     r = fs.getNextRecordDict()
     self.assertEqual(103, state["val"])
Example #3
0
 def testResetAndSequenceFields(self):
     fs = FunctionSource(dataFunction,
                         state=None,
                         resetFieldName="reset",
                         sequenceIdFieldName="sequence")
     self.assertIsNotNone(fs)
     r = fs.getNextRecordDict()
     self.assertIsNotNone(r)
Example #4
0
 def testDefaultArgs(self):
     fs = FunctionSource(dataFunction,
                         state=None,
                         resetFieldName=None,
                         sequenceIdFieldName=None)
     self.assertIsNotNone(fs)
     r = fs.getNextRecordDict()
     self.assertIsNotNone(r)
Example #5
0
    def testPickle(self):
        state = dict(val=100)
        fs = FunctionSource(dataFunction, state=state, resetFieldName="reset", sequenceIdFieldName="sequence")
        self.assertIsNotNone(fs)
        r = fs.getNextRecordDict()
        self.assertIsNotNone(r)

        pkl = pickle.dumps(fs)
        self.assertIsNotNone(pkl)

        fs2 = pickle.loads(pkl)
        self.assertIsNotNone(fs2)

        r = fs2.getNextRecordDict()
        r = fs2.getNextRecordDict()
        self.assertEqual(103, fs2.state["val"])
Example #6
0
    def testPickle(self):
        state = dict(val=100)
        fs = FunctionSource(dataFunction,
                            state=state,
                            resetFieldName="reset",
                            sequenceIdFieldName="sequence")
        self.assertIsNotNone(fs)
        r = fs.getNextRecordDict()
        self.assertIsNotNone(r)

        pkl = pickle.dumps(fs)
        self.assertIsNotNone(pkl)

        fs2 = pickle.loads(pkl)
        self.assertIsNotNone(fs2)

        r = fs2.getNextRecordDict()
        r = fs2.getNextRecordDict()
        self.assertEqual(103, fs2.state["val"])
Example #7
0
 def testResetAndSequenceFields(self):
     fs = FunctionSource(dataFunction, state=None, resetFieldName="reset", sequenceIdFieldName="sequence")
     self.assertIsNotNone(fs)
     r = fs.getNextRecordDict()
     self.assertIsNotNone(r)
Example #8
0
 def testDefaultArgs(self):
     fs = FunctionSource(dataFunction, state=None, resetFieldName=None, sequenceIdFieldName=None)
     self.assertIsNotNone(fs)
     r = fs.getNextRecordDict()
     self.assertIsNotNone(r)