Ejemplo n.º 1
0
    def test_get_and_add_time_range_with_string_with_error(self):
        def query_args_func(key):
            args = {"time_range": "foo"}
            return args.get(key, [])

        expected = {
            "created_on": {
                "$gte":
                datetime.datetime(2015, 7, 13, 11, 35, 00, tzinfo=tz_util.utc),
                "$lt":
                datetime.datetime(2015, 7, 13, 12, 35, 00, tzinfo=tz_util.utc)
            }
        }

        now_value = datetime.datetime(2015,
                                      7,
                                      13,
                                      12,
                                      35,
                                      00,
                                      tzinfo=tz_util.utc)

        patcher = mock.patch("datetime.datetime", spec=True)
        patched_date = patcher.start()
        patched_date.now.return_value = now_value
        self.addCleanup(patcher.stop)

        spec = {}

        get_and_add_time_range(spec, query_args_func)
        self.assertDictEqual(expected, spec)
Ejemplo n.º 2
0
    def test_get_and_add_time_range_no_value(self):
        def query_args_func(key):
            return []

        expected = {}
        spec = {}

        get_and_add_time_range(spec, query_args_func)
        self.assertDictEqual(expected, spec)